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)