[英]Test assert function order with mocks (pytest -> assert_has_calls)
[英]Python pytest-mock assert_has_calls
我正在研究名为 pytest-mock ( https://github.com/pytest-dev/pytest-mock ) 的优秀 pytest 插件,现在我正在尝试一些带有 assert_has_calls 的示例。 简而言之,我正在测试 B 类的一个实例,更具体地说,该实例如何与 A 类的实例交互(我在其中模拟了“time_sumption_task”方法)。
该示例正在使用 alt。 B(见代码注释)。 我更喜欢 alt。 A,而是直接模拟A类中的方法,而不是模拟通过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)])
在 pytest-mock 的作者提供的另一个答案的帮助下设法弄清楚了。
如果使用 alt,则应如下调用断言。 A:
mock_a.assert_has_calls([mocker.call(mocker.ANY, 100), mocker.call(mocker.ANY, 50)])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.