django-tdd

name	django-tdd
description	使用 pytest-django 的 Django 测试策略、TDD 方法论、factory_boy、mock、覆盖率，以及 Django REST Framework API 测试。
origin	ECC

使用 TDD 进行 Django 测试

使用 pytest、factory_boy 和 Django REST Framework 进行 Django 应用的测试驱动开发。

何时激活

• 编写新的 Django 应用
• 实现 Django REST Framework API
• 测试 Django 模型、视图和序列化器
• 为 Django 项目设置测试基础设施

Django 的 TDD 工作流

红-绿-重构循环

# 步骤 1：红 — 编写失败的测试
def test_user_creation():
    user = User.objects.create_user(email='[e-mail]', password='testpass123')
    assert user.email == '[e-mail]'
    assert user.check_password('testpass123')
    assert not user.is_staff

# 步骤 2：绿 — 让测试通过
# 创建 User 模型或工厂

# 步骤 3：重构 — 在保持测试通过的同时改进

设置

pytest 配置

# pytest.ini
[pytest]
DJANGO_SETTINGS_MODULE = config.settings.test
testpaths = tests
python_files = test_*.py
python_classes = Test*
python_functions = test_*
addopts =
    --reuse-db
    --nomigrations
    --cov=apps
    --cov-report=html
    --cov-report=term-missing
    --strict-markers
markers =
    slow: 标记为慢速测试
    integration: 标记为集成测试

测试设置

# config/settings/test.py
from .base import *

DEBUG = True
DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.sqlite3',
        'NAME': ':memory:',
    }
}

# 禁用迁移以加速
class DisableMigrations:
    def __contains__(self, item):
        return True

    def __getitem__(self, item):
        return None

MIGRATION_MODULES = DisableMigrations()

# 更快的密码哈希
PASSWORD_HASHERS = [
    'django.contrib.auth.hashers.MD5PasswordHasher',
]

# 邮件后端
EMAIL_BACKEND = 'django.core.mail.backends.console.EmailBackend'

# Celery 始终立即执行
CELERY_TASK_ALWAYS_EAGER = True
CELERY_TASK_EAGER_PROPAGATES = True

conftest.py

# tests/conftest.py
import pytest
from django.utils import timezone
from django.contrib.auth import get_user_model

User = get_user_model()

@pytest.fixture(autouse=True)
def timezone_settings(settings):
    """确保一致的时区。"""
    settings.TIME_ZONE = 'UTC'

@pytest.fixture
def user(db):
    """创建测试用户。"""
    return User.objects.create_user(
        email='[e-mail]',
        password='testpass123',
        username='testuser'
    )

@pytest.fixture
def admin_user(db):
    """创建管理员用户。"""
    return User.objects.create_superuser(
        email='[e-mail]',
        password='adminpass123',
        username='admin'
    )

@pytest.fixture
def authenticated_client(client, user):
    """返回已认证的客户端。"""
    client.force_login(user)
    return client

@pytest.fixture
def api_client():
    """返回 DRF API 客户端。"""
    from rest_framework.test import APIClient
    return APIClient()

@pytest.fixture
def ():
    
    api_client.force_authenticate(user=user)
     api_client

Factory Boy

工厂设置

# tests/factories.py
import factory
from factory import fuzzy
from datetime import datetime, timedelta
from django.contrib.auth import get_user_model
from apps.products.models import Product, Category

User = get_user_model()

class UserFactory(factory.django.DjangoModelFactory):
    """User 模型的工厂。"""

    class Meta:
        model = User

    email = factory.Sequence(lambda n: f"user{n}@example.com")
    username = factory.Sequence(lambda n: f"user{n}")
    password = factory.PostGenerationMethodCall('set_password', 'testpass123')
    first_name = factory.Faker('first_name')
    last_name = factory.Faker('last_name')
    is_active = True

class CategoryFactory(factory.django.DjangoModelFactory):
    """Category 模型的工厂。"""

    class Meta:
        model = Category

    name = factory.Faker('word')
    slug = factory.LazyAttribute(lambda obj: obj.name.lower())
    description = factory.Faker('text')

