python - 测试异步或多线程代码中回调的执行

Question

我需要检查线程/异步代码中的某些操作是否按特定顺序发生。大致如下:

def test_threaded_stuff():
    # I can define the callbacks to the operation
    op = start_operation(callback1, callback2, cbargs...)
    op.checkpoint1.wait()
    # check that callback1 and callback2 have been invoked,
    # in that order
    op.checkpoint2.wait()
    # check that they have been invoked again, in reverse order

我可以提供将由操作调用的测试回调，但我无法将 py.test 断言放入其中，因为我需要测试它们执行的整体顺序，而不是任何状态个别回调。此外，一些回调是在单独的线程中执行的，这些线程不受 py.test 的控制。

为了测试这样的代码，我想出了以下模式:

def callback1(log):
    log(1)

def callback2(log):
    log(2)

def test_threaded_stuff():
    events = []
    op = start_operation(cb1, cb2, events.append)
    op.checkpoint1.wait()
    assert events == [1, 2]
    op.checkpoint2.wait()
    assert events == [1, 2, 2, 1]

在py.test中是否有一种惯用的方式来表达这一点？例如，一个可自动记录其调用的可调用装置，以便我可以在测试中查询调用。

如果需要具体示例，this file是一个示例，同一目录中的其他文件也是如此。

Answer 1

我认为 pytest 可能不需要具有特定的功能，标准的 Python 单元测试模块就足够了。

您可以利用 Mock 对象来跟踪对自身以及方法和属性的调用，reference 。

您可以将其与 assert_has_calls() 结合使用通过构建 calls 的列表您期望并想要测试。默认情况下，它还允许通过 any_order=False 参数测试调用的特定顺序。

因此，通过充分修补模块并在测试中传递模拟对象而不是回调，您将基本上能够创建测试。