def setUp(self):
super(BETSSuccessfulTreatmentRepeaterTest, self).setUp()
self.repeater = BETSSuccessfulTreatmentRepeater(
domain=self.domain,
url='super-cool-url',
)
self.repeater.white_listed_case_types = ['episode']
self.repeater.save()
class BETSSuccessfulTreatmentRepeaterTest(ENikshayLocationStructureMixin,
ENikshayRepeaterTestBase):
def setUp(self):
super(BETSSuccessfulTreatmentRepeaterTest, self).setUp()
self.repeater = BETSSuccessfulTreatmentRepeater(
domain=self.domain,
url='super-cool-url',
)
self.repeater.white_listed_case_types = ['episode']
self.repeater.save()
def test_treatment_outcome_trigger(self):
# Create case that doesn't meet trigger
cases = self.create_case_structure()
self.assign_person_to_location(self.pcp.location_id)
update_case(
self.domain,
self.episode_id,
{
'treatment_outcome': 'not_evaluated',
'prescription_total_days': 100,
ENROLLED_IN_PRIVATE: "true",
},
)
case = cases[self.episode_id]
self.assertEqual(0, len(self.repeat_records().all()))
# Meet trigger
update_case(self.domain, case.case_id, {"treatment_outcome": "cured"})
self.assertEqual(1, len(self.repeat_records().all()))
# Make sure same case doesn't trigger event again
payload_generator = BETSSuccessfulTreatmentPayloadGenerator(None)
payload_generator.handle_success(
MockResponse(201, {"success": "hooray"}), case, None)
update_case(self.domain, case.case_id, {"foo": "bar"})
self.assertEqual(1, len(self.repeat_records().all()))
def test_prescription_total_days_trigger(self):
# Create case that doesn't meet trigger
cases = self.create_case_structure()
self.assign_person_to_location(self.pcp.location_id)
update_case(
self.domain,
self.episode_id,
{
'treatment_outcome': 'not_evaluated',
'prescription_total_days': 100,
'treatment_options': 'fdc',
ENROLLED_IN_PRIVATE: "true",
},
)
case = cases[self.episode_id]
self.assertEqual(0, len(self.repeat_records().all()))
# Meet trigger
update_case(self.domain, case.case_id,
{"prescription_total_days": "169"})
self.assertEqual(1, len(self.repeat_records().all()))
# Make sure same case doesn't trigger event again
payload_generator = BETSSuccessfulTreatmentPayloadGenerator(None)
payload_generator.handle_success(
MockResponse(201, {"success": "hooray"}), case, None)
update_case(self.domain, case.case_id, {"foo": "bar"})
self.assertEqual(1, len(self.repeat_records().all()))