def test_bets_180_treatment_payload(self):
self.episode.attrs['update'][TREATMENT_OUTCOME] = "cured"
self.episode.attrs['update'][TREATMENT_OUTCOME_DATE] = "2017-08-15"
self.episode.attrs['update'][
LAST_VOUCHER_CREATED_BY_ID] = self.user.user_id
cases = self.create_case_structure()
self.assign_person_to_location(self.pcp.location_id)
episode = cases[self.episode_id]
expected_payload = {
"incentive_details": [{
"EventID":
six.text_type(TREATMENT_180_EVENT),
"EventOccurDate":
"2017-08-15",
"BeneficiaryUUID":
self.user.user_id,
"BeneficiaryType":
"mbbs",
"Location":
self.pcp.location_id,
"DTOLocation":
self.dto.location_id,
"EpisodeID":
self.episode_id,
"PersonId":
self.person.attrs['update']['person_id'],
"AgencyId":
self.username.split('@')[0],
"EnikshayApprover":
None,
"EnikshayRole":
None,
"EnikshayApprovalDate":
None,
}]
}
self.assertDictEqual(
expected_payload,
json.loads(
BETS180TreatmentPayloadGenerator(None).get_payload(
None, episode)))
def test_bets_180_treatment_payload(self):
self.episode.attrs['update'][TREATMENT_OUTCOME_DATE] = "2017-08-15"
self.episode.attrs['update'][
LAST_VOUCHER_CREATED_BY_ID] = self.user.user_id
cases = self.create_case_structure()
self.assign_person_to_location(self.pcp.location_id)
episode = cases[self.episode_id]
expected_payload = {
"incentive_details": [{
u"EventID": unicode(TREATMENT_180_EVENT),
u"EventOccurDate": u"2017-08-15",
u"BeneficiaryUUID": self.user.user_id,
u"BeneficiaryType": u"mbbs",
u"Location": self.pcp.location_id,
u"DTOLocation": self.dto.location_id,
u"EpisodeID": self.episode_id,
}]
}
self.assertDictEqual(
expected_payload,
json.loads(
BETS180TreatmentPayloadGenerator(None).get_payload(
None, episode)))