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)))