將 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中文網其他相關文章！