使用 pytest-django 客户端的 Django 响应上下文总是无

Question

我正在使用pytest-django 来测试一些 Django 视图。

我想测试响应上下文是否包含某些值，但它始终是None。

我的看法：

from django.views.generic import TemplateView

class MyView(TemplateView):
    template_name = 'my_template.html'

    def get_context_data(self, **kwargs):
        context = super(MyView, self).get_context_data(**kwargs)
        context['hello'] = 'hi'
        return context

我的测试：

def test_context(client):
    response = client.get('/test/')
    print('status', response.status_code)
    print('content', response.content)
    print('context', response.context)

如果我使用-s 标志运行此程序以查看打印语句，则状态代码为200，content 包含呈现的模板，包括上下文中的"hi"。但是context 是None。

我认为这个 client 与 django.test.Client 相同，这应该让我看到上下文......那我错过了什么？

我试过this answer，但得到了

RuntimeError: setup_test_environment() 已被调用，如果不先调用 teardown_test_environment() 就无法再次调用。

Answer 1

在您提供的client link 中，声明client 是django.test.Client 的一个实例，因此实际上它在那里并没有做任何特别的事情，也不应该成为问题。

您需要按照正确的说明设置环境。
现在让我们看一下错误：

来自setup_test_environment()源代码：

if hasattr(_TestState, 'saved_data'):
      # Executing this function twice would overwrite the saved values.
      raise RuntimeError(
          "setup_test_environment() was already called and can't be called "
          "again without first calling teardown_test_environment()."
)

这就是上面提到的RuntimeError。

现在让我们看看teardown_test_environment() 方法：

...
del _TestState.saved_data

所以它删除了上述异常的罪魁祸首。

因此：

from django.test.utils import teardown_test_environment, setup_test_environment

try:
    # If setup_test_environment haven't been called previously this
    # will produce an AttributeError.
    teardown_test_environment()
except AttributeError:
    pass

setup_test_environment() 

...