Question

我使用py.test并且非常喜欢将对象注入测试函数的funcarg方法。在我的测试中，我需要使用Mock对象，因为我有很多外部依赖项。我使用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

这里的诀窍是双重的：

我们使用monkeypatch获取config.pluginmanager插件。
我们通过调用monkeypatch funcarg-interface和我们自己的请求对象来欺骗pytest_funcarg__monkeypatch()插件来思考它被py.test的依赖注入代码调用。

py.test在自定义函数中使用monkeypatch

2 个答案: