def test_tag_all(self): req = self.factory.post('/', {}) req.user = self.user factories.FoiRequestFactory(site=self.site) factories.FoiRequestFactory(site=self.site) frs = FoiRequest.objects.all()[:2] result = self.request_admin.tag_all(req, frs) self.assertEqual(result.status_code, 200) self.assertEqual(frs[0].tags.count(), 0) self.assertEqual(frs[1].tags.count(), 0) req = self.factory.post('/', {'tags': 'one, two'}) req.user = self.user req._messages = default_storage(req) frs = FoiRequest.objects.filter(id__in=[frs[0].id, frs[1].id]) result = self.request_admin.tag_all(req, frs) self.assertIsNone(result) frs = list(frs) frs[0] = FoiRequest.objects.get(id=frs[0].id) frs[1] = FoiRequest.objects.get(id=frs[1].id) self.assertEqual(frs[0].tags.count(), 2) self.assertEqual(frs[1].tags.count(), 2) self.assertEqual(set([t.name for t in frs[0].tags.all()]), set(['one', 'two']))
def test_mark_same_as(self): req = self.factory.post('/', {}) req.user = self.user factories.FoiRequestFactory(site=self.site) factories.FoiRequestFactory(site=self.site) frs = FoiRequest.objects.all()[:2] result = self.request_admin.mark_same_as(req, frs) self.assertEqual(result.status_code, 200) same_as = factories.FoiRequestFactory(site=self.site) same_as.save() req = self.factory.post('/', {'req_id': same_as.id}) req.user = self.user req._messages = default_storage(req) frs = FoiRequest.objects.filter(id__in=[frs[0].id, frs[1].id]) result = self.request_admin.mark_same_as(req, frs) self.assertIsNone(result) same_as = FoiRequest.objects.get(id=same_as.id) self.assertEqual(same_as.same_as_count, 2) frs = list(frs) frs[0] = FoiRequest.objects.get(id=frs[0].id) frs[1] = FoiRequest.objects.get(id=frs[1].id) self.assertEqual(frs[0].same_as, same_as) self.assertEqual(frs[1].same_as, same_as)