def test_spotinst_account_resolution_failure():
with responses.RequestsMock() as rsps:
rsps.add(rsps.GET, '{}/setup/account?awsAccountId=12345'.format(SPOTINST_API_URL), status=200,
json={"response": {
"items": [],
}})
with pytest.raises(MissingSpotinstAccount):
resolve_account_id("fake-token", "12345")
def test_spotinst_account_resolution():
with responses.RequestsMock() as rsps:
rsps.add(rsps.GET, '{}/setup/account?awsAccountId=12345'.format(SPOTINST_API_URL), status=200,
json={"response": {
"items": [
{"accountId": "act-1234abcd", "name": "expected-match"},
{"accountId": "act-xyz", "name": "second-match"}
],
}})
account_id = resolve_account_id("fake-token", "12345")
assert account_id == "act-1234abcd"