def test_find_usage_nat_gateways_exception(self): subnets = result_fixtures.VPC.test_find_usage_nat_gateways_subnets def se_exc(*args, **kwargs): raise ClientError({'Error': {}}, 'opname') mock_conn = Mock() mock_conn.describe_nat_gateways.side_effect = se_exc cls = _VpcService(21, 43, {}, None) cls._current_account_id = '0123456789' cls.conn = mock_conn with patch('%s.logger' % self.pbm, autospec=True) as mock_logger: cls._find_usage_nat_gateways(subnets) assert len(cls.limits['NAT Gateways per AZ'].get_current_usage()) == 0 assert mock_conn.mock_calls == [ call.describe_nat_gateways(), ] assert mock_logger.mock_calls == [ call.error( 'Caught exception when trying to list NAT Gateways; ' 'perhaps NAT service does not exist in this region?', exc_info=1) ]
def test_find_usage_nat_gateways(self): subnets = result_fixtures.VPC.test_find_usage_nat_gateways_subnets response = result_fixtures.VPC.test_find_usage_nat_gateways mock_conn = Mock() mock_conn.describe_nat_gateways.return_value = response with patch('%s.logger' % self.pbm) as mock_logger: cls = _VpcService(21, 43) cls.conn = mock_conn cls._find_usage_nat_gateways(subnets) assert len(cls.limits['NAT Gateways per AZ'].get_current_usage()) == 2 az2 = cls.limits['NAT Gateways per AZ'].get_current_usage()[0] assert az2.get_value() == 3 assert az2.resource_id == 'az2' az3 = cls.limits['NAT Gateways per AZ'].get_current_usage()[1] assert az3.get_value() == 1 assert az3.resource_id == 'az3' assert mock_conn.mock_calls == [ call.describe_nat_gateways(), ] assert mock_logger.mock_calls == [ call.error( 'ERROR: NAT Gateway %s in SubnetId %s, but SubnetId not ' 'found in subnet_to_az; Gateway cannot be counted!', 'nat-124', 'subnet4' ), call.debug( 'Skipping NAT Gateway %s in state: %s', 'nat-125', 'deleted' ), call.debug( 'Skipping NAT Gateway %s in state: %s', 'nat-127', 'failed' ) ]
def test_find_usage_nat_gateways(self): response = result_fixtures.VPC.test_find_usage_nat_gateways mock_conn = Mock() mock_conn.describe_nat_gateways.return_value = response cls = _VpcService(21, 43) cls.conn = mock_conn cls._find_usage_nat_gateways() assert len(cls.limits['NAT gateways'].get_current_usage()) == 1 assert cls.limits['NAT gateways'].get_current_usage()[0].get_value( ) == 1 assert mock_conn.mock_calls == [ call.describe_nat_gateways(), ]
def test_find_usage_nat_gateways(self): subnets = result_fixtures.VPC.test_find_usage_nat_gateways_subnets response = result_fixtures.VPC.test_find_usage_nat_gateways mock_conn = Mock() mock_conn.describe_nat_gateways.return_value = response cls = _VpcService(21, 43) cls.conn = mock_conn cls._find_usage_nat_gateways(subnets) assert len(cls.limits['NAT Gateways per AZ'].get_current_usage()) == 2 az2 = cls.limits['NAT Gateways per AZ'].get_current_usage()[0] assert az2.get_value() == 2 assert az2.resource_id == 'az2' az3 = cls.limits['NAT Gateways per AZ'].get_current_usage()[1] assert az3.get_value() == 1 assert az3.resource_id == 'az3' assert mock_conn.mock_calls == [ call.describe_nat_gateways(), ]
def test_find_usage_nat_gateways_exception(self): subnets = result_fixtures.VPC.test_find_usage_nat_gateways_subnets def se_exc(*args, **kwargs): raise ClientError({'Error': {}}, 'opname') mock_conn = Mock() mock_conn.describe_nat_gateways.side_effect = se_exc cls = _VpcService(21, 43) cls.conn = mock_conn with patch('%s.logger' % self.pbm, autospec=True) as mock_logger: cls._find_usage_nat_gateways(subnets) assert len(cls.limits['NAT Gateways per AZ'].get_current_usage()) == 0 assert mock_conn.mock_calls == [ call.describe_nat_gateways(), ] assert mock_logger.mock_calls == [ call.error('Caught exception when trying to list NAT Gateways; ' 'perhaps NAT service does not exist in this region?', exc_info=1) ]