def test_add_to_skunk_single_region_multiple_instances(self, session_mock):
testSkunky = Skunky(session_mock)
testSkunky.add_to_skunk(TEST_IDENTITY_1_A_US_WEST_2)
testSkunky.add_to_skunk(TEST_IDENTITY_2_A_US_WEST_2)
assert len(testSkunky.to_be_skunked['000000000001']['us-west-2']) == 2
def test_add_to_skunk_single_region(self, session_mock):
identity = TEST_IDENTITY_1_A_US_WEST_2
testSkunky = Skunky(session_mock)
testSkunky.add_to_skunk(identity)
assert len(testSkunky.to_be_skunked['000000000001']['us-west-2']) == 1
def test_add_to_skunk_multiple_regions(self, session_mock):
testSkunky = Skunky(session_mock)
testSkunky.add_to_skunk(TEST_IDENTITY_1_A_US_WEST_2)
testSkunky.add_to_skunk(TEST_IDENTITY_1_A_US_EAST_1)
assert len(testSkunky.to_be_skunked['000000000001']['us-west-2']) == 1
assert len(testSkunky.to_be_skunked['000000000001']['us-east-1']) == 1
def test_skunk_1_account_1_region_multiple_instances(
self, session_mock, skunky_put_mock, skunky_tag_mock):
testSkunky = Skunky(session_mock)
testSkunky.add_to_skunk(TEST_IDENTITY_1_A_US_WEST_2)
testSkunky.add_to_skunk(TEST_IDENTITY_2_A_US_WEST_2)
testSkunky.skunk_instances()
assert skunky_put_mock.call_count == 2
assert skunky_tag_mock.call_count == 1
assert len(testSkunky.to_be_skunked['000000000001']['us-west-2']) == 0