python - Django:在测试客户端上设置 cookie？

Question

我的 Django 站点设置了一些中间件，这些中间件在每个请求中检查 cookie，如果未设置，则将用户转发到其他地方。

我现在想在网站上运行一些测试。这是我的代码:

def test_contactform(self):
    response = self.client.get('/contact/')
    self.assertEqual(response.status_code, 200)
    print response
    self.assertTrue('Contact me' in response.content)

不幸的是，这失败了:

Vary: Cookie
Content-Type: text/html; charset=utf-8
Location: http://testserver/ldap/?next=/contact/
Traceback (most recent call last):
  File "tests.py", line 43, in test_contactform
    self.assertEqual(response.status_code, 200)
AssertionError: 302 != 200

我可以 (i) 在 Django 测试客户端上设置 cookie(如果可以的话，如何设置)或(ii)要求 Django 测试客户端遵循重定向并针对最终页面进行测试？

Answer 1

client.get方法需要一个 follow允许它跟随重定向的参数:

response = self.client.get('/contact/', follow=True)