python - Django 测试客户端 : How to remove http headers

Question

我正在尝试为一段代码编写单元测试，以检查是否缺少 header 。如何从 Django 的测试客户端中省略 http_referer header (或与此相关的任何 header )？

def testCase(self):
    response = self.client.get('/some_url/')
    self.assertNotEqual(500, response.status_code)

Answer 1

您可以使用 Django's RequestFactory 而不是使用内置的测试客户端构建您的请求，然后在调用您的函数之前修改请求对象:

from django.test.client import RequestFactory
from django.test import TestCase

from yourapp.view import yourfunction

class YourTest(TestCase)
    def setUp(self):
        self.factory = RequestFactory()

    def testCase(self):
        request = self.factory.get('/yoururl/')
        del request.META['HTTP_REFERER']
        yourfunction(request)