Add test for http code 401

transifex · Jun 9, 2016 · 6ee2b60 · 6ee2b60
1 parent 2d1223c
commit 6ee2b60
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -2,7 +2,7 @@
 from mock import patch, MagicMock
 from urllib3.exceptions import SSLError
 
-from txclib import utils
+from txclib import utils, exceptions
 
 
 class MakeRequestTestCase(unittest.TestCase):
@@ -87,7 +87,7 @@ def test_makes_request_404(self, mock_connection_from_url):
         host = 'http://whynotestsforthisstuff.com'
         url = '/my_test_url/'
         self.assertRaises(
-            utils.HttpNotFound,
+            exceptions.HttpNotFound,
             utils.make_request,
             'GET',
             host,
@@ -118,6 +118,28 @@ def test_makes_request_403(self, mock_connection_from_url):
             'a_pass'
         )
 
+    @patch('urllib3.connection_from_url')
+    def test_makes_request_401(self, mock_connection_from_url):
+        response_mock = MagicMock()
+        response_mock.status = 401
+        response_mock.data = 'test_data'
+
+        mock_connection = MagicMock()
+        mock_connection.request.return_value = response_mock
+        mock_connection_from_url.return_value = mock_connection
+
+        host = 'http://whynotestsforthisstuff.com'
+        url = '/my_test_url/'
+        self.assertRaises(
+            exceptions.HttpNotAuthorized,
+            utils.make_request,
+            'GET',
+            host,
+            url,
+            'a_user',
+            'a_pass'
+        )
+
     @patch('urllib3.connection_from_url')
     def test_makes_request_None(self, mock_connection_from_url):
         response_mock = MagicMock()