Merge pull request #120 from DLHub-Argonne/test-timeout

Add timeout to tests
DLHub-Argonne · Sep 27, 2021 · 81edce5 · 81edce5
2 parents 524993c + 4f8d0cc
commit 81edce5
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/dlhub_sdk/utils/futures.py b/dlhub_sdk/utils/futures.py
@@ -33,7 +33,7 @@ def __init__(self, client, task_id: str, ping_interval: float):
             assert AttributeError('Ping interval must be at least 1 second')
 
         # Start a thread that polls status
-        self._checker_thread = Thread(target=DLHubFuture._ping_server, args=(self,))
+        self._checker_thread = Thread(target=DLHubFuture._ping_server, args=(self,), daemon=True)
         self._checker_thread.start()
 
     def _ping_server(self):

diff --git a/setup.cfg b/setup.cfg
@@ -4,3 +4,6 @@ universal = 1
 [flake8]
 exclude = .git,*.egg*
 max-line-length = 150
+
+[tool:pytest]
+timeout = 60
diff --git a/test-requirements.txt b/test-requirements.txt
@@ -6,3 +6,4 @@ torch>=1.1.0
 pytest
 pytest-cov
 mdf-connect-client>=0.3.8
+pytest-timeout