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)