def test_exporter_get_compliance_visits(self): self.exporter_user.set_role(self.organisation, self.exporter_super_user_role) comp_visit_1 = ComplianceVisitCaseFactory( organisation=self.organisation, status=get_case_status_by_status(CaseStatusEnum.OPEN) ) comp_site_case = comp_visit_1.site_case comp_visit_2 = ComplianceVisitCaseFactory( organisation=self.organisation, status=get_case_status_by_status(CaseStatusEnum.OPEN), site_case=comp_site_case, ) comp_visit_2.case_officer = self.gov_user comp_visit_2.save() url = reverse("compliance:exporter_visit_case_list", kwargs={"pk": comp_site_case.id}) response = self.client.get(url, **self.exporter_headers) self.assertEqual(response.status_code, status.HTTP_200_OK) response_data = response.json()["results"] self.assertEqual(len(response_data), 2) self.compare_compliance_visit_list_details(response_data[0], comp_visit_1) self.compare_compliance_visit_list_details(response_data[1], comp_visit_2)
def test_exporter_get_visit_details(self): self.exporter_user.set_role(self.organisation, self.exporter_super_user_role) comp_visit_1 = ComplianceVisitCaseFactory( organisation=self.organisation, status=get_case_status_by_status(CaseStatusEnum.OPEN) ) comp_visit_1.case_officer = self.gov_user comp_visit_1.save() url = reverse("compliance:exporter_visit_case_detail", kwargs={"pk": comp_visit_1.id}) response = self.client.get(url, **self.exporter_headers) self.assertEqual(response.status_code, status.HTTP_200_OK) response_data = response.json() self.assertEqual(response_data["id"], str(comp_visit_1.id)) self.assertEqual(response_data["reference_code"], comp_visit_1.reference_code) self.assertEqual(response_data["visit_date"], comp_visit_1.visit_date.strftime("%Y-%m-%d")) self.assertEqual(response_data["case_officer_first_name"], comp_visit_1.case_officer.first_name) self.assertEqual(response_data["case_officer_last_name"], comp_visit_1.case_officer.last_name)