python-faust/backport-Fixes-missing-await-in-consumer.wait_empty.patch

26 lines
930 B
Diff
Raw Permalink Normal View History

From 2a9cb8970b051e2536958bf7690b9cdaa22e3b55 Mon Sep 17 00:00:00 2001
From: Ask Solem <asksol@robinhood.com>
Date: Mon, 19 Nov 2018 13:00:47 -0800
Subject: [PATCH] Fixes missing `await` in consumer.wait_empty
---
faust/transport/consumer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/faust/transport/consumer.py b/faust/transport/consumer.py
index 17e60fab..873356d8 100644
--- a/faust/transport/consumer.py
+++ b/faust/transport/consumer.py
@@ -300,7 +300,7 @@ class Consumer(Service, ConsumerT):
self._waiting_for_ack = asyncio.Future(loop=self.loop)
try:
# wait for `ack()` to wake us up
- asyncio.wait_for(
+ await asyncio.wait_for(
self._waiting_for_ack, loop=self.loop, timeout=1)
except (asyncio.TimeoutError,
asyncio.CancelledError): # pragma: no cover
--
2.33.0