def test_add_survey_post_fail(self, mock_request, get_iac_by_enrolment_code):
mock_request.get(url_banner_api, status_code=404)
error_response = Response()
error_response.status_code = 500
error_response.url = url_validate_enrolment
error = ApiError(logger, error_response)
error.status_code = 500
get_iac_by_enrolment_code.side_effect = error
response = self.app.post('/surveys/add-survey', data={'enrolment_code': enrolment_code})
self.assertEqual(response.status_code, 500)
self.assertTrue('An error has occurred'.encode() in response.data)
def test_add_survey_post_fail(self, get_iac_by_enrolment_code):
error_response = Response()
error_response.status_code = 500
error_response.url = url_validate_enrolment
error = ApiError(logger, error_response)
error.status_code = 500
get_iac_by_enrolment_code.side_effect = error
response = self.app.post('/surveys/add-survey',
data={'enrolment_code': enrolment_code})
self.assertEqual(response.status_code, 500)
self.assertTrue('Error 500 - Server error'.encode() in response.data)
def test_add_survey_submit_fail(self, decrypt_enrolment_code, get_iac_from_enrolment):
decrypt_enrolment_code.return_value = enrolment_code.encode()
error_response = Response()
error_response.status_code = 500
error_response.url = url_validate_enrolment
error = ApiError(logger, error_response)
error.status_code = 500
get_iac_from_enrolment.side_effect = error
response = self.app.get(f'/surveys/add-survey/add-survey-submit?encrypted_enrolment_code={encrypted_enrolment_code}')
self.assertEqual(response.status_code, 500)
self.assertTrue('An error has occurred'.encode() in response.data)
def test_survey_confirm_organisation_fail(self, decrypt_enrolment_code, _, get_case):
decrypt_enrolment_code.return_value = enrolment_code.encode()
error_response = Response()
error_response.status_code = 500
error_response.url = url_get_case
error = ApiError(logger, error_response)
error.status_code = 500
get_case.side_effect = error
url = f"/surveys/add-survey/confirm-organisation-survey?encrypted_enrolment_code={encrypted_enrolment_code}"
response = self.app.get(url)
self.assertEqual(response.status_code, 500)
self.assertTrue('Error 500 - Server error'.encode() in response.data)
def test_add_survey_enrolment_code_already_used(self,
get_iac_by_enrolment_code):
error_response = Response()
error_response.status_code = 400
error_response.url = url_validate_enrolment
error = ApiError(logger, error_response)
error.status_code = 400
get_iac_by_enrolment_code.side_effect = error
response = self.app.post('/surveys/add-survey',
data={'enrolment_code': enrolment_code})
self.assertEqual(response.status_code, 200)
self.assertTrue('Enrolment code not valid'.encode() in response.data)
self.assertTrue(
'Please re-enter the code and try again'.encode() in response.data)
def test_add_survey_enrolment_code_already_used(self, mock_request,
get_iac_by_enrolment_code):
mock_request.get(url_banner_api, status_code=404)
error_response = Response()
error_response.status_code = 400
error_response.url = url_validate_enrolment
error = ApiError(logger, error_response)
error.status_code = 400
get_iac_by_enrolment_code.side_effect = error
response = self.app.post("/surveys/add-survey",
data={"enrolment_code": enrolment_code})
self.assertEqual(response.status_code, 200)
self.assertTrue(
"Enter a valid enrolment code".encode() in response.data)
def test_survey_confirm_organisation_fail(self, mock_request,
decrypt_enrolment_code, _,
get_case):
mock_request.get(url_banner_api, status_code=404)
decrypt_enrolment_code.return_value = enrolment_code.encode()
error_response = Response()
error_response.status_code = 500
error_response.url = url_get_case
error = ApiError(logger, error_response)
error.status_code = 500
get_case.side_effect = error
url = f"/surveys/add-survey/confirm-organisation-survey?encrypted_enrolment_code={encrypted_enrolment_code}"
response = self.app.get(url)
self.assertEqual(response.status_code, 500)
self.assertTrue("An error has occurred".encode() in response.data)