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 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"))
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 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"))
