From 2c9b3c941cd1879fe6135fadf83362360d90012b Mon Sep 17 00:00:00 2001
From: Christian Schwarz <christian@neon.tech>
Date: Tue, 9 Jan 2024 15:11:03 +0000
Subject: [PATCH] python tests: broker: support stopping using SIGKILL

---
 test_runner/fixtures/broker.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/test_runner/fixtures/broker.py b/test_runner/fixtures/broker.py
index fa8b816e69..8aca90a097 100644
--- a/test_runner/fixtures/broker.py
+++ b/test_runner/fixtures/broker.py
@@ -54,7 +54,10 @@ class NeonBroker:
             else:
                 break  # success
 
-    def stop(self):
+    def stop(self, immediate: bool = False):
         if self.handle is not None:
-            self.handle.terminate()
+            if immediate:
+                self.handle.kill()
+            else:
+                self.handle.terminate()
             self.handle.wait()