xref: /aosp_15_r20/external/grpc-grpc/src/ruby/end2end/graceful_sig_handling_test.rb (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
1#!/usr/bin/env ruby
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# smoke test for a grpc-using app that receives and
18# handles process-ending signals
19
20require_relative './end2end_common'
21
22def main
23  STDERR.puts 'start server'
24  echo_service = EchoServerImpl.new
25  server_runner = ServerRunner.new(echo_service)
26  server_port = server_runner.run
27  STDERR.puts 'start client'
28  client_controller = ClientController.new(
29    'graceful_sig_handling_client.rb', server_port)
30  # now get the client to send an RPC
31  client_controller.stub.do_echo_rpc(
32    ClientControl::DoEchoRpcRequest.new(request: 'hello'))
33  STDERR.puts 'killing client'
34  Process.kill('SIGINT', client_controller.client_pid)
35  Process.wait(client_controller.client_pid)
36  client_exit_status = $CHILD_STATUS
37  if client_exit_status.exited?
38    if client_exit_status.exitstatus != 0
39      STDERR.puts 'Client did not close gracefully'
40      exit(1)
41    end
42  else
43    STDERR.puts 'Client did not close gracefully'
44    exit(1)
45  end
46
47  STDERR.puts 'Client ended gracefully'
48  server_runner.stop
49end
50
51main
52