fix for consumer errors in app #166

faust-streaming · Jun 29, 2021 · e3fe807 · e3fe807
1 parent 75a1231
commit e3fe807
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit/transport/test_consumer.py b/tests/unit/transport/test_consumer.py
@@ -1386,7 +1386,7 @@ async def test_verify_all_partitions_active(self, *, consumer):
 
         with patch("faust.transport.consumer.monotonic") as monotonic:
             now = monotonic.return_value = 391243.231
-            consumer.verify_all_partitions_active()
+            await consumer.verify_all_partitions_active()
 
             consumer.verify_event_path.assert_has_calls(
                 [
@@ -1405,13 +1405,13 @@ async def test_verify_all_partitions_active__bail_on_sleep(self, *, consumer):
         consumer.sleep = AsyncMock()
 
         async def on_sleep(secs):
-            if consumer.sleep.call_count == 2:
+            if consumer.sleep.call_count == 4:
                 consumer._stopped.set()
 
         consumer.sleep.side_effect = on_sleep
 
         with patch("faust.transport.consumer.monotonic") as monotonic:
             now = monotonic.return_value = 391243.231
-            consumer.verify_all_partitions_active()
+            await consumer.verify_all_partitions_active()
 
             consumer.verify_event_path.assert_called_with(now, TP3)