Pytest 基于 pytest 命令行参数运行“设置”方法

Question

我的问题是，当我运行 pytest 时，如何执行基于 pytest 参数的设置方法？基本上我希望这个“设置方法”在运行任何测试之前先运行。

为简单起见，假设我有一个文件 'text_something.py'

import somemodule

def test_some(usefakemodule):
   ...

“somemodule”要么安装在环境中，要么没有安装。如果它已经安装，那么它将没有问题。如果没有，我需要通过运行一个基本上是 sys.path.append(XXX) 的方法来添加假模块的路径。

简而言之，如果我想使用假模块目录，我想运行“pytest -v --usefakemodule”，或者只使用“pytest -v”来使用本地安装的库（假设它安装在那里）。

我尝试在 conftest.py 中添加一个夹具 (scope=session)，但它似乎在执行“test_something.py”之前没有先运行/执行它，然后它会声明没有名为“somemodule”的模块

我可以在 conftest 中运行常规方法，但我不知道它如何依赖于 pytest 命令行参数 'usefakemodule'。

Answer 1

在您的 conftest.py 中，您使用 pytest_addoption 结合 pytest_cmdline_main， 详情请参阅文档。

import pytest


def pytest_addoption(parser):
    parser.addoption(
        "--usefakemodule", action="store_true", default=False, help="Use fake module "
    )

def pytest_cmdline_main(config):
    usefakemodule = config.getoption("--usefakemodule")
    print(usefakemodule)
    if usefakemodule:
        print("OK fake module ")
    else:
        print("no fakes")

Answer 2

我不知道你是否/如何让 pytest 接受额外的参数，但这里有一些其他的想法可以实现这一点：

1. 尝试导入真正的模块，如果遇到 ImportError，请更新加载路径：

try:
  import somemodule
except ImportError:
  sys.path.append(XXX)
  import somemodule

1. 或者，使用环境变量，并使用USE_FAKE_MODULE=true pytest -v 运行：

import os
if os.environ.get('USE_FAKE_MODULE'):
  sys.path.append(XXX)