def test_handle_success(self):
payload_generator = RegisterPatientPayloadGenerator(None)
payload_generator.handle_success(MockResponse(201, {"success": "hooray"}), self.cases[self.episode_id])
updated_episode_case = CaseAccessors(self.domain).get_case(self.episode_id)
self.assertEqual(
updated_episode_case.dynamic_case_properties().get('dots_99_registered'),
'true'
)
self.assertEqual(
updated_episode_case.dynamic_case_properties().get('dots_99_error'),
''
)
def test_get_payload(self):
payload_generator = RegisterPatientPayloadGenerator(None)
person = self.cases[self.person_id].dynamic_case_properties()
expected_numbers = u"+91{}, +91{}".format(
person['phone_number'].replace("0", ""),
person['backup_number'].replace("0", "")
)
expected_payload = json.dumps({
'beneficiary_id': self.person_id,
'phone_numbers': expected_numbers,
'merm_id': person['merm_id']
})
actual_payload = payload_generator.get_payload(None, self.cases[self.episode_id])
self.assertEqual(expected_payload, actual_payload)
def test_handle_success(self):
cases = self.create_case_structure()
cases[self.person_id] = self.assign_person_to_location(self.phi.location_id)
payload_generator = RegisterPatientPayloadGenerator(None)
payload_generator.handle_success(MockResponse(201, {"success": "hooray"}), cases[self.episode_id], None)
updated_episode_case = CaseAccessors(self.domain).get_case(self.episode_id)
self.assertEqual(
updated_episode_case.dynamic_case_properties().get('dots_99_registered'),
'true'
)
self.assertEqual(
updated_episode_case.dynamic_case_properties().get('dots_99_error'),
''
)
def test_handle_failure(self):
payload_generator = RegisterPatientPayloadGenerator(None)
error = {
"error": "Something went terribly wrong",
}
payload_generator.handle_failure(MockResponse(400, error), self.cases[self.episode_id])
updated_episode_case = CaseAccessors(self.domain).get_case(self.episode_id)
self.assertEqual(
updated_episode_case.dynamic_case_properties().get('dots_99_registered'),
'false'
)
self.assertEqual(
updated_episode_case.dynamic_case_properties().get('dots_99_error'),
"400: {}".format(error['error'])
)
def _assert_payload_equal(self, casedb, expected_numbers):
expected_payload = json.dumps({
'beneficiary_id': self.person_id,
'phone_numbers': expected_numbers,
'merm_id': casedb[self.person_id].dynamic_case_properties().get('merm_id')
})
actual_payload = RegisterPatientPayloadGenerator(None).get_payload(None, casedb[self.episode_id])
self.assertEqual(expected_payload, actual_payload)
def _get_actual_payload(self, casedb):
return RegisterPatientPayloadGenerator(None).get_payload(
None, casedb[self.episode_id])
