Tests: Stop draining LiveViewer when iterator is inactive

Change-Id: Ic67cea7c3fa57ed5c48bbab7d21d206a827852ad
Signed-off-by: Kienan Stewart <kstewart@efficios.com>
Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>

diff --git a/tests/utils/lttngtest/environment.py b/tests/utils/lttngtest/environment.py
index c6dd0397a773e6bf263c1082456576f422b4dc9c..fe37cfa7c19f8ac97eecb901985370a7c66c1b0e 100644 (file)
--- a/tests/utils/lttngtest/environment.py
+++ b/tests/utils/lttngtest/environment.py
@@ -140,6 +140,8 @@ class _LiveViewer:
         while True:
             try:
                 for msg in self._live_iterator:
+                    if type(msg) is bt2._MessageIteratorInactivityMessageConst:
+                        break
                     self._events.append(msg)
                 break
             except bt2.TryAgain as e: