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