def test_send_email_error_call_send_email(self): trace_id = DataTestSes.TRACE_ID region_name = DataTestSes.REGIONS sender = DataTestSes.SENDER subject = DataTestSes.SUBJECT bcc = DataTestSes.BCC body = DataTestSes.BODY error_code = mock_common_utils.get_error_code() error_message = mock_common_utils.get_error_message() client = boto3.client(service_name='ses', region_name=region_name) response = {'ResponseMetadata': {'HTTPStatusCode': 200}} with patch.object(boto3, 'client', return_value=client): with patch.object(client, 'send_email', return_value=response) as mock_send_email: mock_send_email.side_effect = ClientError( {'Error': { 'Code': error_code, 'Message': error_message }}, 'description error') # call function test with self.assertRaises(PmError) as exception: aws_common.send_email(trace_id, region_name, sender, bcc, subject, body) # check result cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], error_code) self.assertEqual(cause_error['Message'], error_message)
def test_update_cognito_user_attributes_error_call_admin_update_user_attributes( self): trace_id = copy.deepcopy(DataTestCognitoIdp.TRACE_ID) user_name = "test_user" os.environ["COGNITO_USER_POOL_ID"] = user_pool_id user_attributes = copy.deepcopy( DataTestCognitoIdp.USER_ATTRIBUTES_UPDATE) error_code = mock_common_utils.get_error_code() error_message = mock_common_utils.get_error_message() # create mock throw error when called function admin_update_user_attributes with patch.object(boto3.client("cognito-idp"), 'admin_update_user_attributes') as mock_method: mock_method.side_effect = ClientError( {'Error': { 'Code': error_code, 'Message': error_message }}, 'description error') # call function test with self.assertRaises(PmError) as exception: aws_common.update_cognito_user_attributes( trace_id, user_name, user_attributes) # check result cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], error_code) self.assertEqual(cause_error['Message'], error_message)
def test_get_cognito_user_info_by_user_name_error_connect_cognito(self): # create boto3 client error mock_common_utils.set_error_response("500", "ERROR") self.create_mock_boto3_client_error() # call function test with self.assertRaises(PmError) as exception: aws_common.get_cognito_user_info_by_user_name(trace_id, user_name) # check error cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], mock_common_utils.get_error_code()) self.assertEqual(cause_error['Message'], mock_common_utils.get_error_message())
def test_get_cognitor_user_pool_error_connect_cognito(self): attributesToGet = "email" # create boto3 client error mock_common_utils.set_error_response("500", "ERROR") self.create_mock_boto3_client_error() # call function test with self.assertRaises(PmError) as exception: aws_common.get_cognito_user_pools(trace_id, filter, attributesToGet) # check error cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], mock_common_utils.get_error_code()) self.assertEqual(cause_error['Message'], mock_common_utils.get_error_message())
def test_process_admin_create_user_pools_error_connect_cognito(self): error_code = mock_common_utils.get_error_code() error_message = mock_common_utils.get_error_message() # create boto3 client error mock_common_utils.set_error_response(error_code, error_message) self.create_mock_boto3_client_error() # call function test with self.assertRaises(PmError) as exception: aws_common.process_admin_create_user_pools(trace_id, user_name, user_attributes, temporary_password, message_action) # check error cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], error_code) self.assertEqual(cause_error['Message'], error_message)
def test_get_cognito_user_info_by_user_name_error_call_admin_get_user_by_user_not_found_exception( self): # create mock throw error when called function admin_get_user with patch.object(boto3.client("cognito-idp"), 'admin_get_user') as mock_method: mock_method.side_effect = ClientError( { 'Error': { 'Code': CommonConst.COGNITO_USER_NOT_FOUND_EXCEPTION, 'Message': mock_common_utils.get_error_message() } }, 'description error') # call function test response = aws_common.get_cognito_user_info_by_user_name( trace_id, user_name) # check result self.assertEqual(response, None)
def test_update_cognito_user_attributes_error_connect_cognito(self): trace_id = copy.deepcopy(DataTestCognitoIdp.TRACE_ID) user_name = "test_user" user_attributes = copy.deepcopy( DataTestCognitoIdp.USER_ATTRIBUTES_UPDATE) # create boto3 client error mock_common_utils.set_error_response("500", "ERROR") self.create_mock_boto3_client_error() # call function test with self.assertRaises(PmError) as exception: aws_common.update_cognito_user_attributes(trace_id, user_name, user_attributes) # check error cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], mock_common_utils.get_error_code()) self.assertEqual(cause_error['Message'], mock_common_utils.get_error_message())
def test_get_cognitor_user_pool_error_call_admin_get_user(self): error_code = mock_common_utils.get_error_code() error_message = mock_common_utils.get_error_message() # create mock throw error when called function admin_get_user with patch.object(boto3.client("cognito-idp"), 'admin_get_user') as mock_method: mock_method.side_effect = ClientError( {'Error': { 'Code': error_code, 'Message': error_message }}, 'description error') with self.assertRaises(PmError) as exception: # call function test aws_common.get_cognito_user_info_by_user_name( trace_id, user_name) # check result cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], error_code) self.assertEqual(cause_error['Message'], error_message)
def test_get_cognitor_user_pool_error_call_list_users(self): attributesToGet = "email" error_code = mock_common_utils.get_error_code() error_message = mock_common_utils.get_error_message() # create mock throw error when called function list_users with patch.object(boto3.client("cognito-idp"), 'list_users') as mock_method: mock_method.side_effect = ClientError( {'Error': { 'Code': error_code, 'Message': error_message }}, 'description error') # call function test with self.assertRaises(PmError) as exception: aws_common.get_cognito_user_pools(trace_id, filter, attributesToGet) # check result cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], error_code) self.assertEqual(cause_error['Message'], error_message)
def test_process_admin_create_user_pools_error_call_admin_create_user( self): error_code = mock_common_utils.get_error_code() error_message = mock_common_utils.get_error_message() # create mock throw error when called function list_users with patch.object(boto3.client("cognito-idp"), 'admin_create_user') as mock_method: mock_method.side_effect = ClientError( {'Error': { 'Code': error_code, 'Message': error_message }}, 'description error') # call function test with self.assertRaises(PmError) as exception: aws_common.process_admin_create_user_pools( trace_id, user_name, user_attributes, temporary_password, message_action) # check result cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], error_code) self.assertEqual(cause_error['Message'], error_message)
def test_send_email_error_client_connect_ses(self): trace_id = DataTestSes.TRACE_ID region_name = DataTestSes.REGIONS sender = DataTestSes.SENDER subject = DataTestSes.SUBJECT bcc = DataTestSes.BCC body = DataTestSes.BODY self.create_mock_boto3_client_error() mock_common_utils.set_error_response("500", "ERROR") with patch.object(PmLogAdapter, 'error', return_value=None) as error_method: # call function test with self.assertRaises(PmError) as exception: aws_common.send_email(trace_id, region_name, sender, bcc, subject, body) # check result cause_error = exception.exception.cause_error.response['Error'] self.assertEqual(cause_error['Code'], mock_common_utils.get_error_code()) self.assertEqual(cause_error['Message'], mock_common_utils.get_error_message()) error_method.assert_any_call("[%s] SESクライアント作成に失敗しました。", region_name)