Django:测试页面是否已重定向到所需的URL

Question

在我的django应用程序中,我有一个身份验证系统.因此,如果我没有登录并尝试访问某个配置文件的个人信息,我会被重定向到登录页面.

现在,我需要为此编写一个测试用例.我得到的浏览器的回复是:

GET /myprofile/data/some_id/ HTTP/1.1 302 0
GET /account/login?next=/myprofile/data/some_id/ HTTP/1.1 301 0
GET /account/login?next=/myprofile/data/some_id/ HTTP/1.1 200 6533

我该如何写测试？这就是我到目前为止:

self.client.login(user="user", password="passwd")
response = self.client.get('/myprofile/data/some_id/')
self.assertEqual(response.status,200)
self.client.logout()
response = self.client.get('/myprofile/data/some_id/')

接下来会发生什么？

Answer 1

Django 1.4:

https://docs.djangoproject.com/en/1.4/topics/testing/#django.test.TestCase.assertRedirects

Django 2.0:

https://docs.djangoproject.com/en/2.0/topics/testing/tools/#django.test.SimpleTestCase.assertRedirects

SimpleTestCase.assertRedirects(response, expected_url, status_code=302, target_status_code=200, msg_prefix='', fetch_redirect_response=True)

断言响应返回status_code重定向状态,重定向到expected_url(包括任何GET数据),并且最终页面是使用target_status_code接收的.

如果您的请求使用了follow参数,那么expected_url和target_status_code将是重定向链最后一个点的url和status代码.

如果fetch_redirect_response为False,则不会加载最终页面.由于测试客户端无法获取外部URL,因此如果expected_url不是Django应用程序的一部分,则此功能特别有用.

在两个URL之间进行比较时,可以正确处理Scheme.如果在我们被重定向到的位置中没有指定任何方案,则使用原始请求的方案.如果存在,expected_url中的方案是用于进行比较的方案.

Answer 2

你也可以按照重定向:

response = self.client.get('/myprofile/data/some_id/', follow=True)

这将反映浏览器中的用户体验并断言您期望在那里找到的内容,例如:

self.assertContains(response, "You must be logged in", status_code=401)

Answer 3

您可以检查response['Location']它是否与预期的网址匹配.还要检查状态代码是否为302.

Answer 4

response['Location']在1.9中不存在.请改用:

response = self.client.get('/myprofile/data/some_id/', follow=True)
last_url, status_code = response.redirect_chain[-1]
print(last_url)