Django request.user 修改未在测试中发生

Question

I have a view that is changing a field of request.user :

def test(request):                  
    request.user.is_provider = False
    request.user.save()
    print(request.user.is_provider)
    return HttpResponse(status=200)

Now I am testing the function and I have the following test:

class RoleSwitchTests(TestCase):
    def test_switch_to_customer(self):
        User = get_user_model()
        user = User.objects.create_user(
            username='test',
            email='test',
            password='test',
            first_name='test',
            last_name='test',
            is_provider=True,
            is_admin=False,
        )
        self.client.login(username='test', password='test')
        response = self.client.post('/test/', follow=True)
        print(user.is_provider)
        self.assertEqual(response.status_code, 200)
        self.assertFalse(user.is_provider)

self.assertFalse(user.is_provider) fails here. For some reason, request.user.is_provider is False in test , but in test_switch_to_customer , user.is_provider is True . I know these refer to the same user because they have the same id , so why is the modification not being preserved here?

Answer 1

The view and the test are working on different User objects, that both have the same primary key. Hence changes made by the view are not done in the user in the test.

You can refetch the User object, for example with:

class RoleSwitchTests(TestCase):
    def test_switch_to_customer(self):
        User = get_user_model()
        user = User.objects.create_user(
            username='test',
            email='test',
            password='test',
            first_name='test',
            last_name='test',
            is_provider=True,
            is_admin=False,
        )
        self.client.login(username='test', password='test')
        response = self.client.post('/test/', follow=True)
          # fetch from database
        print(user.is_provider)
        self.assertEqual(response.status_code, 200)
        self.assertFalse(user.is_provider)

Answer 2

您应该在发布后使​​用refresh_from_db从数据库重新加载更新的用户

user.refresh_from_db()