def test_get_facilities_pending_approval(self): self.maxDiff = None facility = mommy.make(Facility) facility_2 = mommy.make(Facility) mommy.make(FacilityApproval, facility=facility_2) facility_3 = mommy.make(Facility) mommy.make(FacilityApproval, facility=facility_3) facility_3.name = 'new name' facility_3.save() url = self.url + "?pending_approval=true" response = self.client.get(url) self.assertEquals(200, response.status_code) self.assertEquals(2, response.data.get("count")) expected_results = [ FacilitySerializer(facility).data, FacilitySerializer(facility_3).data ] self.assertListEqual( sorted(load_dump(expected_results, default=default)), sorted(load_dump(response.data.get("results"), default=default))) url = self.url + "?pending_approval=false" response = self.client.get(url) self.assertEquals(200, response.status_code) expected_results = [FacilitySerializer(facility_2).data] self.assertListEqual(expected_results, response.data.get("results"))
def test_get_facilities_pending_approval(self): self.maxDiff = None facility = mommy.make(Facility) facility_2 = mommy.make(Facility) mommy.make(FacilityApproval, facility=facility_2) facility_3 = mommy.make(Facility) mommy.make(FacilityApproval, facility=facility_3) facility_3.name = 'new name' facility_3.save() url = self.url + "?pending_approval=true" response = self.client.get(url) self.assertEquals(200, response.status_code) self.assertEquals(2, response.data.get("count")) expected_results = [ FacilitySerializer(facility).data, FacilitySerializer(facility_3).data] self.assertListEqual( sorted(load_dump(expected_results, default=default)), sorted(load_dump(response.data.get("results"), default=default))) url = self.url + "?pending_approval=false" response = self.client.get(url) self.assertEquals(200, response.status_code) expected_results = [ FacilitySerializer(facility_2).data ] self.assertListEqual(expected_results, response.data.get("results"))
def test_admin_user_sees_all_fields_on_detail(self): perm = Permission.objects.get(codename="view_all_facility_fields") self.admin_group.permissions.add(perm.id) facility = mommy.make(Facility) mommy.make(FacilityApproval, facility=facility) self.client.logout self.client.force_authenticate(self.admin_user) url = self.url + "{}/".format(facility.id) response = self.client.get(url) data = load_dump(response.data, default=default) self.confirm_data_detail_endpoint_contains_keys(data)
def test_admin_user_sees_all_fields_list_endpoint(self): perm = Permission.objects.get(codename="view_all_facility_fields") self.admin_group.permissions.add(perm.id) facility = mommy.make(Facility) mommy.make(FacilityApproval, facility=facility) facility.is_published = True facility.save() self.client.force_authenticate(self.admin_user) response = self.client.get(self.url) all_data = load_dump(response.data['results'], default=default) data = all_data[0] self.assertIn('has_edits', data) self.assertIn('is_approved', data) self.assertIn('latest_update', data) self.assertIn('deleted', data) self.assertIn('active', data) self.assertIn('is_classified', data) self.assertIn('is_published', data) self.assertIn('created_by', data) self.assertIn('updated_by', data)