Django - 从 TemplateResponse 获取 URL

Question

我有一个使用内置登录系统的 Django 应用程序。

我正在尝试编写一个单元测试来验证 login url 是否重定向到 home 如果用户已经通过身份验证。这意味着，如果已通过身份验证的用户尝试访问 login 网址，他们将被重定向到 home。

这是 Django 的LoginView 的内置功能，我已经验证它确实可以工作。但是，我不确定如何为此编写单元测试。

问题：如何通过单元测试验证此重定向是否发生？ LoginView 返回一个TemplateResponse，它没有url 参数。

到目前为止，我所拥有的是：

from django.test import TestCase
from django.urls import reverse

home  = reverse('home')
login = reverse('login')
logout = reverse('logout')
signup = reverse('signup')

USER_CREDS = { some users credentials }

class TestLoginView(TestCase):
    def test_login_redirect(self):
        self.client.post(signup, USER_CREDS)
        self.client.get(logout)
        self.client.post(login, USER_CREDS)
        response = self.client.get(login)
        self.assertRedirects(response, home)

编辑： SimpleTestCase.assertRedirects 不适用于 TemplateResponse 对象。

TemplateResponse 对象不包含url 参数，这是SimpleTestCase.assertRedirect 所必需的。

该方法引发以下错误：AttributeError: 'TemplateResponse' object has no attribute 'url'

这与链接问题不同，因为来自LoginView 的响应对象是TemplateResponse，而不是HttpResponse。

Answer 1

您可以使用assertRedirects 检查登录过程是否重定向到所需的命名 URL：

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 的最后一页。

然后在你的测试中：

self.assertRedirects(response, home)

编辑

我也遇到过类似的问题，注意几点：

您可以使用跳过身份验证和验证步骤的TestCase.force_login，而不是执行登录过程。

class TestLoginView(TestCase):
    def test_login_redirect(self):
        self.client.force_login(user)

        # logged user retrieves the login page
        response = self.client.get(login)

        self.assertRedirects(response, home)

响应应该是 302 重定向到 home: <HttpResponseRedirect status_code=302, "text/html; charset=utf-8", url="/">

如果您在执行response = self.client.get(login) 时没有收到重定向，则表示它正在加载登录页面，这是您不想要的，因此您必须修复该错误。

所以问题不在于 TemplateResponse 对象不包含 url 参数，而是您的 login 页面没有将登录用户重定向到主页。