我正在研究名为 pytest-mock ( https://github.com/pytest-dev/pytest-mock ) 的优秀 pytest 插件,现在我正在尝试一些带有 assert_has_calls 的示例。简而言之,我正在测试 B 类的一个实例,更具体地说,该实例如何与 A 类的实例交互(我在其中模拟了“time_sumption_task”方法)。
该示例正在使用 alt。B(见代码中的注释)。我更喜欢alt。A,而是直接mock A类中的方法,而不是mock通过B类的实例(obj)访问的A类实例中的方法。
class A(object):
def do_time_consuming_task(self, timeout):
return True
class B(object):
def __init__(self):
self.a = A()
def do_work(self, timeout):
return self.a.do_time_consuming_task(timeout)
def test_calls(mocker):
# Prepare
obj = B()
#mock_a = mocker.patch.object(A, 'do_time_consuming_task', autospec=True) # Alt. A
mock_a = mocker.patch.object(obj.a, 'do_time_consuming_task', autospec=True) # Alt. B
mock_a.return_value = True
# Exercise
obj.do_work(timeout=100)
obj.do_work(timeout=50)
# Assert
mock_a.assert_has_calls([mocker.call(100), mocker.call(50)])