def go():
conn = Connection(Endpoint('http', 'host', 9999), loop=self.loop)
resp = mock.Mock()
resp.status = 404
r2 = asyncio.Future(loop=self.loop)
r2.set_result('{"a": 1}')
resp.text.return_value = r2
fut = asyncio.Future(loop=self.loop)
fut.set_result(resp)
conn._session.request = mock.Mock(return_value=fut)
with self.assertRaises(NotFoundError) as ctx:
yield from conn.perform_request('GET', '/data', None, None)
self.assertEqual(404, ctx.exception.status_code)
self.assertEqual('{"a": 1}', ctx.exception.error)
self.assertEqual({"a": 1}, ctx.exception.info)
def test_bad_status_409(loop):
conn = Connection(Endpoint('http', 'host', 9999), loop=loop)
resp = mock.Mock()
resp.status = 409
r2 = asyncio.Future(loop=loop)
r2.set_result('{"a": 1}')
resp.text.return_value = r2
fut = asyncio.Future(loop=loop)
fut.set_result(resp)
conn._session.request = mock.Mock(return_value=fut)
with pytest.raises(ConflictError) as ctx:
yield from conn.perform_request('GET', '/data', None, None)
assert 409 == ctx.value.status_code
assert '{"a": 1}' == ctx.value.error
assert {"a": 1} == ctx.value.info
def go():
conn = Connection(Endpoint('http', 'host', 9999), loop=self.loop)
resp = mock.Mock()
resp.status = 404
r2 = asyncio.Future(loop=self.loop)
r2.set_result('{"a": 1}')
resp.text.return_value = r2
fut = asyncio.Future(loop=self.loop)
fut.set_result(resp)
conn._session.request = mock.Mock(return_value=fut)
with self.assertRaises(NotFoundError) as ctx:
yield from conn.perform_request('GET', '/data', None, None)
self.assertEqual(404, ctx.exception.status_code)
self.assertEqual('{"a": 1}', ctx.exception.error)
self.assertEqual({"a": 1}, ctx.exception.info)
