def test04_not_enough_credits(self):
job_id = 100
with mock.patch(
"JobContactById.lambda_function.Session") as mock_session:
mock_query = mock_session.return_value.query
mock_get = mock_query.return_value.get
mock_app_id = mock_get.return_value.job_applications
mock_get.return_value = mock.Mock(
job_applications=[mock.Mock(applicant_id="*****@*****.**")],
people_contacted=3)
with mock.patch(
"JobContactById.lambda_function.client") as mock_client:
mock_client.get_user = mock.MagicMock(
return_value={
"UserAttributes": [{
"Name": "email",
"Value": "*****@*****.**"
}, {
"Name": "custom:credits",
"Value": 4
}]
})
ret = contact.handler(apigw_contact_event(job_id), "")
data = json.loads(ret["body"])
print(data)
self.assertEqual(ret["statusCode"], 402,
self.msg_status_code.format(402, ret["statusCode"]))
def test_contact_200(apigw_contact_header_event, apigw_deleteAppId_event):
event = {}
request = {
"job_id": "%s" % (ids_created[0]),
"job_application_status": "OFFER_REJECT",
"email": "*****@*****.**",
"resumes": "resumes/path-goes-here",
"cover_letters": "cover_letters/path-goes-here"
}
event["body"] = json.dumps(request)
actual = {}
actual = createApp.handler(event, "")
assert actual["statusCode"] == 201
acc_data = json.loads(actual["body"])
ret = contact.handler(
apigw_contact_header_event(
ids_created[0], response["AuthenticationResult"]["AccessToken"]),
"")
data = json.loads(ret["body"])
resp = client.admin_update_user_attributes(
UserPoolId='us-east-1_OiH5DGpGX',
Username='******',
UserAttributes=[
{
'Name': 'custom:credits',
'Value': '50'
},
],
)
assert ret["statusCode"] == 200
delete_app = deleteApp.handler(
apigw_deleteAppId_event(acc_data["job_application_id"]), "")
assert delete_app["statusCode"] == 200
def test_contact_428(apigw_contact_header_event):
ret = contact.handler(
apigw_contact_header_event(
ids_created[0], response["AuthenticationResult"]["AccessToken"]),
"")
data = json.loads(ret["body"])
assert ret["statusCode"] == 428
def test02_no_auth_header(self):
job_id = 100
with mock.patch(
"JobContactById.lambda_function.Session") as mock_session:
ret = contact.handler(apigw_contact_no_auth_event(job_id), "")
data = json.loads(ret["body"])
self.assertEqual(ret["statusCode"], 401,
self.msg_status_code.format(401, ret["statusCode"]))
def test01_invalid_id(self):
job_id = 1
with mock.patch(
"JobContactById.lambda_function.Session") as mock_session:
mock_query = mock_session.return_value.query
mock_get = mock_query.return_value.get
mock_get.return_value = None
ret = contact.handler(apigw_contact_event(job_id), context)
self.assertEqual(ret["statusCode"], 404,
self.msg_status_code.format(404, ret["statusCode"]))
def test03_apply_before(self):
job_id = 100
with mock.patch(
"JobContactById.lambda_function.Session") as mock_session:
with mock.patch(
"JobContactById.lambda_function.client") as mock_client:
mock_getuser = mock_client.return_value.get_user
mock_getuser.return_value = "123"
ret = contact.handler(apigw_contact_event(job_id), "")
data = json.loads(ret["body"])
print(data)
self.assertEqual(ret["statusCode"], 428,
self.msg_status_code.format(428, ret["statusCode"]))
def test_contact_no_auth_401(apigw_contact_event):
ret = contact.handler(apigw_contact_event(ids_created[0]), "")
data = json.loads(ret["body"])
assert ret["statusCode"] == 401
def test_contact_404(apigw_contact_event):
ret = contact.handler(apigw_contact_event(-1), "")
assert ret["statusCode"] == 404