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()