Question

我在module_name / app.py

中有两个函数

async def f1(arg):
    # do something
    return arg + '1'


async def f2(arg):
    result = await f1(arg)
    return result

我尝试使用pytest和asynctest测试f2并模拟f1 只有我这样做才有效。

def sf_f1(arg):
    return 'some value'

@pytest.mark.asyncio
async def test_f2():
    with asynctest.mock.patch('module_name.app.f1', side_effect=sf_f1):
        assert 'some value' == await f2('test')

测试PASSED

但是，我希望做到这样的事情

import module_name

@pytest.fixture()
def mock_f1():
    return asynctest.CoroutineMock(module_name.app.f1, side_effect=sf_f1)


@pytest.mark.asyncio
async def test_f2_2(mock_f1):
    assert 'some value' == await f2('test')

我得到了

   assert 'some value' == await f2('test')
   AssertionError: assert 'some value' == 'test1'
     - some value
     + test1

为什么第二种方式不起作用？

Answer 1

在第二个示例中，在mock_f1 fixture中创建一个CoroutineMock对象并将其返回。但是，您不能覆盖module_name.app.f1功能：Mock - 类似对象不会自动修补任何内容。

以下是对您的示例的说明：

@pytest.mark.asyncio
async def test_f2_2(mock_f1):
    print('fixture value:', mock_f1)
    print('actual module_name.app.f1 function:', module_name.app.f1)
    assert 'some value' == await f2('test')

这会打印一些像这样的想法

fixture value: <CoroutineMock spec='function' id='139756096130688'>
actual module_name.app.f1 function: <function f1 at 0x7f1b7e1139d8>

当您致电f2时，它会使用模块中的f1功能，但不会覆盖该功能。

以下是这对你有用的方法：

@pytest.fixture
def mock_f1(monkeypatch):
    fake_f1 = asynctest.CoroutineMock(module_name.app.f1, side_effect=sf_f1)
    monkeypatch.setattr(module_name.app, 'f1', fake_f1)
    return fake_f1

您可能知道，monkeypatch将确保仅在灯具处于活动状态时应用更改。

Python模拟补丁使用pytest在另一个函数内的coroutine函数？

1 个答案: