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