async def test_add_trace_request_ctx(aiohttp_client, loop):
actual_request_contexts = []
async def on_request_start(
_: ClientSession,
trace_config_ctx: SimpleNamespace,
__: TraceRequestStartParams,
) -> None:
actual_request_contexts.append(trace_config_ctx)
test_app = App()
trace_config = TraceConfig()
trace_config.on_request_start.append(on_request_start) # type: ignore
retry_client = RetryClient()
retry_client._client = await aiohttp_client(
test_app.get_app(),
trace_configs=[trace_config]
)
async with retry_client.get('/sometimes_error', trace_request_ctx={'foo': 'bar'}):
assert test_app.counter == 3
assert actual_request_contexts == [
SimpleNamespace(
trace_request_ctx={
'foo': 'bar',
'current_attempt': i + 1,
},
)
for i in range(3)
]
async def test_not_found_error(aiohttp_client, loop):
test_app = App()
app = test_app.get_app()
client = await aiohttp_client(app)
retry_client = RetryClient()
retry_client._client = client
retry_options = RetryOptions(attempts=5, statuses={404})
async with retry_client.get('/not_found_error', retry_options) as response:
assert response.status == 404
assert test_app.counter == 5
await retry_client.close()
await client.close()
async def test_internal_error(aiohttp_client, loop):
test_app = App()
app = test_app.get_app()
client = await aiohttp_client(app)
retry_client = RetryClient()
retry_client._client = client
retry_options = ExponentialRetry(attempts=5)
async with retry_client.get('/internal_error', retry_options) as response:
assert response.status == 500
assert test_app.counter == 5
await retry_client.close()
await client.close()
async def test_hello_with_context(aiohttp_client, loop):
test_app = App()
app = test_app.get_app()
client = await aiohttp_client(app)
async with RetryClient() as retry_client:
retry_client._client = client
async with retry_client.get('/ping') as response:
text = await response.text()
assert response.status == 200
assert text == 'Ok!'
assert test_app.counter == 1
await client.close()
async def test_hello_awaitable(aiohttp_client, loop):
test_app = App()
app = test_app.get_app()
client = await aiohttp_client(app)
retry_client = RetryClient()
retry_client._client = client
response = await retry_client.get('/ping')
text = await response.text()
assert response.status == 200
assert text == 'Ok!'
assert test_app.counter == 1
await retry_client.close()
await client.close()
async def test_sometimes_error(aiohttp_client, loop):
test_app = App()
app = test_app.get_app()
client = await aiohttp_client(app)
retry_client = RetryClient()
retry_client._client = client
retry_options = RetryOptions(attempts=5)
async with retry_client.get('/sometimes_error', retry_options) as response:
text = await response.text()
assert response.status == 200
assert text == 'Ok!'
assert test_app.counter == 3
await retry_client.close()
await client.close()
async def test_sometimes_error_with_raise_for_status(aiohttp_client, loop):
test_app = App()
app = test_app.get_app()
client = await aiohttp_client(app, raise_for_status=True)
retry_client = RetryClient()
retry_client._client = client
retry_options = RetryOptions(attempts=5, exceptions={ClientResponseError})
async with retry_client.get('/sometimes_error', retry_options) \
as response:
text = await response.text()
assert response.status == 200
assert text == 'Ok!'
assert test_app.counter == 3
await retry_client.close()
await client.close()