在 Django 项目中集成 pytest 可以显著提升测试的灵活性和可维护性。pytest 是一个功能强大的第三方测试框架，它支持简洁的语法、丰富的插件生态，并与 Django 项目无缝集成。 ### 配置 pytest-django 要在 Django 项目中使用 pytest，首先需要安装 `pytest` 和 `pytest-django` 包。建议在虚拟环境中进行操作： ```bash python -m venv venv source venv/bin/activate # Linux/Mac pip install pytest pytest-django ``` 然后，在项目的根目录下创建 `pytest.ini` 文件，用于配置 pytest 的运行参数： ```ini [pytest] DJANGO_SETTINGS_MODULE = myproject.settings python_files = tests.py test_*.py *_tests.py addopts = --verbosity=2 --tb=short ``` 这样配置后，pytest 将自动识别 Django 环境并加载相关设置[^3]。 ### 使用自定义测试运行器（manage.py） 如果希望通过 Django 的 `manage.py test` 命令来运行 pytest，则可以创建一个自定义的测试运行器。在项目中添加一个 `test_runner.py` 文件，内容如下： ```python import pytest class PytestTestRunner: def run_tests(self, test_labels): pytest_args = ['--pyargs'] if test_labels: pytest_args.extend(test_labels) return pytest.main(pytest_args) ``` 接着，在 `settings.py` 中设置测试运行器： ```python TEST_RUNNER = 'myproject.test_runner.PytestTestRunner' ``` 通过这种方式，可以在不改变现有命令的前提下，使用 pytest 执行测试任务[^1]。 ### 测试代码的组织结构 为了使测试代码更易于维护和扩展，建议采用模块化的方式组织测试目录。通常，Django 项目的测试可以分为单元测试、集成测试和功能测试等不同类型： ``` myproject/ ├── app/ │ ├── models.py │ ├── views.py │ └── utils.py └── tests/ ├── unit/ │ ├── test_models.py │ ├── test_views.py │ └── test_utils.py ├── integration/ │ └── test_api.py └── functional/ └── test_user_flows.py ``` 这种结构有助于清晰地划分不同层次的测试，便于团队协作和持续集成流程的构建[^2]。 ### Fixtures 的使用 pytest 提供了强大的 fixture 功能，用于管理测试上下文。例如，可以在 `conftest.py` 文件中定义通用的 fixture，以便在整个测试套件中复用： ```python import pytest from django.contrib.auth.models import User @pytest.fixture def create_user(): def _create_user(username='testuser', password='12345'): return User.objects.create_user(username=username, password=password) return _create_user ``` 在具体的测试文件中，可以直接调用该 fixture 来创建用户对象： ```python def test_user_creation(create_user): user = create_user() assert user.username == 'testuser' ``` 这种方式不仅提高了代码的可读性和可维护性，还简化了测试逻辑的编写过程[^4]。 ### 模拟外部依赖 在某些情况下，测试可能会涉及到外部服务或复杂的数据处理逻辑。此时，可以使用 `pytest-mock` 插件提供的 `mocker` fixture 来模拟这些依赖项的行为： ```python def test_external_service(mocker): mock_request = mocker.patch('requests.get') mock_request.return_value.status_code = 200 response = some_function_that_uses_requests() assert response.status_code == 200 ``` 通过这种方式，可以有效地隔离测试环境，确保测试结果的一致性和可靠性[^5]。 ---