13306: Adds test for when get_current_container results in a 404
authorEric Biagiotti <ebiagiotti@veritasgenetcs.com>
Thu, 17 Jan 2019 16:11:35 +0000 (11:11 -0500)
committerEric Biagiotti <ebiagiotti@veritasgenetcs.com>
Thu, 17 Jan 2019 16:11:35 +0000 (11:11 -0500)
Arvados-DCO-1.1-Signed-off-by:  Eric Biagiotti <ebiagiotti@veritasgenetics.com>

sdk/cwl/tests/test_util.py

index 2200d995166eb325c30cad7db02cef4b26b3ab05..3ca02c7df65f023c32dab341224534f588d32e67 100644 (file)
@@ -45,4 +45,12 @@ class TestUtil(unittest.TestCase):
         logger = mock.MagicMock()
 
         with self.assertRaises(ApiError):
-            get_current_container(api, num_retries=0, logger=logger)
\ No newline at end of file
+            get_current_container(api, num_retries=0, logger=logger)
+
+    def test_get_current_container_404_error(self):
+        api = mock.MagicMock()
+        api.containers().current().execute.side_effect = ApiError(httplib2.Response({"status": 404}), bytes(b""))
+        logger = mock.MagicMock()
+
+        current_container = get_current_container(api, num_retries=0, logger=logger)
+        self.assertEqual(current_container, None)
\ No newline at end of file