diff options
Diffstat (limited to 'test/test_consumer_group.py')
| -rw-r--r-- | test/test_consumer_group.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/test/test_consumer_group.py b/test/test_consumer_group.py index 9fb057e..1acde5e 100644 --- a/test/test_consumer_group.py +++ b/test/test_consumer_group.py @@ -95,7 +95,12 @@ def test_group(kafka_broker, topic): generations = set([consumer._coordinator.generation for consumer in list(consumers.values())]) - if len(generations) == 1: + # New generation assignment is not complete until + # coordinator.rejoining = False + rejoining = any([consumer._coordinator.rejoining + for consumer in list(consumers.values())]) + + if not rejoining and len(generations) == 1: for c, consumer in list(consumers.items()): logging.info("[%s] %s %s: %s", c, consumer._coordinator.generation, |