testit-python-commons 4.2.2__tar.gz → 4.2.3__tar.gz

{testit_python_commons-4.2.2 → testit_python_commons-4.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: testit-python-commons
-Version: 4.2.2
+Version: 4.2.3
 Summary: Python commons for Test IT
 Home-page: https://github.com/testit-tms/adapters-python/
 Author: Integration team
@@ -17,7 +17,7 @@ Classifier: Programming Language :: Python :: 3.12
 Description-Content-Type: text/markdown
 Requires-Dist: pluggy
 Requires-Dist: tomli
-Requires-Dist: testit-api-client==7.5.6
+Requires-Dist: testit-api-client==7.5.10
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{testit_python_commons-4.2.2 → testit_python_commons-4.2.3}/setup.py

@@ -1,6 +1,6 @@
 from setuptools import find_packages, setup
 
-VERSION = "4.2.2"
+VERSION = "4.2.3"
 
 setup(
     name='testit-python-commons',
@@ -25,5 +25,5 @@ setup(
     py_modules=['testit', 'testit_python_commons'],
     packages=find_packages(where='src'),
     package_dir={'': 'src'},
-    install_requires=['pluggy', 'tomli', 'testit-api-client==7.5.6']
+    install_requires=['pluggy', 'tomli', 'testit-api-client==7.5.10']
 )

{testit_python_commons-4.2.2 → testit_python_commons-4.2.3}/src/testit_python_commons/client/api_client.py

@@ -24,7 +24,6 @@ from testit_api_client.models import (
     TestRunV2ApiResult,
     LinkAutoTestToWorkItemRequest,
     AutoTestWorkItemIdentifierApiResult,
-    ProjectModel,
     WorkflowApiResult,
 )
 
@@ -547,7 +546,7 @@ class ApiClientWorker:
 
     @adapter_logger
     @retry
-    def __get_project(self) -> ProjectModel:
+    def __get_project(self):
         return self.__projects_api.get_project_by_id(id=self.__config.get_project_id())
 
     @adapter_logger
@@ -557,7 +556,7 @@ class ApiClientWorker:
 
     @adapter_logger
     def __get_status_codes(self) -> List[str]:
-        project: ProjectModel = self.__get_project()
-        workflow: WorkflowApiResult = self.__get_workflow_by_id(project.workflow_id)
+        project = self.__get_project()
+        workflow = self.__get_workflow_by_id(project.workflow_id)
 
         return [status.code for status in workflow.statuses]

{testit_python_commons-4.2.2 → testit_python_commons-4.2.3}/src/testit_python_commons/services/sync_storage/sync_storage_runner.py

@@ -1,3 +1,4 @@
+import json
 import logging
 import os
 import platform
@@ -39,13 +40,15 @@ class SyncStorageRunner:
     across multiple workers.
     """
 
-    SYNC_STORAGE_VERSION = "v0.3.2"
+    SYNC_STORAGE_VERSION = "v0.3.3"
     SYNC_STORAGE_REPO_URL = (
         "https://github.com/testit-tms/sync-storage-public/releases/download/"
     )
     AMD64 = "amd64"
     ARM64 = "arm64"
     SYNC_STORAGE_STARTUP_TIMEOUT = 5  # seconds
+    KEEP_ALIVE_INTERVAL_SECONDS = 30
+    KEEP_ALIVE_REQUEST_TIMEOUT_SECONDS = 5
 
     def __init__(
         self,
@@ -82,6 +85,9 @@ class SyncStorageRunner:
         self.workers_api: Optional[WorkersApi] = None
         self.test_results_api: Optional[TestResultsApi] = None
 
+        self._keep_alive_stop_event = threading.Event()
+        self._keep_alive_thread: Optional[threading.Thread] = None
+
         logger.debug(
             f"Initialized SyncStorageRunner with test_run_id={test_run_id}, port={self.port}"
         )
@@ -113,6 +119,7 @@ class SyncStorageRunner:
                 except Exception as e:
                     logger.error(f"Error registering worker: {e}")
 
+                self._start_keep_alive()
                 return True
 
             # Get executable file name for current platform
@@ -170,6 +177,7 @@ class SyncStorageRunner:
                 except Exception as e:
                     logger.error(f"Error registering worker: {e}")
 
+                self._start_keep_alive()
                 return True
             else:
                 raise RuntimeError("Cannot start the SyncStorage until timeout")
@@ -213,6 +221,49 @@ class SyncStorageRunner:
         """Get the Sync Storage URL."""
         return f"http://localhost:{self.port}"
 
+    def _keep_alive(self) -> None:
+        try:
+            payload = json.dumps({
+                "pid": self.worker_pid,
+                "testRunId": self.test_run_id,
+            }).encode("utf-8")
+            request = urllib.request.Request(
+                f"{self.get_url()}/keep_alive",
+                data=payload,
+                headers={"Content-Type": "application/json"},
+                method="POST",
+            )
+            urllib.request.urlopen(
+                request,
+                timeout=self.KEEP_ALIVE_REQUEST_TIMEOUT_SECONDS,
+            )
+        except Exception:
+            pass
+
+    def _keep_alive_loop(self) -> None:
+        while not self._keep_alive_stop_event.is_set():
+            self._keep_alive()
+            if self._keep_alive_stop_event.wait(self.KEEP_ALIVE_INTERVAL_SECONDS):
+                break
+
+    def _start_keep_alive(self) -> None:
+        if self._keep_alive_thread and self._keep_alive_thread.is_alive():
+            return
+
+        self._keep_alive_stop_event.clear()
+        self._keep_alive_thread = threading.Thread(
+            target=self._keep_alive_loop,
+            name="sync-storage-keep-alive",
+            daemon=True,
+        )
+        self._keep_alive_thread.start()
+        logger.debug("Sync Storage keep-alive thread started")
+
+    def _stop_keep_alive(self) -> None:
+        self._keep_alive_stop_event.set()
+        if self._keep_alive_thread and self._keep_alive_thread.is_alive():
+            self._keep_alive_thread.join(timeout=1)
+
     def send_in_progress_test_result(
         self, model: TestResultCutApiModel
     ) -> bool:

{testit_python_commons-4.2.2 → testit_python_commons-4.2.3}/src/testit_python_commons.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: testit-python-commons
-Version: 4.2.2
+Version: 4.2.3
 Summary: Python commons for Test IT
 Home-page: https://github.com/testit-tms/adapters-python/
 Author: Integration team
@@ -17,7 +17,7 @@ Classifier: Programming Language :: Python :: 3.12
 Description-Content-Type: text/markdown
 Requires-Dist: pluggy
 Requires-Dist: tomli
-Requires-Dist: testit-api-client==7.5.6
+Requires-Dist: testit-api-client==7.5.10
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

testit_python_commons-4.2.3/src/testit_python_commons.egg-info/requires.txt

@@ -0,0 +1,3 @@
+pluggy
+tomli
+testit-api-client==7.5.10

testit_python_commons-4.2.2/src/testit_python_commons.egg-info/requires.txt

@@ -1,3 +0,0 @@
-pluggy
-tomli
-testit-api-client==7.5.6