Bug fix: Fixed status

DLHub-Argonne · Mar 14, 2019 · be3421e · be3421e
1 parent a06cc8e
commit be3421e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/dlhub_sdk/client.py b/dlhub_sdk/client.py
@@ -126,7 +126,7 @@ def get_task_status(self, task_id):
         """
 
         r = self.get("{task_id}/status".format(task_id=task_id))
-        return r.json()
+        return r.data
 
     def describe_servable(self, name):
         """Get the description for a certain servable

diff --git a/dlhub_sdk/tests/test_dlhub_client.py b/dlhub_sdk/tests/test_dlhub_client.py
@@ -192,3 +192,8 @@ def test_basic_search(self):
     @skipUnless(is_travis, 'Namespace test is only valid with credentials used on Travis')
     def test_namespace(self):
         self.assertEqual(self.dl.get_username(), 'dlhub.test_gmail')
+
+    @skipUnless(is_travis, 'Status')
+    def test_status(self):
+        self.assertEqual('SUCCEEDED', self.dl.get_task_status('9f37b474-0f9f-4f3c-846b-314fc4241fe2')['status'])
+