From 901fa77be8ae238b1c1477021987adfbd9a0e791 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Piotr=20Ma=C5=9Blanka?= <piotr.maslanka@henrietta.com.pl>
Date: Fri, 3 Jan 2020 18:44:56 +0100
Subject: [PATCH] client set to terminate itself

---
 stress_tests/client/__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/stress_tests/client/__init__.py b/stress_tests/client/__init__.py
index 5fc64fd..dd108fe 100644
--- a/stress_tests/client/__init__.py
+++ b/stress_tests/client/__init__.py
@@ -119,9 +119,9 @@ def run(client_notify, result_client, server_notify, server_result):
     client_notify.get()
 
     cad.start()
-
+    started_at = time.monotonic()
     terminating = False
-    while not terminating:
+    while not terminating and (time.monotonic() < started_at + 40):  # run for 40 seconds
         try:
             client_notify.get(timeout=1.0)
             terminating = True
@@ -130,5 +130,7 @@ def run(client_notify, result_client, server_notify, server_result):
         except KeyboardInterrupt:
             break
 
+    server_notify.put(None)
+
     lftf.close()
     # logger.warning('Got %s connections', len(cad.connections))
-- 
GitLab