xref: /aosp_15_r20/external/grpc-grpc/test/core/end2end/tests/invoke_large_request.cc (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
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 <string.h>
20 
21 #include <memory>
22 
23 #include "gtest/gtest.h"
24 
25 #include <grpc/impl/channel_arg_names.h>
26 #include <grpc/status.h>
27 
28 #include "src/core/lib/channel/channel_args.h"
29 #include "src/core/lib/gprpp/time.h"
30 #include "src/core/lib/slice/slice.h"
31 #include "test/core/end2end/end2end_tests.h"
32 
33 namespace grpc_core {
34 namespace {
35 
CORE_END2END_TEST(Http2SingleHopTest,InvokeLargeRequest)36 CORE_END2END_TEST(Http2SingleHopTest, InvokeLargeRequest) {
37   const size_t kMessageSize = 10 * 1024 * 1024;
38   auto send_from_client = RandomSlice(kMessageSize);
39   auto send_from_server = RandomSlice(kMessageSize);
40   InitServer(
41       ChannelArgs().Set(GRPC_ARG_MAX_RECEIVE_MESSAGE_LENGTH, kMessageSize));
42   InitClient(
43       ChannelArgs().Set(GRPC_ARG_MAX_RECEIVE_MESSAGE_LENGTH, kMessageSize));
44   auto c = NewClientCall("/foo").Timeout(Duration::Minutes(5)).Create();
45   CoreEnd2endTest::IncomingStatusOnClient server_status;
46   CoreEnd2endTest::IncomingMetadata server_initial_metadata;
47   CoreEnd2endTest::IncomingMessage server_message;
48   c.NewBatch(1)
49       .SendInitialMetadata({})
50       .SendMessage(send_from_client.Ref())
51       .SendCloseFromClient()
52       .RecvInitialMetadata(server_initial_metadata)
53       .RecvMessage(server_message)
54       .RecvStatusOnClient(server_status);
55   auto s = RequestCall(101);
56   Expect(101, true);
57   Step(Duration::Minutes(1));
58   CoreEnd2endTest::IncomingMessage client_message;
59   s.NewBatch(102).SendInitialMetadata({}).RecvMessage(client_message);
60   Expect(102, true);
61   Step(Duration::Minutes(1));
62   CoreEnd2endTest::IncomingCloseOnServer client_close;
63   s.NewBatch(103)
64       .SendStatusFromServer(GRPC_STATUS_UNIMPLEMENTED, "xyz", {})
65       .SendMessage(send_from_server.Ref())
66       .RecvCloseOnServer(client_close);
67   Expect(103, true);
68   Expect(1, true);
69   Step(Duration::Minutes(1));
70   EXPECT_EQ(server_status.status(), GRPC_STATUS_UNIMPLEMENTED);
71   EXPECT_EQ(server_status.message(), "xyz");
72   EXPECT_EQ(s.method(), "/foo");
73   EXPECT_FALSE(client_close.was_cancelled());
74   EXPECT_EQ(client_message.payload(), send_from_client);
75   EXPECT_EQ(server_message.payload(), send_from_server);
76 }
77 
78 }  // namespace
79 }  // namespace grpc_core
80