diff --git a/project/tests/test_delete_request.py b/project/tests/test_delete_request.py index 2b98448c..1ce28dd3 100644 --- a/project/tests/test_delete_request.py +++ b/project/tests/test_delete_request.py @@ -1,7 +1,10 @@ +from http import HTTPStatus + from django.test import Client, TestCase -from silk.models import Request from django.urls import reverse -from http import HTTPStatus + +from silk.models import Request + class TestDeleteRequest(TestCase): def setUp(self): @@ -11,7 +14,7 @@ def setUp(self): self.request.method = 'get' self.request.body = b'a' * 1000 self.request.save() - + def test_delete(self): - response = self.client.post(reverse('silk:request_detail', kwargs={'request_id':str(self.request.id)})) + response = self.client.post(reverse('silk:request_detail', kwargs={'request_id': str(self.request.id)})) self.assertEqual(response.status_code, HTTPStatus.FOUND) diff --git a/silk/views/request_detail.py b/silk/views/request_detail.py index 9e170c67..38bf009d 100644 --- a/silk/views/request_detail.py +++ b/silk/views/request_detail.py @@ -1,6 +1,6 @@ import json -from django.shortcuts import render, redirect, HttpResponseRedirect +from django.shortcuts import HttpResponseRedirect, redirect, render from django.utils.decorators import method_decorator from django.views.generic import View @@ -40,7 +40,7 @@ def get(self, request, request_id): 'request': request } return render(request, 'silk/request.html', context) - + @method_decorator(login_possibly_required) @method_decorator(permissions_possibly_required) def post(self, request, request_id): @@ -48,5 +48,5 @@ def post(self, request, request_id): try: silk_request.delete() except Request.DoesNotExist: - return HttpResponseRedirect(request.META.get('HTTP_REFERER', '/silk')) + return HttpResponseRedirect(request.headers.get('Referer', '/silk')) return redirect('/silk')