diff --git a/tests/integration/backward_compatible/lifecycle_test.py b/tests/integration/backward_compatible/lifecycle_test.py index 3eef40facd..1c63d9ca1a 100644 --- a/tests/integration/backward_compatible/lifecycle_test.py +++ b/tests/integration/backward_compatible/lifecycle_test.py @@ -67,7 +67,12 @@ def test_lifecycle_listener_receives_disconnected_event(self): ) client.lifecycle_service.add_listener(collector) member.shutdown() - self.assertEqual(collector.events, [LifecycleState.DISCONNECTED]) + + def assertion(): + self.assertEqual(collector.events, [LifecycleState.DISCONNECTED]) + + self.assertTrueEventually(assertion) + client.shutdown() def test_remove_lifecycle_listener(self):