def test_get_allowed(self): """ SuccessView should be successfull if referrer is recognized """ self.request.META["HTTP_REFERER"] = reverse("tracking:charter_add") response = SuccessView.as_view()(self.request, obj_type="charter", pk=9999) self.assertEqual(response.status_code, 200)
def setUp(self): user, _ = User.objects.get_or_create( username="******", email="*****@*****.**", password="******", ) language = Language.objects.create( id=9999, code="ts", name="Test Language", ) department = Department.objects.create(name="Test Department", ) self.charter = Charter.objects.create( id=9999, language=language, start_date=timezone.now().date(), end_date=timezone.now().date(), lead_dept=department, ) self.event = Event.objects.create( id=9999, charter=self.charter, start_date=timezone.now().date(), end_date=timezone.now().date(), lead_dept=department, ) self.request = RequestFactory().get('/tracking/success/') self.request.user = user self.view = setup_view(SuccessView(), self.request, obj_type="charter", pk=9999)
def test_get_no_allowed(self): """ SuccessView should redirect to home page if referrer is not recognized """ self.request.META["HTTP_REFERER"] = "wrong/url" response = SuccessView.as_view()(self.request, obj_type="charter", pk=9999) self.assertEqual(response.status_code, 302) self.assertEqual(response.url, reverse("tracking:project_list"))