mirror of https://github.com/grpc/grpc.git
Client context test peer (#25610)
* Create client_context_test_peer.h
mock-up class for ClientContext
* Add ClientContextTestPeer as a friend for test
* Update client_context_test_peer.h
* Update client_context_test_peer.h
* unit test for client_context_test_peer.h
* unit test for client_context_test_peer.h
* fix for tools/buildgen/generate_projects.sh
* Revert "fix for tools/buildgen/generate_projects.sh"
This reverts commit b007aa8c2b.
* re-fix for tools/buildgen/generate_projects.sh
* Remove an empty line
Co-authored-by: Vijay Pai <vpai@google.com>
pull/26369/head
shangwu-google
4 years ago
committed by
GitHub
parent
66253c521a
commit
8b3f4fca25
8 changed files with 235 additions and 0 deletions
@ -0,0 +1,62 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2021 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 GRPCPP_TEST_CLIENT_CONTEXT_TEST_PEER_H |
||||
#define GRPCPP_TEST_CLIENT_CONTEXT_TEST_PEER_H |
||||
|
||||
#include <grpcpp/client_context.h> |
||||
|
||||
#include <map> |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
|
||||
/// A test-only class to access private members and methods of ClientContext.
|
||||
class ClientContextTestPeer { |
||||
public: |
||||
explicit ClientContextTestPeer(ClientContext* const ctx) : ctx_(ctx) {} |
||||
|
||||
/// Inject metadata to the ClientContext for the test. The test peer
|
||||
/// must be alive when a ClientContext::GetServerInitialMetadata is called.
|
||||
void AddServerInitialMetadata(const std::string& key, |
||||
const std::string& value) { |
||||
server_initial_metadata_storage_.insert( |
||||
std::pair<std::string, std::string>(key, value)); |
||||
ctx_->initial_metadata_received_ = true; |
||||
ctx_->recv_initial_metadata_.map()->clear(); |
||||
for (const auto& item : server_initial_metadata_storage_) { |
||||
ctx_->recv_initial_metadata_.map()->insert( |
||||
std::pair<grpc::string_ref, grpc::string_ref>( |
||||
item.first.c_str(), |
||||
grpc::string_ref(item.second.data(), item.second.size()))); |
||||
} |
||||
} |
||||
|
||||
std::multimap<std::string, std::string> GetSendInitialMetadata() const { |
||||
return ctx_->send_initial_metadata_; |
||||
} |
||||
|
||||
private: |
||||
ClientContext* const ctx_; // not owned
|
||||
std::multimap<std::string, std::string> server_initial_metadata_storage_; |
||||
}; |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc
|
||||
|
||||
#endif // GRPCPP_TEST_CLIENT_CONTEXT_TEST_PEER_H
|
||||
@ -0,0 +1,81 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2021 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 <grpcpp/test/client_context_test_peer.h> |
||||
|
||||
#include <cstring> |
||||
#include <vector> |
||||
|
||||
#include <grpcpp/impl/grpc_library.h> |
||||
#include <gtest/gtest.h> |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
|
||||
static internal::GrpcLibraryInitializer g_initializer; |
||||
|
||||
const char key1[] = "metadata-key1"; |
||||
const char key2[] = "metadata-key2"; |
||||
const char val1[] = "metadata-val1"; |
||||
const char val2[] = "metadata-val2"; |
||||
|
||||
bool ServerInitialMetadataContains(const ClientContext& context, |
||||
const grpc::string_ref& key, |
||||
const grpc::string_ref& value) { |
||||
const auto& server_metadata = context.GetServerInitialMetadata(); |
||||
for (auto iter = server_metadata.begin(); iter != server_metadata.end(); |
||||
++iter) { |
||||
if (iter->first == key && iter->second == value) { |
||||
return true; |
||||
} |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
TEST(ClientContextTestPeerTest, AddServerInitialMetadata) { |
||||
ClientContext context; |
||||
ClientContextTestPeer peer(&context); |
||||
|
||||
peer.AddServerInitialMetadata(key1, val1); |
||||
ASSERT_TRUE(ServerInitialMetadataContains(context, key1, val1)); |
||||
peer.AddServerInitialMetadata(key2, val2); |
||||
ASSERT_TRUE(ServerInitialMetadataContains(context, key1, val1)); |
||||
ASSERT_TRUE(ServerInitialMetadataContains(context, key2, val2)); |
||||
} |
||||
|
||||
TEST(ClientContextTestPeerTest, GetSendInitialMetadata) { |
||||
ClientContext context; |
||||
ClientContextTestPeer peer(&context); |
||||
std::multimap<std::string, std::string> metadata; |
||||
|
||||
context.AddMetadata(key1, val1); |
||||
metadata.insert(std::pair<std::string, std::string>(key1, val1)); |
||||
ASSERT_EQ(metadata, peer.GetSendInitialMetadata()); |
||||
|
||||
context.AddMetadata(key2, val2); |
||||
metadata.insert(std::pair<std::string, std::string>(key2, val2)); |
||||
ASSERT_EQ(metadata, peer.GetSendInitialMetadata()); |
||||
} |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc
|
||||
|
||||
int main(int argc, char** argv) { |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
return RUN_ALL_TESTS(); |
||||
} |
||||
Loading…
Reference in new issue