1 //
2 //
3 // Copyright 2015 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 <memory>
20
21 #include "gtest/gtest.h"
22
23 #include <grpc/impl/channel_arg_names.h>
24 #include <grpc/status.h>
25
26 #include "src/core/lib/channel/channel_args.h"
27 #include "src/core/lib/gprpp/time.h"
28 #include "src/core/lib/slice/slice.h"
29 #include "test/core/end2end/end2end_tests.h"
30
31 namespace grpc_core {
32
BinaryMetadata(CoreEnd2endTest & test,bool server_true_binary,bool client_true_binary)33 static void BinaryMetadata(CoreEnd2endTest& test, bool server_true_binary,
34 bool client_true_binary) {
35 test.InitServer(
36 ChannelArgs().Set(GRPC_ARG_HTTP2_ENABLE_TRUE_BINARY, server_true_binary));
37 test.InitClient(
38 ChannelArgs().Set(GRPC_ARG_HTTP2_ENABLE_TRUE_BINARY, client_true_binary));
39
40 auto key1_payload = RandomBinarySlice(32);
41 auto key2_payload = RandomBinarySlice(18);
42 auto key3_payload = RandomBinarySlice(17);
43 auto key4_payload = RandomBinarySlice(68);
44 auto key5_payload = RandomBinarySlice(33);
45 auto key6_payload = RandomBinarySlice(2);
46 auto request_payload = RandomBinarySlice(7);
47 auto response_payload = RandomBinarySlice(9);
48 auto status_string = RandomBinarySlice(256);
49
50 auto c = test.NewClientCall("/foo").Timeout(Duration::Minutes(1)).Create();
51 CoreEnd2endTest::IncomingMetadata server_initial_md;
52 CoreEnd2endTest::IncomingMessage server_message;
53 CoreEnd2endTest::IncomingStatusOnClient server_status;
54 c.NewBatch(1)
55 .SendInitialMetadata({
56 {"key1-bin", key1_payload.as_string_view()},
57 {"key2-bin", key2_payload.as_string_view()},
58 })
59 .SendMessage(request_payload.Ref())
60 .SendCloseFromClient()
61 .RecvInitialMetadata(server_initial_md)
62 .RecvMessage(server_message)
63 .RecvStatusOnClient(server_status);
64 auto s = test.RequestCall(101);
65 test.Expect(101, true);
66 test.Step();
67 CoreEnd2endTest::IncomingMessage client_message;
68 s.NewBatch(102)
69 .SendInitialMetadata({
70 {"key3-bin", key3_payload.as_string_view()},
71 {"key4-bin", key4_payload.as_string_view()},
72 })
73 .RecvMessage(client_message);
74 test.Expect(102, true);
75 test.Step();
76 CoreEnd2endTest::IncomingCloseOnServer client_close;
77 s.NewBatch(103)
78 .RecvCloseOnServer(client_close)
79 .SendMessage(response_payload.Ref())
80 .SendStatusFromServer(GRPC_STATUS_OK, status_string.as_string_view(),
81 {
82 {"key5-bin", key5_payload.as_string_view()},
83 {"key6-bin", key6_payload.as_string_view()},
84 });
85 test.Expect(103, true);
86 test.Expect(1, true);
87 test.Step();
88
89 EXPECT_EQ(server_status.status(), GRPC_STATUS_OK);
90 EXPECT_EQ(server_status.message(), status_string.as_string_view());
91 EXPECT_EQ(s.method(), "/foo");
92 EXPECT_FALSE(client_close.was_cancelled());
93 EXPECT_EQ(client_message.payload(), request_payload);
94 EXPECT_EQ(server_message.payload(), response_payload);
95 EXPECT_EQ(s.GetInitialMetadata("key1-bin"), key1_payload.as_string_view());
96 EXPECT_EQ(s.GetInitialMetadata("key2-bin"), key2_payload.as_string_view());
97 EXPECT_EQ(server_initial_md.Get("key3-bin"), key3_payload.as_string_view());
98 EXPECT_EQ(server_initial_md.Get("key4-bin"), key4_payload.as_string_view());
99 EXPECT_EQ(server_status.GetTrailingMetadata("key5-bin"),
100 key5_payload.as_string_view());
101 EXPECT_EQ(server_status.GetTrailingMetadata("key6-bin"),
102 key6_payload.as_string_view());
103 }
104
CORE_END2END_TEST(CoreEnd2endTest,BinaryMetadataServerTrueBinaryClientHttp2Fallback)105 CORE_END2END_TEST(CoreEnd2endTest,
106 BinaryMetadataServerTrueBinaryClientHttp2Fallback) {
107 BinaryMetadata(*this, true, false);
108 }
109
CORE_END2END_TEST(CoreEnd2endTest,BinaryMetadataServerHttp2FallbackClientTrueBinary)110 CORE_END2END_TEST(CoreEnd2endTest,
111 BinaryMetadataServerHttp2FallbackClientTrueBinary) {
112 BinaryMetadata(*this, false, true);
113 }
114
CORE_END2END_TEST(CoreEnd2endTest,BinaryMetadataServerTrueBinaryClientTrueBinary)115 CORE_END2END_TEST(CoreEnd2endTest,
116 BinaryMetadataServerTrueBinaryClientTrueBinary) {
117 BinaryMetadata(*this, true, true);
118 }
119
CORE_END2END_TEST(CoreEnd2endTest,BinaryMetadataServerHttp2FallbackClientHttp2Fallback)120 CORE_END2END_TEST(CoreEnd2endTest,
121 BinaryMetadataServerHttp2FallbackClientHttp2Fallback) {
122 BinaryMetadata(*this, false, false);
123 }
124
125 } // namespace grpc_core
126