Django rest 框架：客户端中的自定义标头与单元测试不同

Question

我有一个 Django Rest API，然后单独有一个使用这个 API 的客户端。

我使用客户端中的请求传递自定义标头：

r = requests.get(url, params=params, headers={'license': '12345'})

然后在 API 中，在我的自定义权限中，我得到如下标题：

app_license = request.META['HTTP_LICENSE']

我知道 django出于安全原因重写了自定义标头，所以它工作正常。

我的问题是当我在 Django rest API 中编写单元测试时：

response = self.client.get(self.url, params=params, headers={'license': '12345'})

然后它提出：

KeyError: 'HTTP_LICENSE'

但是如果我像这样更改代码，测试通过没有问题但消费者不起作用：

request.META['headers']['license']

我可以检查是否有 'headers' 键，但我不想更改代码只是为了通过单元测试，它必须是编写符合现实的单元测试的某种方式，对吗？

我尝试使用：

from django.test import TestCase

和：

from rest_framework.test import APITestCase

两者结果相同。有什么解决办法吗？谢谢！

Answer 1

TL;DR — 如果您想在 META 中显示某些内容，请将其设为kwarg. 该client.get方法与requests.get.

这是因为 kwargs直接self.client.get映射到 HTTP 属性。

快来看看源。你会注意到堆栈是：

1. get
2. generic
3. request
4. _base_environ

在那个过程中，kwargs只进行了_base_environ很少的改动。然后将它们合并到一个字典中，其中包含您期望的所有基本标题：

    # This is a minimal valid WSGI environ dictionary, plus:
    # - HTTP_COOKIE: for cookie support,
    # - REMOTE_ADDR: often useful, see #8551.
    # See http://www.python.org/dev/peps/pep-3333/#environ-variables
    environ = {
        'HTTP_COOKIE': self.cookies.output(header='', sep='; '),
        'PATH_INFO': '/',
        'REMOTE_ADDR': '127.0.0.1',
        'REQUEST_METHOD': 'GET',
        'SCRIPT_NAME': '',
        'SERVER_NAME': 'testserver',
        'SERVER_PORT': '80',
        'SERVER_PROTOCOL': 'HTTP/1.1',
        'wsgi.version': (1, 0),
        'wsgi.url_scheme': 'http',
        'wsgi.input': FakePayload(b''),
        'wsgi.errors': self.errors,
        'wsgi.multiprocess': True,
        'wsgi.multithread': False,
        'wsgi.run_once': False,
    }
    environ.update(self.defaults)
    environ.update(request)
    return environ

然后将上面的内容传递到方法中的WSGIRequest对象中request。然后该类采用提供的environdict 并将其分配给 META 属性：

class WSGIRequest(HttpRequest):
    def __init__(self, environ):
        script_name = get_script_name(environ)
        # If PATH_INFO is empty (e.g. accessing the SCRIPT_NAME URL without a
        # trailing slash), operate as if '/' was requested.
        path_info = get_path_info(environ) or '/'

        ##############################
        ##                          ##
        ## This is the line you're  ##
        ##       looking for.       ##
        ##                          ##
        ##############################
        self.environ = environ
        self.path_info = path_info