def test_authorized_request_raises(self, is_logged_in): is_logged_in.return_value = False mock_response = MagicMock(headers={'content-type': 'text/html'}) self.session.request = Mock(return_value=mock_response) wc = BaseWebClient(self.session) with self.assertRaises(NotLoggedIn): wc.authorized_request(Mock(), 'http://example.com/', 'GET')
def test_authorized_request_returns_response(self): mock_response = MagicMock() self.session.request = Mock(return_value=mock_response) wc = BaseWebClient(self.session) resp = wc.authorized_request(Mock(), 'http://example.com/', 'GET') self.assertIs(mock_response, resp)