Add EvaluateArgs test.
This commit is contained in:
Ashitha Santhosh
parent
dca24b8de5
commit
b1fd4a45a4
|
|
@ -14,7 +14,9 @@
|
|||
|
||||
#include <grpc/support/port_platform.h>
|
||||
|
||||
#include <gmock/gmock.h>
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include "absl/strings/string_view.h"
|
||||
|
||||
#include "src/core/lib/security/authorization/evaluate_args.h"
|
||||
|
|
@ -45,27 +47,172 @@ class EvaluateArgsTest : public ::testing::Test {
|
|||
|
||||
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.";
|
||||
EXPECT_EQ(src_address, local_address_);
|
||||
}
|
||||
|
||||
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.";
|
||||
EXPECT_EQ(src_port, local_port_);
|
||||
}
|
||||
|
||||
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. ";
|
||||
EXPECT_EQ(dest_address, peer_address_);
|
||||
}
|
||||
|
||||
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.";
|
||||
EXPECT_EQ(dest_port, peer_port_);
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsMetadataTest, HandlesNullMetadata) {
|
||||
EvaluateArgs eval_args(nullptr, nullptr, nullptr);
|
||||
EXPECT_EQ(eval_args.GetPath(), nullptr);
|
||||
EXPECT_EQ(eval_args.GetMethod(), nullptr);
|
||||
EXPECT_EQ(eval_args.GetHost(), nullptr);
|
||||
EXPECT_THAT(eval_args.GetHeaders(), ::testing::ElementsAre());
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsMetadataTest, HandlesEmptyMetadata) {
|
||||
grpc_metadata_batch metadata;
|
||||
grpc_metadata_batch_init(&metadata);
|
||||
EvaluateArgs eval_args(&metadata, nullptr, nullptr);
|
||||
EXPECT_EQ(eval_args.GetPath(), nullptr);
|
||||
EXPECT_EQ(eval_args.GetMethod(), nullptr);
|
||||
EXPECT_EQ(eval_args.GetHost(), nullptr);
|
||||
EXPECT_THAT(eval_args.GetHeaders(), ::testing::ElementsAre());
|
||||
grpc_metadata_batch_destroy(&metadata);
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsMetadataTest, GetPathSuccess) {
|
||||
grpc_init();
|
||||
const char* kPath = "/some/path";
|
||||
grpc_metadata_batch metadata;
|
||||
grpc_metadata_batch_init(&metadata);
|
||||
grpc_slice fake_val = grpc_slice_intern(grpc_slice_from_static_string(kPath));
|
||||
grpc_mdelem fake_val_md = grpc_mdelem_from_slices(GRPC_MDSTR_PATH, fake_val);
|
||||
grpc_linked_mdelem storage;
|
||||
storage.md = fake_val_md;
|
||||
ASSERT_EQ(grpc_metadata_batch_link_head(&metadata, &storage),
|
||||
GRPC_ERROR_NONE);
|
||||
EvaluateArgs eval_args(&metadata, nullptr, nullptr);
|
||||
EXPECT_EQ(eval_args.GetPath(), kPath);
|
||||
grpc_metadata_batch_destroy(&metadata);
|
||||
grpc_shutdown();
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsMetadataTest, GetHostSuccess) {
|
||||
grpc_init();
|
||||
const char* kHost = "host";
|
||||
grpc_metadata_batch metadata;
|
||||
grpc_metadata_batch_init(&metadata);
|
||||
grpc_slice fake_val = grpc_slice_intern(grpc_slice_from_static_string(kHost));
|
||||
grpc_mdelem fake_val_md = grpc_mdelem_from_slices(GRPC_MDSTR_HOST, fake_val);
|
||||
grpc_linked_mdelem storage;
|
||||
storage.md = fake_val_md;
|
||||
ASSERT_EQ(grpc_metadata_batch_link_head(&metadata, &storage),
|
||||
GRPC_ERROR_NONE);
|
||||
EvaluateArgs eval_args(&metadata, nullptr, nullptr);
|
||||
EXPECT_EQ(eval_args.GetHost(), kHost);
|
||||
grpc_metadata_batch_destroy(&metadata);
|
||||
grpc_shutdown();
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsMetadataTest, GetMethodSuccess) {
|
||||
grpc_init();
|
||||
const char* kMethod = "GET";
|
||||
grpc_metadata_batch metadata;
|
||||
grpc_metadata_batch_init(&metadata);
|
||||
grpc_slice fake_val =
|
||||
grpc_slice_intern(grpc_slice_from_static_string(kMethod));
|
||||
grpc_mdelem fake_val_md =
|
||||
grpc_mdelem_from_slices(GRPC_MDSTR_METHOD, fake_val);
|
||||
grpc_linked_mdelem storage;
|
||||
storage.md = fake_val_md;
|
||||
ASSERT_EQ(grpc_metadata_batch_link_head(&metadata, &storage),
|
||||
GRPC_ERROR_NONE);
|
||||
EvaluateArgs eval_args(&metadata, nullptr, nullptr);
|
||||
EXPECT_EQ(eval_args.GetMethod(), kMethod);
|
||||
grpc_metadata_batch_destroy(&metadata);
|
||||
grpc_shutdown();
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsMetadataTest, GetHeadersSuccess) {
|
||||
grpc_init();
|
||||
const char* kPath = "/some/path";
|
||||
const char* kHost = "host";
|
||||
grpc_metadata_batch metadata;
|
||||
grpc_metadata_batch_init(&metadata);
|
||||
grpc_slice fake_path =
|
||||
grpc_slice_intern(grpc_slice_from_static_string(kPath));
|
||||
grpc_mdelem fake_path_md =
|
||||
grpc_mdelem_from_slices(GRPC_MDSTR_PATH, fake_path);
|
||||
grpc_linked_mdelem storage;
|
||||
storage.md = fake_path_md;
|
||||
ASSERT_EQ(grpc_metadata_batch_link_head(&metadata, &storage, GRPC_BATCH_PATH),
|
||||
GRPC_ERROR_NONE);
|
||||
grpc_slice fake_host =
|
||||
grpc_slice_intern(grpc_slice_from_static_string(kHost));
|
||||
grpc_mdelem fake_host_md =
|
||||
grpc_mdelem_from_slices(GRPC_MDSTR_HOST, fake_host);
|
||||
grpc_linked_mdelem storage2;
|
||||
storage2.md = fake_host_md;
|
||||
ASSERT_EQ(
|
||||
grpc_metadata_batch_link_tail(&metadata, &storage2, GRPC_BATCH_HOST),
|
||||
GRPC_ERROR_NONE);
|
||||
EvaluateArgs eval_args(&metadata, nullptr, nullptr);
|
||||
EXPECT_THAT(
|
||||
eval_args.GetHeaders(),
|
||||
::testing::UnorderedElementsAre(
|
||||
::testing::Pair(StringViewFromSlice(GRPC_MDSTR_HOST), kHost),
|
||||
::testing::Pair(StringViewFromSlice(GRPC_MDSTR_PATH), kPath)));
|
||||
grpc_metadata_batch_destroy(&metadata);
|
||||
grpc_shutdown();
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsAuthContextTest, HandlesNullAuthContext) {
|
||||
EvaluateArgs eval_args(nullptr, nullptr, nullptr);
|
||||
EXPECT_EQ(eval_args.GetSpiffeId(), nullptr);
|
||||
EXPECT_EQ(eval_args.GetCertServerName(), nullptr);
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsAuthContextTest, HandlesEmptyAuthCtx) {
|
||||
grpc_auth_context auth_context(nullptr);
|
||||
EvaluateArgs eval_args(nullptr, &auth_context, nullptr);
|
||||
EXPECT_EQ(eval_args.GetSpiffeId(), nullptr);
|
||||
EXPECT_EQ(eval_args.GetCertServerName(), nullptr);
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsAuthContextTest, GetSpiffeIdSuccessOneProperty) {
|
||||
grpc_auth_context auth_context(nullptr);
|
||||
const char* kId = "spiffeid";
|
||||
auth_context.add_cstring_property(GRPC_PEER_SPIFFE_ID_PROPERTY_NAME, kId);
|
||||
EvaluateArgs eval_args(nullptr, &auth_context, nullptr);
|
||||
EXPECT_EQ(eval_args.GetSpiffeId(), kId);
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsAuthContextTest, GetSpiffeIdFailDuplicateProperty) {
|
||||
grpc_auth_context auth_context(nullptr);
|
||||
auth_context.add_cstring_property(GRPC_PEER_SPIFFE_ID_PROPERTY_NAME, "id1");
|
||||
auth_context.add_cstring_property(GRPC_PEER_SPIFFE_ID_PROPERTY_NAME, "id2");
|
||||
EvaluateArgs eval_args(nullptr, &auth_context, nullptr);
|
||||
EXPECT_EQ(eval_args.GetSpiffeId(), nullptr);
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsAuthContextTest, GetCertServerNameSuccessOneProperty) {
|
||||
grpc_auth_context auth_context(nullptr);
|
||||
const char* kServer = "server";
|
||||
auth_context.add_cstring_property(GRPC_X509_CN_PROPERTY_NAME, kServer);
|
||||
EvaluateArgs eval_args(nullptr, &auth_context, nullptr);
|
||||
EXPECT_EQ(eval_args.GetCertServerName(), kServer);
|
||||
}
|
||||
|
||||
TEST(EvaluateArgsAuthContextTest, GetCertServerNameFailDuplicateProperty) {
|
||||
grpc_auth_context auth_context(nullptr);
|
||||
auth_context.add_cstring_property(GRPC_X509_CN_PROPERTY_NAME, "server1");
|
||||
auth_context.add_cstring_property(GRPC_X509_CN_PROPERTY_NAME, "server2");
|
||||
EvaluateArgs eval_args(nullptr, &auth_context, nullptr);
|
||||
EXPECT_EQ(eval_args.GetCertServerName(), nullptr);
|
||||
}
|
||||
|
||||
} // namespace grpc_core
|
||||
|
|
|
|||
Loading…
Reference in New Issue