Update raised exception for invalid server response

ohsu-comp-bio · Mar 2, 2024 · 8178eab · 8178eab
1 parent d4fa6e5
commit 8178eab
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/tes/client.py b/tes/client.py
@@ -241,7 +241,7 @@ def check_success(data: Task) -> bool:
             try:
                 response = self.get_task(task_id, "MINIMAL")
             except Exception:
-                pass
+                raise Exception(f"Failed to get task {task_id}")
 
             if response is not None:
                 if check_success(response):

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -206,6 +206,18 @@ def test_wait(self):
             )
             self.cli.wait(self.mock_id, timeout=2)
 
+    def test_wait_exception(self):
+        with requests_mock.Mocker() as m:
+            m.get(
+                "%s/ga4gh/tes/v1/tasks/%s" % (self.mock_url, self.mock_id),
+                status_code=200,
+                json={
+                    "Error": "Error",
+                }
+            )
+            with self.assertRaises(Exception):
+                self.cli.wait(self.mock_id, timeout=2)
+
     def test_wait_no_state_change(self):
         with requests_mock.Mocker() as m:
             m.get(