django如何在客户端测试中设置请求用户

Question

我正在测试一个视图,我的测试看起来像:

def test_profile(self, user_id):
    user = User.objects.create_user(username="myusername", password="password", email="abc@testmail.com")
    self.client.user = user

    print(user.id)

    request = self.client.get("/account/profile/{}/".format(user_id), follow=True)
    self.assertEqual(request.status_code, 200)

这里我的个人资料视图有login_required装饰.那么如何将用户设置为request.user

Answer 1

我试图自己做同样的事情,但发现Django Test Client没有在请求中设置用户,并且在通过任何其他方式使用Client时无法设置request.user.我使用了RequestFactory.

def setUp(self):
    self.request_factory = RequestFactory()
    self.user = User.objects.create_user(
        username='javed', email='javed@javed.com', password='my_secret')

def test_my_test_method(self):
    request = self.request_factory.post(reverse('home'), {'question_title_name':'my first question title', 'question_name':'my first question', 'question_tag_name':'first, question'})
    request.user = self.user
    response = home_page(request)

更多关于请求工厂的信息

Answer 2

尝试这个：

from django.test import TestCase, Client

from django.contrib.auth.models import User

class YourTestCase(TestCase):
    def test_profile(self, user_id):
        user = User.objects.create(username='testuser')
        user.set_password('12345')
        user.save()
        client = Client()
        client.login(username='testuser', password='12345')
        response = client.get("/account/profile/{}/".format(user.id), follow=True)
        self.assertEqual(response.status_code, 200)

在这里，我首先创建用户并设置用户的登录凭据。然后，我创建一个客户端并使用该用户登录。因此，在您的views.py中，当您执行request.user时，您将获得此用户。