qubes/tests: do not deadlock on .drain()

This commit is contained in:
Wojtek Porczyk 2017-09-21 14:17:36 +02:00
parent 71eefbea14
commit b78ad1993c

View File

@ -410,7 +410,11 @@ class QubesTestCase(unittest.TestCase):
# Check for unfinished libvirt business. # Check for unfinished libvirt business.
if libvirt_event_impl is not None: if libvirt_event_impl is not None:
self.loop.run_until_complete(libvirt_event_impl.drain()) try:
self.loop.run_until_complete(asyncio.wait_for(
libvirt_event_impl.drain(), timeout=4))
except asyncio.TimeoutError:
raise AssertionError('libvirt event impl drain timeout')
# Check there are no Tasks left. # Check there are no Tasks left.
assert not self.loop._ready assert not self.loop._ready