Add a unit test

pull/2313/head
yang-g 10 years ago
parent 5ea46ab248
commit 8068f86974
  1. 45
      Makefile
  2. 13
      build.json
  3. 6
      src/core/security/client_auth_filter.c
  4. 6
      src/cpp/common/secure_auth_context.cc
  5. 2
      test/core/security/auth_context_test.c
  6. 79
      test/cpp/common/secure_auth_context_test.cc
  7. 13
      tools/run_tests/sources_and_headers.json
  8. 9
      tools/run_tests/tests.json

File diff suppressed because one or more lines are too long

@ -2290,6 +2290,19 @@
"grpc++_test_config"
]
},
{
"name": "secure_auth_context_test",
"build": "test",
"language": "c++",
"src": [
"test/cpp/common/secure_auth_context_test.cc"
],
"deps": [
"grpc++",
"grpc",
"gpr"
]
},
{
"name": "server_crash_test",
"build": "test",

@ -214,10 +214,12 @@ static void auth_start_transport_op(grpc_call_element *elem,
sec_ctx = op->context[GRPC_CONTEXT_SECURITY].value;
if (sec_ctx->auth_context == NULL) {
sec_ctx->auth_context =
grpc_auth_context_ref(chand->security_connector->base.auth_context);
GRPC_AUTH_CONTEXT_REF(chand->security_connector->base.auth_context,
"client_auth_filter");
} else {
sec_ctx->auth_context->chained =
grpc_auth_context_ref(chand->security_connector->base.auth_context);
GRPC_AUTH_CONTEXT_REF(chand->security_connector->base.auth_context,
"client_auth_filter chained");
}
}

@ -38,9 +38,11 @@
namespace grpc {
SecureAuthContext::SecureAuthContext(grpc_auth_context* ctx)
: ctx_(grpc_auth_context_ref(ctx)) {}
: ctx_(GRPC_AUTH_CONTEXT_REF(ctx, "SecureAuthContext")) {}
SecureAuthContext::~SecureAuthContext() { grpc_auth_context_unref(ctx_); }
SecureAuthContext::~SecureAuthContext() {
GRPC_AUTH_CONTEXT_UNREF(ctx_, "SecureAuthContext");
}
std::vector<grpc::string> SecureAuthContext::GetPeerIdentity() const {
if (!ctx_) {

@ -31,7 +31,7 @@
*
*/
#include<string.h>
#include <string.h>
#include "src/core/security/security_context.h"
#include "src/core/support/string.h"

@ -0,0 +1,79 @@
/*
*
* Copyright 2015, Google Inc.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following disclaimer
* in the documentation and/or other materials provided with the
* distribution.
* * Neither the name of Google Inc. nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
#include <grpc++/auth_context.h>
#include <gtest/gtest.h>
#include "src/cpp/common/secure_auth_context.h"
#include "src/core/security/security_context.h"
namespace grpc {
namespace {
class SecureAuthContextTest : public ::testing::Test {};
// Created with nullptr
TEST_F(SecureAuthContextTest, EmptyContext) {
SecureAuthContext context(nullptr);
EXPECT_TRUE(context.GetPeerIdentity().empty());
EXPECT_TRUE(context.GetPeerIdentityPropertyName().empty());
EXPECT_TRUE(context.FindPropertyValues("").empty());
EXPECT_TRUE(context.FindPropertyValues("whatever").empty());
}
TEST_F(SecureAuthContextTest, Properties) {
grpc_auth_context* ctx = grpc_auth_context_create(NULL, 3);
ctx->properties[0] = grpc_auth_property_init_from_cstring("name", "chapi");
ctx->properties[1] = grpc_auth_property_init_from_cstring("name", "chapo");
ctx->properties[2] = grpc_auth_property_init_from_cstring("foo", "bar");
ctx->peer_identity_property_name = ctx->properties[0].name;
SecureAuthContext context(ctx);
std::vector<grpc::string> peer_identity = context.GetPeerIdentity();
EXPECT_EQ(2, peer_identity.size());
EXPECT_EQ("chapi", peer_identity[0]);
EXPECT_EQ("chapo", peer_identity[1]);
EXPECT_EQ("name", context.GetPeerIdentityPropertyName());
std::vector<grpc::string> bar = context.FindPropertyValues("foo");
EXPECT_EQ(1, bar.size());
EXPECT_EQ("bar", bar[0]);
GRPC_AUTH_CONTEXT_UNREF(ctx, "SecureAuthContextTest");
}
} // namespace
} // namespace grpc
int main(int argc, char **argv) {
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}

@ -1441,6 +1441,19 @@
"test/cpp/qps/worker.cc"
]
},
{
"deps": [
"gpr",
"grpc",
"grpc++"
],
"headers": [],
"language": "c++",
"name": "secure_auth_context_test",
"src": [
"test/cpp/common/secure_auth_context_test.cc"
]
},
{
"deps": [
"gpr",

@ -702,6 +702,15 @@
"posix"
]
},
{
"flaky": false,
"language": "c++",
"name": "secure_auth_context_test",
"platforms": [
"windows",
"posix"
]
},
{
"flaky": false,
"language": "c++",

Loading…
Cancel
Save