def test_name_to_sgid(self):
_groupname_cache.pop(self.sg_name,
None) # make sure test group isn't in the cache
sg_name = name_to_sgid(self.aws, self.sg_name)
self.aws.SecurityGroups.get.called_with(
filters={'group-name': self.sg_name})
self.assertEqual(self.group["GroupName"], sg_name)
def test_lookup_uses_cache(self):
# prime the cache
_groupname_cache[self.sg_name] = self.group["GroupName"]
name_to_sgid(self.aws, self.sg_name)
self.assertFalse(self.aws.get_all_security_groups.called)
def test_lookup_caches_name(self):
_groupname_cache.pop(self.sg_name, None) # make sure test group isn't in the cache
name_to_sgid(self.aws, self.sg_name)
self.assertIn(self.sg_name, _groupname_cache)
def test_multiple_matches(self):
self.aws.ec2.SecurityGroups.get.return_value = [self.group, self.group]
_groupname_cache.pop(self.sg_name, None) # make sure test group isn't in the cache
with self.assertRaises(MultipleNameMatches):
name_to_sgid(self.aws, self.sg_name)
def test_name_to_sgid(self):
_groupname_cache.pop(self.sg_name, None) # make sure test group isn't in the cache
sg_name = name_to_sgid(self.aws, self.sg_name)
self.aws.SecurityGroups.get.called_with(filters={'group-name': self.sg_name})
self.assertEqual(self.group["GroupName"], sg_name)
def test_lookup_uses_cache(self):
# prime the cache
_groupname_cache[self.sg_name] = self.group["GroupName"]
name_to_sgid(self.aws, self.sg_name)
self.assertFalse(self.aws.get_all_security_groups.called)
def test_lookup_caches_name(self):
_groupname_cache.pop(self.sg_name,
None) # make sure test group isn't in the cache
name_to_sgid(self.aws, self.sg_name)
self.assertIn(self.sg_name, _groupname_cache)
def test_multiple_matches(self):
self.aws.ec2.SecurityGroups.get.return_value = [self.group, self.group]
_groupname_cache.pop(self.sg_name,
None) # make sure test group isn't in the cache
with self.assertRaises(MultipleNameMatches):
name_to_sgid(self.aws, self.sg_name)
