From 9e26599a7677453722b1de7b1756cd1675f81603 Mon Sep 17 00:00:00 2001 From: Jean-Paul Calderone Date: Fri, 16 Oct 2020 13:44:37 -0400 Subject: [PATCH] Fix the race condition --- integration/test_streaming_logs.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/integration/test_streaming_logs.py b/integration/test_streaming_logs.py index b921937f1..52c813f9b 100644 --- a/integration/test_streaming_logs.py +++ b/integration/test_streaming_logs.py @@ -136,10 +136,13 @@ def _test_streaming_logs(reactor, temp_dir, alice): client.on_close = Deferred() client.on_message = Deferred() + # Capture this now before on_message perhaps goes away. + racing = _race(client.on_close, client.on_message) + # Provoke _some_ log event. yield treq.get(node_url) - result = yield _race(client.on_close, client.on_message) + result = yield racing assert isinstance(result, Right) json.loads(result.value)