def test_find_usage_security_groups(self): """test find usage for security groups""" data = { 'DescribeCacheSecurityGroupsResponse': { 'DescribeCacheSecurityGroupsResult': { 'Marker': None, 'CacheSecurityGroups': [ { 'OwnerId': '123456789012', 'CacheSecurityGroupName': 'default', 'Description': 'default', 'EC2SecurityGroups': [] }, { 'OwnerId': '123456789012', 'CacheSecurityGroupName': 'csg1', 'Description': 'foo bar', 'EC2SecurityGroups': [ { 'EC2SecurityGroupName': 'ec2-sg1', 'Status': 'authorized', 'EC2SecurityGroupOwnerId': '123456789012' } ] } ] }, 'ResponseMetadata': { 'RequestId': 'be15fa9c-26ac-11e5-a849-894e77ed58a8' } } } mock_conn = Mock(spec_set=ElastiCacheConnection) mock_conn.describe_cache_security_groups.return_value = data cls = _ElastiCacheService(21, 43) cls.conn = mock_conn cls._find_usage_security_groups() assert mock_conn.mock_calls == [ call.describe_cache_security_groups(), ] usage = cls.limits['Security Groups'].get_current_usage() assert len(usage) == 1 assert usage[0].get_value() == 2
def test_find_usage_security_groups_exception(self): """test find usage for security groups""" def se_exc(): raise BotoServerError(None, None, None) mock_conn = Mock(spec_set=ElastiCacheConnection) mock_conn.describe_cache_security_groups.side_effect = se_exc cls = _ElastiCacheService(21, 43) cls.conn = mock_conn cls._find_usage_security_groups() assert mock_conn.mock_calls == [ call.describe_cache_security_groups(), ] usage = cls.limits['Security Groups'].get_current_usage() assert len(usage) == 1 assert usage[0].get_value() == 0