Update test

Author: robmarkcole

tests/test_deepstack.py

@@ -1,7 +1,9 @@
 # Placeholder
 
 import deepstack.core as ds
+import requests
 import requests_mock
+import pytest
 
 MOCK_IP_ADDRESS = "localhost"
 MOCK_PORT = 5000
@@ -46,7 +48,8 @@
 CONFIDENCE_THRESHOLD = 0.7
 
 
-def test_DeepstackObject():
+def test_DeepstackObject_process_image_bytes():
+    """Test a good response from server."""
     with requests_mock.Mocker() as mock_req:
         mock_req.post(MOCK_URL, status_code=ds.HTTP_OK, json=MOCK_RESPONSE)
 
@@ -55,6 +58,17 @@ def test_DeepstackObject():
         assert dsobject.predictions == MOCK_PREDICTIONS
 
 
+def test_DeepstackObject_process_image_bytes_timeout():
+    """Test a timeout. THIS SHOULD FAIL"""
+    with pytest.raises(ds.DeepstackException) as excinfo:
+        with requests_mock.Mocker() as mock_req:
+            mock_req.post(MOCK_URL, exc=requests.exceptions.ConnectTimeout)
+            dsobject = ds.DeepstackObject(MOCK_IP_ADDRESS, MOCK_PORT)
+            dsobject.process_image_bytes(MOCK_BYTES)
+            assert False
+            assert "SHOULD FAIL" in str(excinfo.value)
+
+
 def test_get_object_labels():
     """Cant always be sure order of returned list items."""
     object_labels = ds.get_object_labels(MOCK_PREDICTIONS)