class HamQTHClientRequestNotOkTestCase(TestCase):
@classmethod
def setUpClass(cls):
class RaisedError(Exception):
pass
class NotOkResponse:
ok = False
def raise_for_status(self):
raise RaisedError("raise_for_status was called")
cls._raised_error = RaisedError
cls._response = NotOkResponse
def setUp(self):
self.client = HamQTHClient()
def test_request_not_ok_raises(self):
url = 'https://some-url.test/'
payload = {'foo': 'bar'}
with patch.object(requests, 'get',
return_value=self._response()) as mock_method:
with self.assertRaises(self._raised_error):
self.client.request(url, payload=payload)
mock_method.assert_called_once_with('https://some-url.test/',
params='foo=bar')
class HamQTHClientRequestTestCase(TestCase):
@classmethod
def setUpClass(cls):
class OkResponse:
ok = True
cls._response = OkResponse
def setUp(self):
self.client = HamQTHClient()
def test_request(self):
url = 'https://some-url.test/'
payload = {'foo': 'bar'}
with patch.object(requests, 'get',
return_value=self._response()) as mock_method:
self.client.request(url, payload=payload)
mock_method.assert_called_once_with('https://some-url.test/',
params='foo=bar')
