def test_add_full_mapping(self): ami_map = AmiRegionMap() stile_map = { 'us-east-1': {'NAT': 'ami-303b1458', 'BASTION': 'ami-60b6c60a'}, 'us-west-1': {'NAT': 'ami-7da94839', 'BASTION': 'ami-d5ea86b5'}, 'us-west-2': {'NAT': 'ami-69ae8259', 'BASTION': 'ami-f0091d91'} } ami_map.add_map(stile_map) self.assertEquals('ami-69ae8259', ami_map['us-west-2']['NAT']) self.assertEquals('ami-d5ea86b5', ami_map['us-west-1']['BASTION'])
def test_add_single_region_mapping(self): ami_map = AmiRegionMap() ami_map.add_ami_map('BIGGLES', {'us-east-1': 'ami-deadbeef'}) self.assertEqual('ami-deadbeef', ami_map['us-east-1']['BIGGLES'])
def test_add_bogus_region(self): ami_map = AmiRegionMap() ami_map.add_ami_map('BIGGLES', {'anthrax': 'ami-deadbeef'}) self.assertNotIn('anthrax', ami_map)
def test_add_to_one_region_not_in_another(self): ami_map = AmiRegionMap() ami_map.add_ami_map('BIGGLES', {'us-east-1': 'ami-deadbeef'}) self.assertNotIn('BIGGLES', ami_map['us-west-1'])