class ProductFactory(factory.django.DjangoModelFactory):
    """Product 模型的工厂。"""

    class Meta:
        model = Product

    name = factory.Faker('sentence', nb_words=3)
    slug = factory.LazyAttribute( obj: obj.name.lower().replace(, ))
    description = factory.Faker()
    price = fuzzy.FuzzyDecimal(, , )
    stock = fuzzy.FuzzyInteger(, )
    is_active = 
    category = factory.SubFactory(CategoryFactory)
    created_by = factory.SubFactory(UserFactory)


     ():
        
          create:
            
         extracted:
             tag  extracted:
                .tags.add(tag)

使用工厂

# tests/test_models.py
import pytest
from tests.factories import ProductFactory, UserFactory

def test_product_creation():
    """使用工厂测试产品创建。"""
    product = ProductFactory(price=100.00, stock=50)
    assert product.price == 100.00
    assert product.stock == 50
    assert product.is_active is True

def test_product_with_tags():
    """测试带标签的产品。"""
    tags = [TagFactory(name='electronics'), TagFactory(name='new')]
    product = ProductFactory(tags=tags)
    assert product.tags.count() == 2

def test_multiple_products():
    """测试创建多个产品。"""
    products = ProductFactory.create_batch(10)
    assert len(products) == 10

模型测试

模型测试

# tests/test_models.py
import pytest
from django.core.exceptions import ValidationError
from tests.factories import UserFactory, ProductFactory

class TestUserModel:
    """测试 User 模型。"""

    def test_create_user(self, db):
        """测试创建普通用户。"""
        user = UserFactory(email='[e-mail]')
        assert user.email == '[e-mail]'
        assert user.check_password('testpass123')
        assert not user.is_staff
        assert not user.is_superuser

    def test_create_superuser(self, db):
        """测试创建超级用户。"""
        user = UserFactory(
            email='[e-mail]',
            is_staff=True,
            is_superuser=True
        )
        assert user.is_staff
        assert user.is_superuser

    def test_user_str(self, db):
        """测试用户字符串表示。"""
        user = UserFactory(email='[e-mail]')
        assert str(user) == '[e-mail]'

class TestProductModel:
    """测试 Product 模型。"""

    def test_product_creation(self, db):
        
        product = ProductFactory()
         product.   
         product.is_active  
         product.created_at   

     ():
        
        product = ProductFactory(name=)
         product.slug == 

     ():
        
        product = ProductFactory(price=-)
         pytest.raises(ValidationError):
            product.full_clean()

     ():
        
        ProductFactory.create_batch(, is_active=)
        ProductFactory.create_batch(, is_active=)

        active_count = Product.objects.active().count()
         active_count == 

     ():
        
        product = ProductFactory(stock=)
        product.reduce_stock()
        product.refresh_from_db()
         product.stock == 

         pytest.raises(ValueError):
            product.reduce_stock()  

视图测试

Django 视图测试

# tests/test_views.py
import pytest
from django.urls import reverse
from tests.factories import ProductFactory, UserFactory

class TestProductViews:
    """测试产品视图。"""

    def test_product_list(self, client, db):
        """测试产品列表视图。"""
        ProductFactory.create_batch(10)

        response = client.get(reverse('products:list'))

        assert response.status_code == 200
        assert len(response.context['products']) == 10

    def test_product_detail(self, client, db):
        """测试产品详情视图。"""
        product = ProductFactory()

        response = client.get(reverse('products:detail', kwargs={'slug': product.slug}))

        assert response.status_code == 200
        assert response.context['product'] == product

    def test_product_create_requires_login(self, client, db):
        """测试创建产品需要登录。"""
        response = client.get(reverse('products:create'))

        assert response.status_code == 302
        assert response.url.startswith('/accounts/login/')

    def test_product_create_authenticated(self, authenticated_client, db):
        """测试已认证用户创建产品。"""
        response = authenticated_client.get(reverse())

         response.status_code == 

     ():
        
        data = {
            : ,
            : ,
            : ,
            : ,
            : category.,
        }

        response = authenticated_client.post(reverse(), data)

         response.status_code == 
         Product.objects.(name=).exists()

DRF API 测试

