如何在测试Django中禁用CSRF？

Question

我在使用csrf令牌测试视图时遇到问题。

这段代码

class ViewTests(TestCase):
    def test_bets_view(self):
        login_page = self.client.get('/users/login/')
        print(login_page.content)

返回带有CSRF隐藏输入的HTML。

我需要将它与以前的HTML进行比较，

expected_html = render_to_response('login.html',
                                   dictionary={'form': LoginForm()})

没有隐藏的CSRF输入。 因此断言失败。

如何在测试客户端中禁用CSRF渲染？

Answer 1

您永远不要比较完整的HTML内容。 只需检查功能即可。 万一您需要不惜一切代价禁用csrf ，以下逻辑应该可以帮助我猜测。

在您的views.py文件中，添加以下包

from django.views.decorators.csrf import csrf_exempt

然后，在要执行检查的功能定义之前，添加以下代码段：

@csrf_exempt

这将禁用csrf的默认验证。 即使您的传入请求具有隐藏的csrf令牌，您的服务器功能也会完全忽略它。 这应该是禁用csrf的技巧。

Answer 2

首先，您将在测试中找到该页面并从中提取csrf令牌：

page = self.client.get("/users/login")
token = page.context.get("csrf_token")

然后，您使用相同的标记来渲染模板并进行比较：

expected_html = TemplateResponse(
    page.wsgi_request,                                              
    "login.html",                                   
    context={"form": LoginForm(), "csrf_token": token}).render()

assert expected_html.content == page.content

Answer 3

您可以在单元测试中覆盖中间件设置，如下所示：

from django.test import override_settings

testing_middleware = [ ... anything but csrf middleware goes here ]


class TestWhatever(TestCase)
    @override_settings(MIDDLEWARE=testing_middleware)
    def testSomething():
        # no csrf error will occur 
        self.client.post('/', data={ ... })