xref: /aosp_15_r20/external/grpc-grpc/test/core/end2end/tests/graceful_server_shutdown.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 <memory>
20 
21 #include "gtest/gtest.h"
22 
23 #include <grpc/status.h>
24 
25 #include "src/core/lib/gprpp/time.h"
26 #include "test/core/end2end/end2end_tests.h"
27 
28 namespace grpc_core {
29 namespace {
30 
CORE_END2END_TEST(Http2Test,GracefulServerShutdown)31 CORE_END2END_TEST(Http2Test, GracefulServerShutdown) {
32   auto c = NewClientCall("/foo").Timeout(Duration::Seconds(10)).Create();
33   CoreEnd2endTest::IncomingStatusOnClient server_status;
34   CoreEnd2endTest::IncomingMetadata server_initial_metadata;
35   c.NewBatch(1)
36       .SendInitialMetadata({})
37       .SendCloseFromClient()
38       .RecvInitialMetadata(server_initial_metadata)
39       .RecvStatusOnClient(server_status);
40   auto s = RequestCall(101);
41   Expect(101, true);
42   Step();
43   // shutdown and destroy the server
44   ShutdownServerAndNotify(200);
45   Step();
46   CoreEnd2endTest::IncomingCloseOnServer client_close;
47   s.NewBatch(102)
48       .SendInitialMetadata({})
49       .SendStatusFromServer(GRPC_STATUS_UNIMPLEMENTED, "xyz", {})
50       .RecvCloseOnServer(client_close);
51   Expect(102, true);
52   Expect(200, true);
53   Expect(1, true);
54   Step();
55 
56   EXPECT_EQ(server_status.status(), GRPC_STATUS_UNIMPLEMENTED);
57   EXPECT_EQ(s.method(), "/foo");
58   EXPECT_FALSE(client_close.was_cancelled());
59 }
60 
61 }  // namespace
62 }  // namespace grpc_core
63