1 /*
2 *
3 * Copyright 2016 gRPC authors.
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 *
17 */
18
19 #include <grpcpp/test/server_context_test_spouse.h>
20
21 #include <cstring>
22 #include <vector>
23
24 #include <grpcpp/impl/grpc_library.h>
25 #include <gtest/gtest.h>
26
27 namespace grpc {
28 namespace testing {
29
30 static internal::GrpcLibraryInitializer g_initializer;
31
32 const char key1[] = "metadata-key1";
33 const char key2[] = "metadata-key2";
34 const char val1[] = "metadata-val1";
35 const char val2[] = "metadata-val2";
36
ClientMetadataContains(const ServerContext & context,const grpc::string_ref & key,const grpc::string_ref & value)37 bool ClientMetadataContains(const ServerContext& context,
38 const grpc::string_ref& key,
39 const grpc::string_ref& value) {
40 const auto& client_metadata = context.client_metadata();
41 for (auto iter = client_metadata.begin(); iter != client_metadata.end();
42 ++iter) {
43 if (iter->first == key && iter->second == value) {
44 return true;
45 }
46 }
47 return false;
48 }
49
TEST(ServerContextTestSpouseTest,ClientMetadata)50 TEST(ServerContextTestSpouseTest, ClientMetadata) {
51 ServerContext context;
52 ServerContextTestSpouse spouse(&context);
53
54 spouse.AddClientMetadata(key1, val1);
55 ASSERT_TRUE(ClientMetadataContains(context, key1, val1));
56
57 spouse.AddClientMetadata(key2, val2);
58 ASSERT_TRUE(ClientMetadataContains(context, key1, val1));
59 ASSERT_TRUE(ClientMetadataContains(context, key2, val2));
60 }
61
TEST(ServerContextTestSpouseTest,InitialMetadata)62 TEST(ServerContextTestSpouseTest, InitialMetadata) {
63 ServerContext context;
64 ServerContextTestSpouse spouse(&context);
65 std::multimap<grpc::string, grpc::string> metadata;
66
67 context.AddInitialMetadata(key1, val1);
68 metadata.insert(std::pair<grpc::string, grpc::string>(key1, val1));
69 ASSERT_EQ(metadata, spouse.GetInitialMetadata());
70
71 context.AddInitialMetadata(key2, val2);
72 metadata.insert(std::pair<grpc::string, grpc::string>(key2, val2));
73 ASSERT_EQ(metadata, spouse.GetInitialMetadata());
74 }
75
TEST(ServerContextTestSpouseTest,TrailingMetadata)76 TEST(ServerContextTestSpouseTest, TrailingMetadata) {
77 ServerContext context;
78 ServerContextTestSpouse spouse(&context);
79 std::multimap<grpc::string, grpc::string> metadata;
80
81 context.AddTrailingMetadata(key1, val1);
82 metadata.insert(std::pair<grpc::string, grpc::string>(key1, val1));
83 ASSERT_EQ(metadata, spouse.GetTrailingMetadata());
84
85 context.AddTrailingMetadata(key2, val2);
86 metadata.insert(std::pair<grpc::string, grpc::string>(key2, val2));
87 ASSERT_EQ(metadata, spouse.GetTrailingMetadata());
88 }
89
90 } // namespace testing
91 } // namespace grpc
92
main(int argc,char ** argv)93 int main(int argc, char** argv) {
94 ::testing::InitGoogleTest(&argc, argv);
95 return RUN_ALL_TESTS();
96 }
97