Question

我正在尝试在Django中执行单元测试。我在index.html中有以下表格：

<form method=POST>
  {% csrf_token %}
  <input name=itemT>
</form>

我正在测试视图是否正确呈现模板：

views.py

def homePage(request):
    return render(request, 'index.html')

tests.py：

request = HttpRequest()

response = homePage(request)
if response:
    response = response.content.decode('UTF-8')

expectedHTML = render_to_string('index.html')
self.assertEqual(response, expectedHTML)

response有一个带有csrf标记的隐藏输入字段;但是，expectedHTML没有（{% csrf_token %}处只有一个空行）。所以断言总是失败。

是否可以让render_to_string()生成csrf输入字段？如果是，response的令牌是否与expectedHTML的令牌相同？

或者，有没有办法忽略输入字段，以便测试成功？

Answer 1

要在使用render_to_string时使csrf令牌正常工作，您需要提供request对象以便上下文处理器运行。

在Django 1.8+中，您可以简单地将请求作为参数传递

return render_to_string('index.html', request=request)

在早期版本中，您可以使用RequestContext。

from django.template import RequestContext
render_to_string('index.html', context_instance=RequestContext(request))

Answer 2

不幸的是，Alasdair的答案不适用于Django 1.10，因为csrf_token会对每个请求进行更改。请参阅适用于1.10的this gist。（改变代码以修复原始要点中的拼写错误）

class HomePageTest(TestCase):

    @staticmethod
    def remove_csrf(html_code):
        csrf_regex = r'<input[^>]+csrfmiddlewaretoken[^>]+>'
        return re.sub(csrf_regex, '', html_code)

    def assertEqualExceptCSRF(self, html_code1, html_code2):
        return self.assertEqual(
            self.remove_csrf(html_code1),
            self.remove_csrf(html_code2)
        )

Answer 3

你可以简单地添加这样的参数。

render_to_string('index.html', request=request)

请参阅document。

Answer 4

从 Django 2 开始，您可以这样做：

html = render_to_string('my_template.html', context, request=request)

Django render_to_string（）忽略{％csrf_token％}

4 个答案: