def test_raise_from_erroneous_response_without_error(self):
err_res = mock.NonCallableMock(requests.Response)
err_res.status_code = 401
err_res.json = mock.MagicMock(return_value={"goldfish": "are pretty."})
err_res.text = json.dumps(err_res.json())
with self.assertRaises(YggdrasilError) as cm:
_raise_from_response(err_res)
message_start = "[401] Malformed error message"
self.assertTrue(str(cm.exception).startswith(message_start))
def test_raise_invalid_json(self):
err_res = mock.NonCallableMock(requests.Response)
err_res.status_code = 401
err_res.json = mock.MagicMock(
side_effect=ValueError("no json could be decoded"))
err_res.text = "{sample invalid json}"
with self.assertRaises(YggdrasilError) as cm:
_raise_from_response(err_res)
message_start = "[401] Malformed error message"
self.assertTrue(str(cm.exception).startswith(message_start))
def test_raise_from_erroneous_request(self):
err_res = mock.NonCallableMock(requests.Response)
err_res.status_code = 401
err_res.json = mock.MagicMock(
return_value={"error": "ThisIsAnException",
"errorMessage": "Went wrong."})
err_res.text = json.dumps(err_res.json())
with self.assertRaises(YggdrasilError) as cm:
_raise_from_response(err_res)
message = "[401] ThisIsAnException: 'Went wrong.'"
self.assertEqual(str(cm.exception), message)
def test_raise_from_healthy_response(self):
res = mock.NonCallableMock(requests.Response)
res.status_code = 200
res.json = mock.MagicMock(return_value={"vegetables": "are healthy."})
res.text = json.dumps(res.json())
self.assertIs(_raise_from_response(res), None)
