def test_update_user_attribute_success_case_exists_record_contain_callerservicename_and_data_body_contain_callerservicename(
self):
# perpare data test
data_update = DataPmUserAttribute.DATA_UPDATE_SIMPLE
data_body = {
"userName": data_update['UserName'],
"companyName": data_update['CompanyName'],
"departmentName": data_update['DepartmentName'],
"countryCode": data_update['CountryCode'],
"companyFlg": data_update['CompanyFlg'],
"callerServiceName": "opswitch"
}
mock_pm_userAttribute.create(data_insert)
event_mock = event_create.get_event_object(
trace_id=data_insert['UserID'], body=json.dumps(data_body))
# Call function test
response = user.update_user_attributes_handler(event_mock, {})
# Check data
status_code = response['statusCode']
response_body = json.loads(response['body'])
self.assertEqual(status_code, HTTPStatus.OK.value)
self.assertEqual(response_body['id'], data_insert['UserID'])
self.assertEqual(response_body['userName'], data_body['userName'])
self.assertEqual(response_body['companyName'],
data_body['companyName'])
self.assertEqual(response_body['depertmentName'],
data_body['departmentName'])
self.assertEqual(response_body['countryCode'],
data_body['countryCode'])
self.assertEqual(response_body['companyFlg'], data_body['companyFlg'])
self.assertEqual(response_body['callerServiceName'],
data_insert['CallerServiceName'])
def test_get_user_success_case_not_caller_service_name(self):
# perpare data test
data_insert = DataPmUserAttribute.DATA_NOT_CALLERSERVICENAME
mock_pm_userAttribute.create(data_insert)
event_mock = event_create.get_event_object(
trace_id=data_insert['UserID'])
# Call function test
result = user.get_user_attributes_handler(event_mock, {})
# Check data
status_code = result['statusCode']
response_body = json.loads(result['body'])
self.assertEqual(status_code, HTTPStatus.OK.value)
self.assertEqual(response_body['id'], data_insert['UserID'])
self.assertEqual(response_body['userName'], data_insert['UserName'])
self.assertEqual(response_body['mailStatus'],
data_insert['MailStatus'])
self.assertEqual(response_body['companyName'],
data_insert['CompanyName'])
self.assertEqual(response_body['depertmentName'],
data_insert['DepartmentName'])
self.assertEqual(response_body['companyFlg'],
data_insert['CompanyFlg'])
self.assertEqual(response_body['countryCode'],
data_insert['CountryCode'])
self.assertFalse("callerServiceName" in response_body.keys())
def test_get_user_attributes_success(self):
# perpare data test
mock_pm_userAttribute.create(data_insert)
# Call function test
result = user.get_user_attributes_handler(event_mock, {})
# Check data
status_code = result['statusCode']
response_body = json.loads(result['body'])
self.assertEqual(status_code, HTTPStatus.OK.value)
self.assertEqual(response_body['id'], data_insert['UserID'])
self.assertEqual(response_body['userName'], data_insert['UserName'])
self.assertEqual(response_body['mailStatus'],
data_insert['MailStatus'])
self.assertEqual(response_body['companyName'],
data_insert['CompanyName'])
self.assertEqual(response_body['depertmentName'],
data_insert['DepartmentName'])
self.assertEqual(response_body['companyFlg'],
data_insert['CompanyFlg'])
self.assertEqual(response_body['countryCode'],
data_insert['CountryCode'])
self.assertEqual(response_body['callerServiceName'],
data_insert['CallerServiceName'])
def test_update(self):
# prepare data
data_insert = DataPmUserAttribute.DATA_SIMPLE
mock_pm_userAttribute.create(data_insert)
data_update_simple = DataPmUserAttribute.DATA_UPDATE_SIMPLE
data_update = {
"UserName": {
"Value": data_update_simple['UserName']
},
"CompanyName": {
"Value": data_update_simple['CompanyName']
},
"DepartmentName": {
"Value": data_update_simple['DepartmentName']
},
"CompanyFlg": {
"Value": data_update_simple['CompanyFlg']
},
"CountryCode": {
"Value": data_update_simple['CountryCode']
},
"MailStatus": {
"Value": data_update_simple['MailStatus']
}
}
# call function test
pm_userAttribute.update(data_insert['UserID'], data_update)
# check data
result = mock_pm_userAttribute.query_key(data_insert['UserID'])
self.assertEqual(result['UserName'], data_update_simple['UserName'])
self.assertEqual(result['CompanyName'],
data_update_simple['CompanyName'])
self.assertEqual(result['DepartmentName'],
data_update_simple['DepartmentName'])
self.assertEqual(result['CompanyFlg'],
data_update_simple['CompanyFlg'])
self.assertEqual(result['CountryCode'],
data_update_simple['CountryCode'])
def test_query_key(self):
# prepare data
data_insert = DataPmUserAttribute.DATA_SIMPLE
mock_pm_userAttribute.create(data_insert)
# call function test
result = pm_userAttribute.query_key(data_insert['UserID'])
# check data
self.assertEqual(result['UserID'], data_insert['UserID'])
self.assertEqual(result['UserName'], data_insert['UserName'])
self.assertEqual(result['MailStatus'], data_insert['MailStatus'])
self.assertEqual(result['CompanyName'], data_insert['CompanyName'])
self.assertEqual(result['DepartmentName'],
data_insert['DepartmentName'])
self.assertEqual(result['CompanyFlg'], data_insert['CompanyFlg'])
self.assertEqual(result['CountryCode'], data_insert['CountryCode'])
self.assertEqual(result['CallerServiceName'],
data_insert['CallerServiceName'])
self.assertEqual(result['CreatedAt'], data_insert['CreatedAt'])
self.assertEqual(result['UpdatedAt'], data_insert['UpdatedAt'])