def test_detect_account_id_session_mismatch(self):
account_id = "234567890121"
region_name = "us-east-1"
session = boto3.Session()
scan_accessor = AWSAccessor(session=session, account_id=account_id, region_name=region_name)
with self.assertRaises(ValueError):
AccountResourceSpec.scan(scan_accessor=scan_accessor)
def test_scan(self):
account_id = "123456789012"
region_name = "us-east-1"
session = boto3.Session()
scan_accessor = AWSAccessor(session=session, account_id=account_id, region_name=region_name)
resources = AccountResourceSpec.scan(scan_accessor=scan_accessor)
expected_resources = [
{
"type": "aws:account",
"links": [{"pred": "account_id", "obj": "123456789012", "type": "simple"}],
}
]
expected_api_call_stats = {
"count": 1,
"123456789012": {
"count": 1,
"us-east-1": {
"count": 1,
"sts": {"count": 1, "GetCallerIdentity": {"count": 1}},
},
},
}
self.assertListEqual([resource.to_dict() for resource in resources], expected_resources)
self.assertDictEqual(scan_accessor.api_call_stats.to_dict(), expected_api_call_stats)
def test_generate_arn(self):
account_id = "234567890121"
region_name = "us-east-1"
resource_id = "123456789012"
expected_arn = "arn:aws::::account/123456789012"
arn = AccountResourceSpec.generate_arn(account_id=account_id,
region=region_name,
resource_id=resource_id)
self.assertEqual(arn, expected_arn)
def test_scan(self):
account_id = "123456789012"
region_name = "us-east-1"
session = boto3.Session()
scan_accessor = AWSAccessor(session=session,
account_id=account_id,
region_name=region_name)
resources = AccountResourceSpec.scan(scan_accessor=scan_accessor)
expected_resources = [
Resource(
resource_id="arn:aws::::account/123456789012",
type="aws:account",
link_collection=LinkCollection(simple_links=(SimpleLink(
pred="account_id", obj="123456789012"), ), ),
)
]
self.assertEqual(resources, expected_resources)
def test_scan(self):
account_id = "123456789012"
region_name = "us-east-1"
session = boto3.Session()
scan_accessor = AWSAccessor(session=session,
account_id=account_id,
region_name=region_name)
scan_result = AccountResourceSpec.scan(scan_accessor=scan_accessor)
scan_result_dict = scan_result.to_dict()
expected_scan_result_dict = {
"resources": [{
"type":
"aws:account",
"links": [{
"pred": "account_id",
"obj": "123456789012",
"type": "simple"
}],
}],
"stats": {
"count": 1,
"123456789012": {
"count": 1,
"us-east-1": {
"count": 1,
"sts": {
"count": 1,
"GetCallerIdentity": {
"count": 1
}
},
},
},
},
"errors": [],
}
self.assertDictEqual(scan_result_dict, expected_scan_result_dict)