将 pytest 命令添加到 Django manage.py

调用 pytest 的首选方法是通过 pytest 命令。即使应用程序 pytest-django 也不提供调用 pytest 的管理命令。

但我更喜欢通过一个地方调用我的 django 项目，即项目根目录下的 manage.py 脚本。所以我只是将其添加到脚本中：-

#!/usr/bin/env python
"""Django's command-line utility for administrative tasks."""
import os
import sys
import pathlib
import subprocess

def main():
    BASE_DIR = pathlib.Path(__file__).parent.absolute()
    sys.path[0] = str(BASE_DIR / "src")

    """Run administrative tasks."""
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "medan.settings")
    try:
        from django.core.management import execute_from_command_line
    except ImportError as exc:
        raise ImportError(
            "Couldn't import Django. Are you sure it's installed and "
            "available on your PYTHONPATH environment variable? Did you "
            "forget to activate a virtual environment?"
        ) from exc

    # wrap pytest command here so we don't have to manage
    # PYTHONPATH in multiple places. django-pytest unfortunately
    # doesn't provide native management commands so we still need
    # to invoke pytest directly
    if len(sys.argv) > 1 and sys.argv[1] == "pytest":
        pytest_args = " ".join(sys.argv[2:])
        os.environ["PYTHONPATH"] = sys.path[0]
        subprocess.run(f".venv/bin/pytest {pytest_args}", shell=True, env=os.environ)
    else:
        execute_from_command_line(sys.argv)


if __name__ == "__main__":
    main()

这允许我们运行 pytest：-

poetry run python manage.py pytest -s -x ./path/to/tests/test_*.py

以上是将 pytest 命令添加到 Django manage.py的详细内容。更多信息请关注PHP中文网其他相关文章！