Django 测试在测试中存储会话数据

使用Django进行测试是开发Web应用程序的关键步骤之一。在测试过程中，我们经常需要模拟用户会话，并在会话中存储一些数据。Django提供了一种简单而强大的方法来处理这个问题，即使用测试客户端来模拟用户请求，并在测试中存储会话数据。

python
from django.test import Client

python
client = Client()

python
client.session['username'] = 'testuser'
client.session['is_logged_in'] = True

python
username = client.session.get('username')
is_logged_in = client.session.get('is_logged_in')

python
client.session.save()

python
from django.test import TestCase, Client
class LoginTestCase(TestCase):
    def test_login_success(self):
        client = Client()
        # 模拟用户登录成功
        client.session['username'] = 'testuser'
        client.session['is_logged_in'] = True
        client.session.save()
        # 发起登录请求
        response = client.get('/login/')
        # 断言登录成功
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.context['username'], 'testuser')
        self.assertEqual(response.context['is_logged_in'], True)