diff --git a/tests/test_utils.py b/tests/test_utils.py index cf3cd30b..b8b66530 100644 --- 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()