async def test_DynamicConfigAuthorizationMappingGenerator(self):
from consoleme.lib.cloud_credential_authorization_mapping import (
DynamicConfigAuthorizationMappingGenerator,
RoleAuthorizations,
)
authorization_mapping = await DynamicConfigAuthorizationMappingGenerator(
).generate_credential_authorization_mapping({})
expected = {
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/userrolename"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={"arn:aws:iam::123456789012:role/rolename"},
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/rolename2"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/roleB",
"arn:aws:iam::123456789012:role/roleA",
},
authorized_roles_cli_only=set(),
),
}
for k, v in expected.items():
self.assertEqual(authorization_mapping.get(k), v)
async def test_InternalPluginAuthorizationMappingGenerator(self):
from consoleme.lib.cloud_credential_authorization_mapping import (
InternalPluginAuthorizationMappingGenerator,
RoleAuthorizations,
)
expected = {
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/roleA",
"arn:aws:iam::123456789012:role/roleB",
},
authorized_roles_cli_only=set(),
),
}
authorization_mapping = await InternalPluginAuthorizationMappingGenerator(
).generate_credential_authorization_mapping(expected)
for k, v in expected.items():
self.assertEqual(authorization_mapping.get(k), v)
async def test_RoleTagAuthorizationMappingGenerator(self):
from consoleme.lib.cloud_credential_authorization_mapping import (
RoleAuthorizations,
RoleTagAuthorizationMappingGenerator,
)
authorization_mapping = await RoleTagAuthorizationMappingGenerator(
).generate_credential_authorization_mapping({})
expected = {
"group8":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
authorized_roles_cli_only=set(),
),
"group8-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
),
"group1":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
authorized_roles_cli_only=set(),
),
"group1-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
),
"group0":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
authorized_roles_cli_only=set(),
),
"group0-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
),
"group5":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
authorized_roles_cli_only=set(),
),
"group5-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
),
"group6":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
authorized_roles_cli_only=set(),
),
"group6-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
),
"group7":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
authorized_roles_cli_only=set(),
),
"group7-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
),
"group4":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
authorized_roles_cli_only=set(),
),
"group4-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
),
"group2":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
authorized_roles_cli_only=set(),
),
"group2-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
),
"group3":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
authorized_roles_cli_only=set(),
),
"group3-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
),
"group9":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
authorized_roles_cli_only=set(),
),
"group9-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
),
}
for k, v in expected.items():
self.assertEqual(authorization_mapping.get(k), v)
async def test_generate_and_store_credential_authorization_mapping(self):
from consoleme.lib.cloud_credential_authorization_mapping import (
RoleAuthorizations,
generate_and_store_credential_authorization_mapping,
)
mapping = await generate_and_store_credential_authorization_mapping()
expected = {
"group8":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
authorized_roles_cli_only=set(),
),
"group8-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
),
"group3":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
authorized_roles_cli_only=set(),
),
"group3-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
),
"group6":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
authorized_roles_cli_only=set(),
),
"group6-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
),
"group9":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
authorized_roles_cli_only=set(),
),
"group9-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
),
"group5":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
authorized_roles_cli_only=set(),
),
"group5-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
),
"group2":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
authorized_roles_cli_only=set(),
),
"group2-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
),
"group1":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber1",
"arn:aws:iam::123456789012:role/rolename",
},
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/rolename2"
},
),
"group1-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
),
"group7":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
authorized_roles_cli_only=set(),
),
"group7-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
),
"group4":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
authorized_roles_cli_only=set(),
),
"group4-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
),
"group0":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
authorized_roles_cli_only=set(),
),
"group0-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/userrolename"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/roleA",
"arn:aws:iam::123456789012:role/roleB",
},
authorized_roles_cli_only=set(),
),
}
for k, v in expected.items():
self.assertEqual(mapping.get(k), v)
async def test_generate_and_store_reverse_authorization_mapping(self):
from consoleme.lib.cloud_credential_authorization_mapping import (
RoleAuthorizations,
generate_and_store_reverse_authorization_mapping,
)
authorization_mapping = {
"group8":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
authorized_roles_cli_only=set(),
),
"group8-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber8"
},
),
"group3":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
authorized_roles_cli_only=set(),
),
"group3-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber3"
},
),
"group6":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
authorized_roles_cli_only=set(),
),
"group6-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber6"
},
),
"group9":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
authorized_roles_cli_only=set(),
),
"group9-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber9"
},
),
"group5":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
authorized_roles_cli_only=set(),
),
"group5-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber5"
},
),
"group2":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
authorized_roles_cli_only=set(),
),
"group2-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber2"
},
),
"group1":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber1",
"arn:aws:iam::123456789012:role/rolename",
},
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/rolename2"
},
),
"group1-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber1"
},
),
"group7":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
authorized_roles_cli_only=set(),
),
"group7-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber7"
},
),
"group4":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
authorized_roles_cli_only=set(),
),
"group4-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber4"
},
),
"group0":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
authorized_roles_cli_only=set(),
),
"group0-cli":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles=set(),
authorized_roles_cli_only={
"arn:aws:iam::123456789012:role/RoleNumber0"
},
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/userrolename"
},
authorized_roles_cli_only=set(),
),
"*****@*****.**":
RoleAuthorizations(
authorized_roles={
"arn:aws:iam::123456789012:role/roleA",
"arn:aws:iam::123456789012:role/roleB",
},
authorized_roles_cli_only=set(),
),
}
reverse_mapping = await generate_and_store_reverse_authorization_mapping(
authorization_mapping)
expected = {
"arn:aws:iam::123456789012:role/rolenumber0": [
"group0",
"*****@*****.**",
"group0-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber1": [
"group1",
"*****@*****.**",
"group1-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber2": [
"group2",
"*****@*****.**",
"group2-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber3": [
"group3",
"*****@*****.**",
"group3-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber4": [
"group4",
"*****@*****.**",
"group4-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber5": [
"group5",
"*****@*****.**",
"group5-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber6": [
"group6",
"*****@*****.**",
"group6-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber7": [
"group7",
"*****@*****.**",
"group7-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber8": [
"group8",
"*****@*****.**",
"group8-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolenumber9": [
"group9",
"*****@*****.**",
"group9-cli",
"*****@*****.**",
],
"arn:aws:iam::123456789012:role/rolea": ["*****@*****.**"],
"arn:aws:iam::123456789012:role/roleb": ["*****@*****.**"],
"arn:aws:iam::123456789012:role/rolename": ["*****@*****.**"],
"arn:aws:iam::123456789012:role/rolename2": ["*****@*****.**"],
"arn:aws:iam::123456789012:role/userrolename":
["*****@*****.**"],
}
self.assertDictEqual(reverse_mapping, expected)