Django DRF API 单元测试完整方案（基于 TestCase）

一、方案概述

使用 django.test.TestCase 和 rest_framework.test.APIClient 进行 API 单元测试，确保 API 正确性、权限控制、数据返回格式、业务逻辑 等。

---

二、基本步骤

1. 使用 setUp() 初始化测试环境

   • 创建 API 客户端 APIClient()
   • 预先插入数据库测试数据（如普通用户、管理员用户等）
   • 生成 Token 进行身份认证（如 TokenAuthentication）

2. 编写测试用例

   • 发送 GET、POST、PUT、DELETE 请求
   • 断言 HTTP 状态码、数据格式、权限逻辑等

3. 运行测试

   • 执行 python manage.py test 运行测试

---

三、测试场景 1：权限控制（用户管理 API）

HTTP 方法	端点	功能描述
POST	/api/users/	创建用户
GET	/api/users/	获取用户列表（需要身份认证）
GET	/api/users/{id}/	获取用户详情（仅管理员可查看）
PUT	/api/users/{id}/	更新用户信息（仅限用户本人）
DELETE	/api/users/{id}/	删除用户（仅管理员可删除）

完整的 API 测试代码

from django.test import TestCase
from rest_framework.test import APIClient
from django.contrib.auth.models import User
from rest_framework.authtoken.models import Tokenclass UserAPITestCase(TestCase):"""用户 API 测试"""def setUp(self):"""初始化 API 客户端，并创建测试用户和管理员"""self.client = APIClient()# 创建普通用户self.user = User.objects.create_user(username="testuser", email="test@example.com", password="password123")# 创建管理员用户self.admin_user = User.objects.create_superuser(username="admin", email="admin@example.com", password="adminpass")# 生成 Tokenself.token, _ = Token.objects.get_or_create(user=self.user)self.admin_token, _ = Token.objects.get_or_create(user=self.admin_user)def test_get_users_without_authentication(self):"""测试未登录获取用户列表，应该返回 403"""response = self.client.get("/api/users/")self.assertEqual(response.status_code, 403)def test_get_users_with_authentication(self):"""测试登录后获取用户列表"""self.client.credentials(HTTP_AUTHORIZATION=f"Token {self.token.key}")response = self.client.get("/api/users/")self.assertEqual(response.status_code, 200)

---

四、测试场景 2：API 依赖前置 setUp()（菜品管理 API）

HTTP 方法	端点	功能描述
POST	/api/categories/	创建菜品分类
GET	/api/categories/	获取菜品分类列表
POST	/api/dishes/	创建菜品（必须指定已有的菜品分类）
GET	/api/dishes/	获取菜品列表

完整的 API 测试代码

from django.test import TestCase
from rest_framework.test import APIClient
from myapp.models import Category, Dishclass DishAPITestCase(TestCase):"""菜品 API 测试"""def setUp(self):"""初始化 API 客户端，并创建测试数据"""self.client = APIClient()# 创建一个分类self.category = Category.objects.create(name="热菜")def test_create_dish_with_valid_category(self):"""测试创建菜品（分类有效）"""data = {"name": "宫保鸡丁", "price": "25.00", "category": self.category.id}response = self.client.post("/api/dishes/", data, format="json")self.assertEqual(response.status_code, 201)self.assertEqual(response.data["name"], "宫保鸡丁")def test_create_dish_with_invalid_category(self):"""测试创建菜品（分类不存在）"""data = {"name": "鱼香肉丝", "price": "22.00", "category": 999}  # 不存在的分类 IDresponse = self.client.post("/api/dishes/", data, format="json")self.assertEqual(response.status_code, 400)self.assertIn("分类不存在", response.data["category"])

---

五、运行测试

python manage.py test myapp

---

六、方案总结

功能	方法	断言
获取用户列表（未登录）	test_get_users_without_authentication()	self.assertEqual(response.status_code, 403)
获取用户列表（已登录）	test_get_users_with_authentication()	self.assertEqual(response.status_code, 200)
创建菜品（分类有效）	test_create_dish_with_valid_category()	self.assertEqual(response.status_code, 201)
创建菜品（分类不存在）	test_create_dish_with_invalid_category()	self.assertEqual(response.status_code, 400)

✅ 覆盖了权限控制 + API 依赖数据，确保 Django DRF API 的稳定性