python - django - 使用 django.test.Client 对用户进行身份验证

Question

在我的测试中，我尝试创建并验证用户，但当我点击 View 时，它仍然返回 302

def test_Home(self):
    self.client = Client()
    self.user = User.objects.create_user("xoxo", password="bar", is_staff=True)
    self.logged_in = self.client.login(
        username="xoxo", password="bar")
    r = self.client.get('/hudson/')
    self.assertEqual(r.status_code, HTTP_200_OK)

我的 /hudson/ View 很简单:

class HomeView(PermissionRequiredMixin, generic.TemplateView):
    template_name = 'foo/base.html'
    permission_required = ('user.is_staff', )
    login_url = reverse_lazy('admin:login')

按照 View 中的定义，我的 admin/login 收到 302 错误。

Answer 1

is_staff 是 user 实例上的属性，而不是权限。您被重定向是因为用户没有权限'user.is_staff'。

您可以使用UserPassesTestMixin改为混合:

from django.contrib.auth.mixins import UserPassesTestMixin

class HomeView(UserPassesTestMixin, generic.TemplateView):
    def test_func(self):
        return self.request.user.is_staff 

    template_name = 'foo/base.html'
    login_url = reverse_lazy('admin:login')

如果您在多个 View 中执行此操作，则可以创建自己的 mixin。

class UserIsStaffMixin(UserPassesTestMixin):
    def test_func(self):
        return self.request.user.is_staff 

    login_url = reverse_lazy('admin:login')

class HomeView(UserIsStaffMixin, generic.TemplateView):
    template_name = 'foo/base.html'