pytest-bec-e2e 3.161.0__tar.gz → 3.162.0__tar.gz

{pytest_bec_e2e-3.161.0 → pytest_bec_e2e-3.162.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-bec-e2e
-Version: 3.161.0
+Version: 3.162.0
 Summary: BEC pytest plugin for end-to-end tests
 Project-URL: Bug Tracker, https://github.com/bec-project/bec/issues
 Project-URL: Homepage, https://github.com/bec-project/bec

{pytest_bec_e2e-3.161.0 → pytest_bec_e2e-3.162.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pytest-bec-e2e"
-version = "3.161.0"
+version = "3.162.0"
 description = "BEC pytest plugin for end-to-end tests"
 requires-python = ">=3.11"
 classifiers = [

{pytest_bec_e2e-3.161.0 → pytest_bec_e2e-3.162.0}/pytest_bec_e2e/plugin.py

@@ -21,6 +21,9 @@ from bec_lib.tests.utils import wait_for_empty_queue
 
 RedisConnector.RETRY_ON_TIMEOUT = 1
 
+CLIENT_SHUTDOWN_TIMEOUT_S = 10
+SERVICE_STOP_TIMEOUT_S = 20
+
 
 class LogTestTool:
     def __init__(self, client: BECIPythonClient):
@@ -29,7 +32,8 @@ class LogTestTool:
 
     def fetch(self, count: int | None = None) -> None:
         """Fetch logs from the server and store them for interrogation, get all by default or get
-        the last `count` logs (latter will not fetch read logs again if they have been seen!)"""
+        the last `count` logs (latter will not fetch read logs again if they have been seen!)
+        """
         log_data = self._conn.xread(MessageEndpoints.log(), from_start=(count is None), count=count)
         if log_data is None:
             self._logs = None
@@ -204,7 +208,7 @@ def bec_servers(
         try:
             yield
         finally:
-            service_handler.stop(processes)
+            service_handler.stop(processes, timeout_s=SERVICE_STOP_TIMEOUT_S)
     else:
         # Nothing to do here: servers are supposed to be started externally.
         yield
@@ -221,7 +225,7 @@ def bec_ipython_client_with_demo_config(
     try:
         yield bec
     finally:
-        bec.shutdown()
+        bec.shutdown(per_thread_timeout_s=CLIENT_SHUTDOWN_TIMEOUT_S)
         bec._client._reset_singleton()
 
 
@@ -234,7 +238,7 @@ def bec_client_lib_with_demo_config(bec_redis_fixture, bec_services_config_file_
     try:
         yield bec
     finally:
-        bec.shutdown()
+        bec.shutdown(per_thread_timeout_s=CLIENT_SHUTDOWN_TIMEOUT_S)
         bec._client._reset_singleton()