[Qemu-devel] [PATCH RFC v4 10/13] iotests: add event_wait to VM class

[Vladimir Sementsov-Ogievskiy <vsementsov@parallels.com>]

Signed-off-by: John Snow <jsnow@redhat.com>
Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@parallels.com>
---
 tests/qemu-iotests/iotests.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/qemu-iotests/iotests.py b/tests/qemu-iotests/iotests.py
index 75640b2..294b158 100644
--- a/tests/qemu-iotests/iotests.py
+++ b/tests/qemu-iotests/iotests.py
@@ -203,6 +203,13 @@ class VM(object):
         self._qmp.clear_events()
         return events
 
+    def event_wait(self, name='BLOCK_JOB_COMPLETED', maxwait=10):
+        for _ in range(maxwait):
+            for event in self.get_qmp_events(wait=True):
+                if event['event'] == name:
+                    return event
+            return None
+
 index_re = re.compile(r'([^\[]+)\[([^\]]+)\]')
 
 class QMPTestCase(unittest.TestCase):
-- 
1.9.1