def test_get_list_policies_error_call_list_policies(self):
# connect client
client_connect_iam = iam_utils.client_connect()
expected_error_response = copy.deepcopy(DataCommon.ERROR_RESPONSE)
expected_operation_name = copy.deepcopy(DataCommon.OPERATION_NAME)
# mock client
with patch.object(session, 'client') as mock_client:
mock_client.return_value = client_connect_iam
# mock error call API list_policies
with patch.object(client_connect_iam,
'list_policies') as mock_method:
mock_method.side_effect = ClientError(
error_response=expected_error_response,
operation_name=expected_operation_name)
with patch.object(
PmLogAdapter, 'error',
return_value=None) as mock_method_error:
with self.assertRaises(PmError) as exception:
IAMUtils.get_list_policies(trace_id, session,
aws_account)
# check error
actual_cause_error = exception.exception.cause_error
self.assertEqual(expected_error_response['Error'],
actual_cause_error.response['Error'])
self.assertEqual(expected_operation_name,
actual_cause_error.operation_name)
# check message log error
mock_method_error.assert_any_call("[%s] ポリシー一覧情報の取得に失敗しました。",
aws_account)
def test_get_list_policies_success_response_not_exists_policies(self):
# connect client
client_connect_iam = iam_utils.client_connect()
expected_list_policies = []
# mock client
with patch.object(session, 'client') as mock_method_client:
mock_method_client.return_value = client_connect_iam
# mock API list_policies
with patch.object(client_connect_iam,
'list_policies') as mock_method:
mock_method.return_value = {'IsTruncated': False}
actual_list_policies = IAMUtils.get_list_policies(
trace_id, session, aws_account)
# check response
self.assertEqual(expected_list_policies, actual_list_policies)
# check connect client
mock_method_client.assert_any_call(service_name="iam")
def test_get_list_policies_success_response_is_truncate_true(self):
# connect client
client_connect_iam = iam_utils.client_connect()
expected_list_policies = copy.deepcopy(
DataTestIAM.LIST_POLICIES_DATA)
# mock client
with patch.object(session, 'client') as mock_method_client:
mock_method_client.return_value = client_connect_iam
# mock response API list_policies
with patch.object(client_connect_iam,
'list_policies') as mock_method:
mock_method.side_effect = iam_utils.side_effect_list_policies
actual_list_policies = IAMUtils.get_list_policies(
trace_id, session, aws_account)
# check response
self.assertEqual(expected_list_policies, actual_list_policies)
# check connect client
mock_method_client.assert_any_call(service_name="iam")