async def test_containermetadatafetcher_retrieve_url_bad_status():
json_body = "not json"
sleep = mock.AsyncMock()
http = fake_aiohttp_session((json_body, 500))
fetcher = utils.AioContainerMetadataFetcher(http, sleep)
with pytest.raises(MetadataRetrievalError):
await fetcher.retrieve_uri('/foo?id=1')
async def test_containermetadatafetcher_retrieve_url():
json_body = json.dumps({
"AccessKeyId": "a",
"SecretAccessKey": "b",
"Token": "c",
"Expiration": "d"
})
sleep = mock.AsyncMock()
http = fake_aiohttp_session((json_body, 200))
fetcher = utils.AioContainerMetadataFetcher(http, sleep)
resp = await fetcher.retrieve_uri('/foo?id=1')
assert resp['AccessKeyId'] == 'a'
assert resp['SecretAccessKey'] == 'b'
assert resp['Token'] == 'c'
assert resp['Expiration'] == 'd'
resp = await fetcher.retrieve_full_uri('http://localhost/foo?id=1',
{'extra': 'header'})
assert resp['AccessKeyId'] == 'a'
assert resp['SecretAccessKey'] == 'b'
assert resp['Token'] == 'c'
assert resp['Expiration'] == 'd'