mirror of https://github.com/grpc/grpc.git
ashithasantosh
4 years ago
committed by
Ashitha Santhosh
parent
c79bef55ee
commit
5820e833fc
26 changed files with 702 additions and 16 deletions
@ -0,0 +1,76 @@ |
|||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include <gtest/gtest.h> |
||||||
|
#include "absl/strings/string_view.h" |
||||||
|
|
||||||
|
#include "src/core/lib/security/authorization/evaluate_args.h" |
||||||
|
#include "test/core/util/eval_args_mock_endpoint.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
class EvaluateArgsTest : public ::testing::Test { |
||||||
|
protected: |
||||||
|
void SetUp() override { |
||||||
|
local_address_ = "255.255.255.255"; |
||||||
|
peer_address_ = "128.128.128.128"; |
||||||
|
local_port_ = 413; |
||||||
|
peer_port_ = 314; |
||||||
|
endpoint_ = CreateEvalArgsMockEndpoint(local_address_.c_str(), local_port_, |
||||||
|
peer_address_.c_str(), peer_port_); |
||||||
|
evaluate_args_ = |
||||||
|
absl::make_unique<EvaluateArgs>(nullptr, nullptr, endpoint_); |
||||||
|
} |
||||||
|
void TearDown() override { grpc_endpoint_destroy(endpoint_); } |
||||||
|
grpc_endpoint* endpoint_; |
||||||
|
std::unique_ptr<EvaluateArgs> evaluate_args_; |
||||||
|
std::string local_address_; |
||||||
|
std::string peer_address_; |
||||||
|
int local_port_; |
||||||
|
int peer_port_; |
||||||
|
}; |
||||||
|
|
||||||
|
TEST_F(EvaluateArgsTest, TestEvaluateArgsLocalAddress) { |
||||||
|
absl::string_view src_address = evaluate_args_->GetLocalAddress(); |
||||||
|
EXPECT_EQ(src_address, local_address_) |
||||||
|
<< "Error: Failed to extract correct Local address from EvaluateArgs."; |
||||||
|
} |
||||||
|
|
||||||
|
TEST_F(EvaluateArgsTest, TestEvaluateArgsLocalPort) { |
||||||
|
int src_port = evaluate_args_->GetLocalPort(); |
||||||
|
EXPECT_EQ(src_port, local_port_) |
||||||
|
<< "Error: Failed to extract correct Local port from EvaluateArgs."; |
||||||
|
} |
||||||
|
|
||||||
|
TEST_F(EvaluateArgsTest, TestEvaluateArgsPeerAddress) { |
||||||
|
absl::string_view dest_address = evaluate_args_->GetPeerAddress(); |
||||||
|
EXPECT_EQ(dest_address, peer_address_) |
||||||
|
<< "Error: Failed to extract correct Peer address from " |
||||||
|
"EvaluateArgs. "; |
||||||
|
} |
||||||
|
|
||||||
|
TEST_F(EvaluateArgsTest, TestEvaluateArgsPeerPort) { |
||||||
|
int dest_port = evaluate_args_->GetPeerPort(); |
||||||
|
EXPECT_EQ(dest_port, peer_port_) |
||||||
|
<< "Error: Failed to extract correct Peer port from EvaluateArgs."; |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
int main(int argc, char** argv) { |
||||||
|
::testing::InitGoogleTest(&argc, argv); |
||||||
|
return RUN_ALL_TESTS(); |
||||||
|
} |
||||||
@ -0,0 +1,118 @@ |
|||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include "test/core/util/eval_args_mock_endpoint.h" |
||||||
|
|
||||||
|
#include <inttypes.h> |
||||||
|
|
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include "absl/strings/str_format.h" |
||||||
|
|
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
#include <grpc/support/string_util.h> |
||||||
|
#include "src/core/lib/iomgr/sockaddr.h" |
||||||
|
#include "src/core/lib/iomgr/sockaddr_utils.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
class EvalArgsMockEndpoint { |
||||||
|
public: |
||||||
|
EvalArgsMockEndpoint(absl::string_view local_uri, absl::string_view peer_uri) |
||||||
|
: local_address_(local_uri), peer_(peer_uri) { |
||||||
|
base_.vtable = &vtable_; |
||||||
|
} |
||||||
|
grpc_endpoint* base() const { return const_cast<grpc_endpoint*>(&base_); } |
||||||
|
static void Read(grpc_endpoint* ep, grpc_slice_buffer* slices, |
||||||
|
grpc_closure* cb, bool unused) {} |
||||||
|
static void Write(grpc_endpoint* ep, grpc_slice_buffer* slices, |
||||||
|
grpc_closure* cb, void* unused) {} |
||||||
|
static void AddToPollset(grpc_endpoint* ep, grpc_pollset* unused) {} |
||||||
|
static void AddToPollsetSet(grpc_endpoint* ep, grpc_pollset_set* unused) {} |
||||||
|
static void DeleteFromPollsetSet(grpc_endpoint* ep, |
||||||
|
grpc_pollset_set* unused) {} |
||||||
|
static void Shutdown(grpc_endpoint* ep, grpc_error* why) {} |
||||||
|
static void Destroy(grpc_endpoint* ep) { |
||||||
|
EvalArgsMockEndpoint* m = reinterpret_cast<EvalArgsMockEndpoint*>(ep); |
||||||
|
delete m; |
||||||
|
} |
||||||
|
|
||||||
|
static absl::string_view GetPeer(grpc_endpoint* ep) { |
||||||
|
EvalArgsMockEndpoint* m = reinterpret_cast<EvalArgsMockEndpoint*>(ep); |
||||||
|
return m->peer_; |
||||||
|
} |
||||||
|
|
||||||
|
static absl::string_view GetLocalAddress(grpc_endpoint* ep) { |
||||||
|
EvalArgsMockEndpoint* m = reinterpret_cast<EvalArgsMockEndpoint*>(ep); |
||||||
|
return m->local_address_; |
||||||
|
} |
||||||
|
|
||||||
|
static grpc_resource_user* GetResourceUser(grpc_endpoint* ep) { |
||||||
|
return nullptr; |
||||||
|
} |
||||||
|
|
||||||
|
static int GetFd(grpc_endpoint* unused) { return -1; } |
||||||
|
static bool CanTrackErr(grpc_endpoint* unused) { return false; } |
||||||
|
|
||||||
|
private: |
||||||
|
static constexpr grpc_endpoint_vtable vtable_ = { |
||||||
|
EvalArgsMockEndpoint::Read, |
||||||
|
EvalArgsMockEndpoint::Write, |
||||||
|
EvalArgsMockEndpoint::AddToPollset, |
||||||
|
EvalArgsMockEndpoint::AddToPollsetSet, |
||||||
|
EvalArgsMockEndpoint::DeleteFromPollsetSet, |
||||||
|
EvalArgsMockEndpoint::Shutdown, |
||||||
|
EvalArgsMockEndpoint::Destroy, |
||||||
|
EvalArgsMockEndpoint::GetResourceUser, |
||||||
|
EvalArgsMockEndpoint::GetPeer, |
||||||
|
EvalArgsMockEndpoint::GetLocalAddress, |
||||||
|
EvalArgsMockEndpoint::GetFd, |
||||||
|
EvalArgsMockEndpoint::CanTrackErr}; |
||||||
|
grpc_endpoint base_; |
||||||
|
std::string local_address_; |
||||||
|
std::string peer_; |
||||||
|
}; |
||||||
|
|
||||||
|
constexpr grpc_endpoint_vtable EvalArgsMockEndpoint::vtable_; |
||||||
|
|
||||||
|
namespace { |
||||||
|
|
||||||
|
std::string NameAndPortToURI(const char* addr, const int port) { |
||||||
|
grpc_sockaddr_in address; |
||||||
|
memset(&address, 0, sizeof(address)); |
||||||
|
address.sin_family = AF_INET; |
||||||
|
address.sin_port = htons(port); |
||||||
|
inet_pton(AF_INET, addr, &address.sin_addr); |
||||||
|
grpc_resolved_address resolved; |
||||||
|
memset(&resolved, 0, sizeof(resolved)); |
||||||
|
memcpy(resolved.addr, &address, sizeof(address)); |
||||||
|
resolved.len = sizeof(address); |
||||||
|
return grpc_sockaddr_to_uri(&resolved); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
grpc_endpoint* CreateEvalArgsMockEndpoint(const char* local_address, |
||||||
|
const int local_port, |
||||||
|
const char* peer_address, |
||||||
|
const int peer_port) { |
||||||
|
EvalArgsMockEndpoint* m = |
||||||
|
new EvalArgsMockEndpoint(NameAndPortToURI(local_address, local_port), |
||||||
|
NameAndPortToURI(peer_address, peer_port)); |
||||||
|
return m->base(); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
@ -0,0 +1,31 @@ |
|||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
#ifndef GRPC_TEST_CORE_UTIL_EVAL_ARGS_MOCK_ENDPOINT_H |
||||||
|
#define GRPC_TEST_CORE_UTIL_EVAL_ARGS_MOCK_ENDPOINT_H |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include "src/core/lib/iomgr/endpoint.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
grpc_endpoint* CreateEvalArgsMockEndpoint(const char* local_address, |
||||||
|
const int local_port, |
||||||
|
const char* peer_address, |
||||||
|
const int peer_port); |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif // GRPC_TEST_CORE_UTIL_EVAL_ARGS_MOCK_ENDPOINT_H
|
||||||
Loading…
Reference in new issue