python - Django 休息框架 : custom headers are different in client than unittests

Question

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

我在客户端中使用请求传递自定义 header :

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

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

app_license = request.META['HTTP_LICENSE']

我知道 django rewrites the custom headers出于安全原因，所以它工作正常。

我的问题是当我在 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 不同的签名。

---

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

拍一个look at the source .您会注意到堆栈是:

1. 获取
2. 通用
3. 请求
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

然后将上述内容传递到 request 方法中的 WSGIRequest 对象。 That class then takes the provided environ dict and assigns it to the META property:

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