py.test 在自定义 funcargs 中使用 Monkeypatch

Question

我使用 py.test 并且非常喜欢 funcarg 方法将对象注入到测试函数中。 在我的测试中，我需要使用模拟对象，因为我有很多外部依赖项。我使用 Monkeypatch 将某些属性替换为模拟对象。

我遇到的问题是，我经常会进行一堆使用特定 funcarg 的测试，并且总是需要修补相同的属性。到目前为止，我替换了每个测试函数中的属性。

有没有办法在我的 funcarg 函数中使用 Monkeypatch，并从各个测试中删除这些重复的代码？

import sys
import pytest
from mock import Mock


#----------------------------------------------------------------------
def pytest_funcarg__api(request):
    """"""
    api = myclass()
    #do some initialisation...
    return api


#----------------------------------------------------------------------
def test_bla1(monkeypatch, api):
    """"""
    monkeypatch.setattr(api,"get_external_stuff",Mock())
    monkeypatch.setattr(api,"morestuff",Mock())

    api.do_something1()
    assert not api.a

#----------------------------------------------------------------------
def test_bla2(monkeypatch, api):
    """"""
    monkeypatch.setattr(api,"get_external_stuff",Mock())
    monkeypatch.setattr(api,"morestuff",Mock())

    api.do_something2()
    assert api.b


if __name__=='__main__':
    pytest.main(args=["-v",sys.argv[0]])

Answer 1

您可以使用记录的 getfuncargvalue 函数在内部使用来自另一个函数参数工厂的函数参数：

def pytest_funcarg__api(request):
    api = myclass()
    #do some initialisation...
    mp = request.getfuncargvalue("monkeypatch")
    mp.setattr(api,"get_external_stuff", Mock())
    mp.setattr(api,"morestuff", Mock())
    return api

Answer 2

这应该可行：

def pytest_funcarg__api(request):
    """"""
    api = myclass()
    #do some initialisation...
    mp_plugin = request.config.pluginmanager.getplugin("monkeypatch")
    monkeypatch = mp_plugin.pytest_funcarg__monkeypatch(request)
    monkeypatch.setattr(api,"get_external_stuff",Mock())
    monkeypatch.setattr(api,"morestuff",Mock())
    return api

这里的技巧有两个：

1. 我们使用 config.pluginmanager 获取 monkeypatch 插件。
2. 我们通过使用我们自己的请求对象调用其 pytest_funcarg__monkeypatch() funcarg 接口来欺骗 monkeypatch 插件，使其认为它是由 py.test 的依赖注入代码调用的。