From 87dda31a2c234a7596b661ad5d63f74f3124ea25 Mon Sep 17 00:00:00 2001
From: Vitaly Baranov <vitlibar@clickhouse.com>
Date: Wed, 3 Jul 2024 20:36:19 +0200
Subject: [PATCH] Add test for GRPCServer's shutdown.

---
 tests/integration/test_grpc_protocol/test.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/integration/test_grpc_protocol/test.py b/tests/integration/test_grpc_protocol/test.py
index 851da99acf3..1ace5b361b8 100644
--- a/tests/integration/test_grpc_protocol/test.py
+++ b/tests/integration/test_grpc_protocol/test.py
@@ -39,6 +39,7 @@ node = cluster.add_instance(
         "TSAN_OPTIONS": "report_atomic_races=0 " + os.getenv("TSAN_OPTIONS", default="")
     },
     ipv6_address=IPV6_ADDRESS,
+    stay_alive=True,
 )
 main_channel = None
 
@@ -763,3 +764,9 @@ def test_opentelemetry_context_propagation():
         )
         == "SELECT 1\tsome custom state\n"
     )
+
+
+def test_restart():
+    assert query("SELECT 1") == "1\n"
+    node.restart_clickhouse()
+    assert query("SELECT 2") == "2\n"