Django单元测试生成器提示

我将为您生成Django单元测试，但您尚未提供具体的Viewset类。请提供您的Viewset代码，我将根据以下模板为您生成完整的测试用例： ```python from django.test import TestCase from rest_framework.test import APIClient from rest_framework import status from django.contrib.auth.models import User from .models import YourModel # 请替换为实际模型 from .serializers import YourSerializer # 请替换为实际序列化器 class YourViewSetTestCase(TestCase): """测试YourViewSet的所有CRUD操作""" def setUp(self): """初始化测试数据""" self.client = APIClient() # 创建测试用户 self.user = User.objects.create_user( username='testuser', password='testpass123' ) self.other_user = User.objects.create_user( username='otheruser', password='testpass123' ) # 创建测试数据 self.test_object = YourModel.objects.create( # 根据实际模型字段填写 name='Test Object', owner=self.user, # 其他必要字段 ) self.other_object = YourModel.objects.create( name='Other Object', owner=self.other_user, # 其他必要字段 ) self.valid_payload = { 'name': 'Updated Object', # 根据实际模型字段填写 } self.invalid_payload = { 'name': '', # 空值示例 # 其他无效字段 } # ========== 创建操作测试 ========== def test_create_object_authenticated(self): """测试已认证用户创建对象""" self.client.force_authenticate(user=self.user) response = self.client.post('/api/your-endpoint/', self.valid_payload, format='json') self.assertEqual(response.status_code, status.HTTP_201_CREATED) self.assertEqual(YourModel.objects.count(), 3) def test_create_object_unauthenticated(self): """测试未认证用户创建对象应返回401""" response = self.client.post('/api/your-endpoint/', self.valid_payload, format='json') self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED) def test_create_object_invalid_data(self): """测试创建对象时提供无效数据""" self.client.force_authenticate(user=self.user) response = self.client.post('/api/your-endpoint/', self.invalid_payload, format='json') self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST) # ========== 读取操作测试 ========== def test_list_objects_authenticated(self): """测试已认证用户获取对象列表""" self.client.force_authenticate(user=self.user) response = self.client.get('/api/your-endpoint/') self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(len(response.data), 2) # 假设返回所有对象 def test_list_objects_unauthenticated(self): """测试未认证用户获取列表应返回401""" response = self.client.get('/api/your-endpoint/') self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED) def test_retrieve_object_authenticated(self): """测试已认证用户获取单个对象""" self.client.force_authenticate(user=self.user) response = self.client.get(f'/api/your-endpoint/{self.test_object.id}/') self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(response.data['name'], 'Test Object') def test_retrieve_object_not_found(self): """测试获取不存在的对象应返回404""" self.client.force_authenticate(user=self.user) response = self.client.get('/api/your-endpoint/99999/') self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND) # ========== 更新操作测试 ========== def test_update_object_owner(self): """测试对象所有者更新对象""" self.client.force_authenticate(user=self.user) response = self.client.put( f'/api/your-endpoint/{self.test_object.id}/', self.valid_payload, format='json' ) self.assertEqual(response.status_code, status.HTTP_200_OK) self.test_object.refresh_from_db() self.assertEqual(self.test_object.name, 'Updated Object') def test_update_object_non_owner(self): """测试非所有者更新对象应返回403""" self.client.force_authenticate(user=self.other_user) response = self.client.put( f'/api/your-endpoint/{self.test_object.id}/', self.valid_payload, format='json' ) self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN) def test_partial_update_object(self): """测试部分更新对象""" self.client.force_authenticate(user=self.user) partial_payload = {'name': 'Partially Updated'} response = self.client.patch( f'/api/your-endpoint/{self.test_object.id}/', partial_payload, format='json' ) self.assertEqual(response.status_code, status.HTTP_200_OK) self.test_object.refresh_from_db() self.assertEqual(self.test_object.name, 'Partially Updated') # ========== 删除操作测试 ========== def test_delete_object_owner(self): """测试对象所有者删除对象""" self.client.force_authenticate(user=self.user) response = self.client.delete(f'/api/your-endpoint/{self.test_object.id}/') self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) self.assertEqual(YourModel.objects.count(), 1) def test_delete_object_non_owner(self): """测试非所有者删除对象应返回403""" self.client.force_authenticate(user=self.other_user) response = self.client.delete(f'/api/your-endpoint/{self.test_object.id}/') self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN) def test_delete_object_unauthenticated(self): """测试未认证用户删除对象应返回401""" response = self.client.delete(f'/api/your-endpoint/{self.test_object.id}/') self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED) # ========== 权限和边界测试 ========== def test_list_objects_filtered_by_user(self): """测试用户只能看到自己的对象（如果ViewSet有此逻辑）""" self.client.force_authenticate(user=self.user) response = self.client.get('/api/your-endpoint/') # 根据实际过滤逻辑调整断言 self.assertEqual(len(response.data), 1) def test_create_object_with_duplicate_name(self): """测试创建重复名称的对象（如果模型有唯一约束）""" self.client.force_authenticate(user=self.user) duplicate_payload = {'name': 'Test Object'} # 与现有对象同名 response = self.client.post('/api/your-endpoint/', duplicate_payload, format='json') self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST) def test_retrieve_object_with_invalid_id(self): """测试使用无效ID获取对象""" self.client.force_authenticate(user=self.user) response = self.client.get('/api/your-endpoint/abc/') self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND) def test_bulk_create_objects(self): """测试批量创建对象（如果支持）""" self.client.force_authenticate(user=self.user) bulk_payload = [ {'name': 'Bulk Object 1'}, {'name': 'Bulk Object 2'} ] response = self.client.post('/api/your-endpoint/bulk/', bulk_payload, format='json') self.assertEqual(response.status_code, status.HTTP_201_CREATED) self.assertEqual(YourModel.objects.count(), 4) ``` 请提供您的Viewset类代码，我将根据实际模型、字段、权限和URL配置来调整上述测试用例。

Django单元测试生成器提示

提示词内容收藏

用法

参考资料

讨论