def test_describe_flow_log_error(self):
# create data mock
ec2_client = ec2_utils_mock.client_connect()
with patch.object(PmLogAdapter, 'error',
return_value=None) as mock_error:
with patch.object(ec2_client, 'describe_flow_logs') as mock_method:
mock_method.side_effect = ClientError(
{
'Error': {
'Code': data_client_error['service_error_code'],
'Message': data_client_error['service_message']
}
}, 'EXCEPTION')
with self.assertRaises(PmError) as exception:
# Call function test
ec2_utils.describe_flow_logs(trace_id, aws_account,
ec2_client, region_name)
cause_error = exception.exception.cause_error.response['Error']
# Check result
self.assertEqual(data_client_error['service_error_code'],
cause_error['Code'])
self.assertEqual(data_client_error['service_message'],
cause_error['Message'])
mock_error.assert_any_call("[%s/%s] VPCフローログ情報の取得に失敗しました。",
aws_account, region_name)
def test_describe_not_exist_flow_logs(self):
# create data mock
ec2_client = ec2_utils_mock.client_connect()
except_result = copy.deepcopy(
DataTestEC2.DATA_DESCRIBE_FLOW_LOGS_EMPTY)
with patch.object(ec2_client, 'describe_flow_logs') as mock_obj:
mock_obj.return_value = except_result
# Call function test
actual_instances = ec2_utils.describe_flow_logs(
trace_id, aws_account, ec2_client, region_name)
# Check result
self.assertListEqual([], actual_instances)
def test_describe_flow_log_exist_next_token(self):
# create data mock
ec2_client = ec2_utils_mock.client_connect()
with patch.object(ec2_client, 'describe_flow_logs') as mock_obj:
mock_obj.side_effect = ec2_utils_mock.side_effect_describe_flow_log
# Call function test
actual_instances = ec2_utils.describe_flow_logs(
trace_id, aws_account, ec2_client, region_name)
# Check result
except_result = copy.deepcopy(
DataTestEC2.DATA_CHECK_ALL_DESCRIBE_FLOW_LOGS['FlowLogs'])
self.assertListEqual(except_result, actual_instances)