Question

我目前正在用pytest编写一些异步测试，发现自己遇到了以下情况。

考虑我们有asyncio.Queue名为peer2_subscriber，我们要检查它是否收到某条消息（触发某些操作后，为简洁起见省略）

peer, cmd, msg = await asyncio.wait_for(
    peer2_subscriber.get(),
    timeout=1,
)

assert peer == peer2
assert isinstance(cmd, Transactions)
assert msg[0].hash == txs[0].hash

现在，请考虑我想测试另一个asyncio.Queue没有被推送的东西。

我发现自己创造了这样一种辅助方法。

async def wait_with_fallback(fn, fallback):
    try:
        return await asyncio.wait_for(
            fn(),
            timeout=1
        )
    except asyncio.TimeoutError:
        return fallback

然后在测试中我写了类似的东西：

val = await wait_with_fallback(
    peer1_subscriber.get,
    None
)

assert val == None

我想知道我现有的模式是否存在？

Answer 1

你的模式是有效的，所以我会说它是正确的＆＃34;，对于某些正确的值......这里主要是风格的观点。我会写

await asyncio.sleep(1)
assert peer1_subscriber.empty()

或

await asyncio.sleep(1)
val = peer1_subscriber.get_nowait()
assert val is None

如何测试asyncio.Queue没有得到推动

1 个答案: