pytest的内置插件盘点7：python

最新推荐文章于 2024-05-20 22:54:06 发布

测试开发三木

最新推荐文章于 2024-05-20 22:54:06 发布

阅读量584

点赞数 7

分类专栏： pytest源码剖析文章标签： pytest python 服务器

本文链接：https://blog.csdn.net/python_sanmu/article/details/135908543

版权

pytest源码剖析专栏收录该内容

18 篇文章 4 订阅

订阅专栏

本文系《pytest源码剖析》系列内容

正在连载，欢迎关注

7. 内置插件 python

插件路径：_pytest.python

实现的 hook

hook	tryfirst	trylast	optionalhook	hookwrapper	wrapper
pytest_addoption	False	False	False	False	False
pytest_cmdline_main	False	False	False	False	False
pytest_collect_file	False	False	False	False	False
pytest_configure	False	False	False	False	False
pytest_generate_tests	False	False	False	False	False
pytest_pycollect_makeitem	False	False	False	False	True
pytest_pycollect_makemodule	False	False	False	False	False
pytest_pyfunc_call	False	False	False	False	True

调用的 hook

pytest_pycollect_makeitem
pytest_collect_file
pytest_pyfunc_call
pytest_make_parametrize_id
pytest_pycollect_makemodule
pytest_ignore_collect

插件功能

创建一系列 ini 配置，指定 python 用例的发现规则：
- python_files: 文件名，默认值 test_*.py, *_test.py
- python_classes：类名前缀，默认值 Test
- python_functions，函数名前缀，默认值 test
创建 Item 子类，作为测试用例对象，_pytest.python.Function
根据规则，从各目录、文件、类中，收集测试用例，实例化 Item
实现 hookpytest_pyfunc_call，执行用例中的代码

代码片段

def pytest_pycollect_makemodule(module_path: Path, parent) -&gt; "Module":
    if module_path.name == "__init__.py":
        pkg: Package = Package.from_parent(parent, path=module_path)
        return pkg
    mod: Module = Module.from_parent(parent, path=module_path)
    return mod

class Module(nodes.File, PyCollector):
    def collect(self) -&gt; Iterable[Union[nodes.Item, nodes.Collector]]:
        self._inject_setup_module_fixture()
        self._inject_setup_function_fixture()
        self.session._fixturemanager.parsefactories(self)
        return super().collect
    
@hookimpl(trylast=True)
def pytest_pyfunc_call(pyfuncitem: "Function") -&gt; Optional[object]:
    testfunction = pyfuncitem.obj
    if is_async_function(testfunction):
        async_warn_and_skip(pyfuncitem.nodeid)
    funcargs = pyfuncitem.funcargs
    testargs = {arg: funcargs[arg] for arg in pyfuncitem._fixtureinfo.argnames}
    result = testfunction(**testargs)
    if hasattr(result, "__await__") or hasattr(result, "__aiter__"):
        async_warn_and_skip(pyfuncitem.nodeid)
    elif result is not None:
        warnings.warn(
            PytestReturnNotNoneWarning(
                f"Expected None, but {pyfuncitem.nodeid} returned {result!r}, which will be an error in a "
                "future version of pytest.  Did you mean to use `assert` instead of `return`?"
            )
        )
    return True