pytest-间谍方法调用因对象

发布于 2025-02-12 16:38:56 字数 704 浏览 0 评论 0原文

class Work:
    pass
class Engineer:

   def delegate(work: Work):
       print('Doing all the hard work')
class Manager:

   def __init__(self, engineer: Engineer):
       self.engineer = engineer

   def perform(work: Work):
       self.engineer.delegate(work)

我想测试Manager的Perrive()方法。在这样做的过程中,我想验证使用正确的参数调用Engineer.delegate()方法:

    # pseudo 

    work = mock(Work)
    engr = mock(Engineer)

    mgr = spy(Manager)
    mgr.perform(work)

    verify(engr.delegate).times(1).capture(args)
    assert args == work
    

我似乎无法将我的头缠住以上如何用pytest和pytest mock完成上述内容。

class Work:
    pass
class Engineer:

   def delegate(work: Work):
       print('Doing all the hard work')
class Manager:

   def __init__(self, engineer: Engineer):
       self.engineer = engineer

   def perform(work: Work):
       self.engineer.delegate(work)

I want to test Manager's perform() method. In doing so I want to verify if engineer.delegate() method was called with the right params:

    # pseudo 

    work = mock(Work)
    engr = mock(Engineer)

    mgr = spy(Manager)
    mgr.perform(work)

    verify(engr.delegate).times(1).capture(args)
    assert args == work
    

I can't seem to wrap my head over how the above can be accomplished with pytest and pytest-mock.

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

深海少女心 2025-02-19 16:38:56

由于您将依赖项注入Manager类,因此测试是直接的。如您的伪代码中所述,我们只需要为依赖项创建一些模拟,然后我们可以使用 间谍 验证呼叫数量和参数,函数被调用。

一件事要提及的是,在您的Manager.perform功能定义中,您忘记了包括self作为传递的参数之一。

from src.manager import Manager, Engineer, Work


def test_manager_delegation(mocker):
    mock_engineer = mocker.Mock(spec=Engineer)
    mock_work = mocker.Mock(spec=Work)
    manager = Manager(mock_engineer)

    perform_spy = mocker.spy(manager, "perform")
    delegate_spy = mocker.spy(mock_engineer, "delegate")
    manager.perform(mock_work)

    perform_spy.assert_called_once_with(mock_work)
    delegate_spy.assert_called_once_with(mock_work)

=========================================== test session starts ===========================================
platform darwin -- Python 3.8.9, pytest-7.0.1, pluggy-1.0.0
rootdir: ***
plugins: asyncio-0.18.3, hypothesis-6.48.1, mock-3.7.0
asyncio: mode=strict
collected 1 item                                                                                          

tests/test_manager.py .                                                                             [100%]

============================================ 1 passed in 0.01s ============================================

Since you inject your dependencies into your Manager class it is straight-forward to test. As noted in your pseudo-code we just need to create some mocks for the dependencies and then we can use spy to verify number of calls and the arguments the functions were called with.

One thing to mention, in your Manager.perform function definition, you forgot to include self as being one of the arguments passed in.

from src.manager import Manager, Engineer, Work


def test_manager_delegation(mocker):
    mock_engineer = mocker.Mock(spec=Engineer)
    mock_work = mocker.Mock(spec=Work)
    manager = Manager(mock_engineer)

    perform_spy = mocker.spy(manager, "perform")
    delegate_spy = mocker.spy(mock_engineer, "delegate")
    manager.perform(mock_work)

    perform_spy.assert_called_once_with(mock_work)
    delegate_spy.assert_called_once_with(mock_work)

=========================================== test session starts ===========================================
platform darwin -- Python 3.8.9, pytest-7.0.1, pluggy-1.0.0
rootdir: ***
plugins: asyncio-0.18.3, hypothesis-6.48.1, mock-3.7.0
asyncio: mode=strict
collected 1 item                                                                                          

tests/test_manager.py .                                                                             [100%]

============================================ 1 passed in 0.01s ============================================
~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文