序列化器测试

# tests/test_serializers.py
import pytest
from rest_framework.exceptions import ValidationError
from apps.products.serializers import ProductSerializer
from tests.factories import ProductFactory

class TestProductSerializer:
    """测试 ProductSerializer。"""

    def test_serialize_product(self, db):
        """测试序列化产品。"""
        product = ProductFactory()
        serializer = ProductSerializer(product)

        data = serializer.data

        assert data['id'] == product.id
        assert data['name'] == product.name
        assert data['price'] == str(product.price)

    def test_deserialize_product(self, db):
        """测试反序列化产品数据。"""
        data = {
            'name': 'Test Product',
            'description': 'Test description',
            'price': '99.99',
            'stock': 10,
            'category': 1,
        }

        serializer = ProductSerializer(data=data)

        assert serializer.is_valid()
        product = serializer.save()

        assert product.name == 'Test Product'
        assert float(product.price) == 99.99

    def ():
        
        data = {
            : ,
            : ,
            : ,
        }

        serializer = ProductSerializer(data=data)

          serializer.is_valid()
           serializer.errors

     ():
        
        data = {
            : ,
            : ,
            : -,
        }

        serializer = ProductSerializer(data=data)

          serializer.is_valid()
           serializer.errors

API ViewSet 测试

# tests/test_api.py
import pytest
from rest_framework.test import APIClient
from rest_framework import status
from django.urls import reverse
from tests.factories import ProductFactory, UserFactory

class TestProductAPI:
    """测试 Product API 端点。"""

    @pytest.fixture
    def api_client(self):
        """返回 API 客户端。"""
        return APIClient()

    def test_list_products(self, api_client, db):
        """测试列出产品。"""
        ProductFactory.create_batch(10)

        url = reverse('api:product-list')
        response = api_client.get(url)

        assert response.status_code == status.HTTP_200_OK
        assert response.data['count'] == 10

    def test_retrieve_product(self, api_client, db):
        """测试获取单个产品。"""
        product = ProductFactory()

        url = reverse('api:product-detail', kwargs={'pk': product.id})
        response = api_client.get(url)

        assert response.status_code == status.HTTP_200_OK
        assert response.data['id'] == product.id

    def test_create_product_unauthorized(self, api_client, db):
        """测试未认证创建产品。"""
        url = reverse()
        data = {: , : }

        response = api_client.post(url, data)

         response.status_code == status.HTTP_401_UNAUTHORIZED

     ():
        
        url = reverse()
        data = {
            : ,
            : ,
            : ,
            : ,
        }

        response = authenticated_api_client.post(url, data)

         response.status_code == status.HTTP_201_CREATED
         response.data[] == 

     ():
        
        product = ProductFactory(created_by=authenticated_api_client.user)

        url = reverse(, kwargs={: product.})
        data = {: }

        response = authenticated_api_client.patch(url, data)

         response.status_code == status.HTTP_200_OK
         response.data[] == 

     ():
        
        product = ProductFactory(created_by=authenticated_api_client.user)

        url = reverse(, kwargs={: product.})
        response = authenticated_api_client.delete(url)

         response.status_code == status.HTTP_204_NO_CONTENT

     ():
        
        ProductFactory(price=)
        ProductFactory(price=)

        url = reverse()
        response = api_client.get(url, {: })

         response.status_code == status.HTTP_200_OK
         response.data[] == 

     ():
        
        ProductFactory(name=)
        ProductFactory(name=)

        url = reverse()
        response = api_client.get(url, {: })

         response.status_code == status.HTTP_200_OK
         response.data[] == 

Mock 和 Patch

Mock 外部服务

# tests/test_views.py
from unittest.mock import patch, Mock
import pytest

class TestPaymentView:
    """测试带 mock 支付网关的支付视图。"""

    @patch('apps.payments.services.stripe')
    def test_successful_payment(self, mock_stripe, client, user, product):
        """测试使用 mock Stripe 的成功支付。"""
        # 配置 mock
        mock_stripe.Charge.create.return_value = {
            'id': 'ch_123',
            'status': 'succeeded',
            'amount': 9999,
        }

        client.force_login(user)
        response = client.post(reverse('payments:process'), {
            'product_id': product.id,
            'token': 'tok_visa',
        })

        assert response.status_code == 302
        mock_stripe.Charge.create.assert_called_once()

    @patch('apps.payments.services.stripe')
    def test_failed_payment(self, mock_stripe, client, user, product):
        """测试失败的支付。"""
        mock_stripe.Charge.create.side_effect = Exception('Card declined')

        client.force_login(user)
        response = client.post(reverse('payments:process'), {
            'product_id': product.id,
            'token': 'tok_visa',
        })

        assert response.status_code == 
           response.url

Mock 邮件发送

# tests/test_email.py
from django.core import mail
from django.test import override_settings

@override_settings(EMAIL_BACKEND='django.core.mail.backends.locmem.EmailBackend')
def test_order_confirmation_email(db, order):
    """测试订单确认邮件。"""
    order.send_confirmation_email()

    assert len(mail.outbox) == 1
    assert order.user.email in mail.outbox[0].to
    assert 'Order Confirmation' in mail.outbox[0].subject

集成测试

完整流程测试

# tests/test_integration.py
import pytest
from django.urls import reverse
from tests.factories import UserFactory, ProductFactory

class TestCheckoutFlow:
    """测试完整结账流程。"""

    def test_guest_to_purchase_flow(self, client, db):
        """测试从访客到购买的完整流程。"""
        # 步骤 1：注册
        response = client.post(reverse('users:register'), {
            'email': '[e-mail]',
            'password': 'testpass123',
            'password_confirm': 'testpass123',
        })
        assert response.status_code == 302

        # 步骤 2：登录
        response = client.post(reverse('users:login'), {
            'email': '[e-mail]',
            'password': 'testpass123',
        })
        assert response.status_code == 302

        # 步骤 3：浏览产品
        product = ProductFactory(price=100)
        response = client.get(reverse('products:detail', kwargs={'slug': product.slug}))
        assert response.status_code == 200

        # 步骤 4：添加到购物车
        response = client.post(reverse('cart:add'), {
            'product_id': product.id,
            'quantity': 1,
        })
        assert response.status_code == 

        
        response = client.get(reverse())
         response.status_code == 
         product.name  response.content.decode()

        
         patch()  mock_payment:
            mock_payment.return_value = 
            response = client.post(reverse())

         response.status_code == 
         Order.objects.(user__email=).exists()

测试最佳实践

应该做

• 使用工厂：替代手动创建对象
• 每个测试一个断言：保持测试聚焦
• 描述性测试名称：test_user_cannot_delete_others_post
• 测试边缘情况：空输入、None 值、边界条件
• Mock 外部服务：不要依赖外部 API
• 使用 fixture：消除重复
• 测试权限：确保授权正确工作
• 保持测试快速：使用 --reuse-db 和 --nomigrations

不应该做

• 不要测试 Django 内部：信任 Django 能正常工作
• 不要测试第三方代码：信任库能正常工作
• 不要忽略失败测试：所有测试必须通过
• 不要让测试相互依赖：测试应该能以任何顺序运行
• 不要过度 mock：只 mock 外部依赖
• 不要测试私有方法：测试公共接口
• 不要使用生产数据库：始终使用测试数据库

覆盖率

覆盖率配置

# 运行带覆盖率的测试
pytest --cov=apps --cov-report=html --cov-report=term-missing

# 生成 HTML 报告
open htmlcov/index.html

覆盖率目标

组件	目标覆盖率
模型	90%+
序列化器	85%+
视图	80%+
服务	90%+
工具类	80%+
总体	80%+

快速参考

模式	用途
@pytest.mark.django_db	启用数据库访问
client	Django 测试客户端
api_client	DRF API 客户端
factory.create_batch(n)	创建多个对象
patch('module.function')	Mock 外部依赖
override_settings	临时更改设置
force_authenticate()	在测试中绕过认证
assertRedirects	检查重定向
assertTemplateUsed	验证模板使用
mail.outbox	检查已发送邮件

记住：测试就是文档。好的测试解释了你的代码应该如何工作。保持它们简单、可读和可维护。