Handle different content types

Author: Maxime Belanger

tests/ubersmith_request_get_test.py

@@ -65,7 +65,7 @@ def test_api_get_method_returns_with_arguments(self, request_mock):
         expected_call()
 
     def expect_a_ubersmith_call(self, requests_mock, returning=None, **kwargs):
-        response = MagicMock(status_code=200)
+        response = MagicMock(status_code=200, headers={"content-type": "application/json"})
         requests_mock.get = MagicMock(return_value=response)
         response.json = MagicMock(return_value=returning)
 

tests/ubersmith_request_post_test.py

@@ -76,7 +76,7 @@ def test_api_raises_exception_with_if_data_status_is_false(self, requests_mock):
         assert_that(calling(ubersmith_api.client.miss), raises(ubersmith_client.exceptions.UbersmithException))
 
     def expect_a_ubersmith_call_post(self, requests_mock, returning=None, status_code=200, **kwargs):
-        response = MagicMock(status_code=status_code)
+        response = MagicMock(status_code=status_code, headers={"content-type": "application/json"})
         requests_mock.post = MagicMock(return_value=response)
         response.json = MagicMock(return_value=returning)
 

tests/ubersmith_request_test.py

@@ -16,10 +16,11 @@
 import ubersmith_client
 from mock import Mock, patch
 
-from hamcrest import assert_that, raises, calling
+from hamcrest import assert_that, raises, calling, equal_to
 from requests.exceptions import ConnectionError, Timeout
 
-from ubersmith_client.exceptions import UbersmithException, BadRequest, UnknownError, Forbidden, NotFound, Unauthorized, UbersmithConnectionError, \
+from ubersmith_client.exceptions import UbersmithException, BadRequest, UnknownError, Forbidden, NotFound, Unauthorized, \
+    UbersmithConnectionError, \
     UbersmithTimeout
 from tests.ubersmith_json.response_data_structure import a_response_data
 from ubersmith_client.ubersmith_request import UbersmithRequest
@@ -32,8 +33,8 @@ def setUp(self):
         self.password = 'test'
 
     def test_process_ubersmith_response(self):
-        response = Mock()
-        response.status_code = 200
+        response = Mock(status_code=200, headers={"content-type": "application/json"})
+
         json_data = {
             'client_id': '1',
             'first': 'Rick',
@@ -45,9 +46,12 @@ def test_process_ubersmith_response(self):
 
         self.assertDictEqual(json_data, UbersmithRequest.process_ubersmith_response(response))
 
+    def test_process_ubersmith_response_not_application_json(self):
+        response = Mock(status_code=200, headers={"content-type": "text/html"}, content="42")
+        assert_that(response.content, equal_to(UbersmithRequest.process_ubersmith_response(response)))
+
     def test_process_ubersmith_response_raise_exception(self):
-        response = Mock()
-        response.status_code = 400
+        response = Mock(status_code=400, headers={"content-type": "application/json"})
         assert_that(calling(UbersmithRequest.process_ubersmith_response).with_args(response), raises(BadRequest))
 
         response.status_code = 401

ubersmith_client/exceptions.py

@@ -70,4 +70,4 @@ def __init__(self, url):
 class UbersmithTimeout(UbersmithException):
     def __init__(self, url, timeout):
         super(UbersmithTimeout, self)\
-            .__init__(message='Trying to connect to {url} times out after {timeout}'.format(url=url, timeout=timeout))
+            .__init__(message='Trying to connect to {url} timed out after {timeout} seconds'.format(url=url, timeout=timeout))

ubersmith_client/ubersmith_request.py

@@ -32,7 +32,7 @@ def __getattr__(self, function):
 
     @abstractmethod
     def __call__(self, **kwargs):
-        raise
+        raise AttributeError
 
     def _process_request(self, method, **kwargs):
         try:
@@ -52,11 +52,14 @@ def process_ubersmith_response(response):
         if response.status_code < 200 or response.status_code >= 400:
             raise get_exception_for(status_code=response.status_code)
 
-        response_json = response.json()
-        if not response_json['status']:
-            raise UbersmithException(
-                response_json['error_code'],
-                response_json['error_message']
-            )
+        if response.headers['content-type'] == 'application/json':
+            response_json = response.json()
+            if not response_json['status']:
+                raise UbersmithException(
+                    response_json['error_code'],
+                    response_json['error_message']
+                )
 
-        return response.json()["data"]
+            return response.json()["data"]
+
+        return response.content