kevinheyer/bootstrap-vz
1
0
Fork 0
mirror of https://github.com/kevingruesser/bootstrap-vz.git synced 2025-10-07 17:40:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix lock handling for virtualbox tests

Browse source
This commit is contained in:
Anders Ingemann 2015-01-25 13:30:32 +01:00
parent ef37cd257f
commit 5b48ce58c5
1 changed files with 24 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
tests/integration/instances/vbox.py
View file

@ -1,6 +1,7 @@
from instance import Instance
import virtualbox
from contextlib import contextmanager
from ..tools import waituntil
import logging
log = logging.getLogger(__name__)
@ -84,7 +85,8 @@ class VirtualBoxInstance(Instance):
def shutdown(self):
log.debug('Shutting down vbox machine `{name}\''.format(name=self.name))
self.session.console.power_down().wait_for_completion(-1)
lock(self.machine, self.session).unlock()
if not waituntil(lambda: self.machine.session_state == virtualbox.library.SessionState.unlocked):
raise LockingException('Timeout while waiting for the machine to become unlocked')
def destroy(self):
log.debug('Destroying vbox machine `{name}\''.format(name=self.name))
@ -105,15 +107,27 @@ class VirtualBoxInstance(Instance):
@contextmanager
def lock(machine, session):
if machine.session_state != virtualbox.library.SessionState.unlocked:
msg = ('Acquiring lock on machine failed, state was `{state}\' '
'instead of `Unlocked\'.'.format(state=str(machine.session_state)))
raise LockingException(msg)
machine.lock_machine(session, virtualbox.library.LockType.write)
yield session.machine
from ..tools import waituntil
if machine.session_state == virtualbox.library.SessionState.unlocked:
return
if machine.session_state == virtualbox.library.SessionState.unlocking:
waituntil(lambda: machine.session_state == virtualbox.library.SessionState.unlocked)
return
if machine.session_state == virtualbox.library.SessionState.spawning:
waituntil(lambda: machine.session_state == virtualbox.library.SessionState.locked)
if machine.session_state != virtualbox.library.SessionState.locked:
if not waituntil(lambda: machine.session_state == virtualbox.library.SessionState.unlocked):
msg = ('Error before trying to release lock on machine, state was `{state}\' '
'instead of `Locked\'.'.format(state=str(machine.session_state)))
raise LockingException(msg)
session.unlock_machine()
waituntil(lambda: machine.session_state == virtualbox.library.SessionState.unlocked)
if not waituntil(lambda: machine.session_state == virtualbox.library.SessionState.unlocked):
msg = ('Timeout while trying to release lock on machine, '
'last state was `{state}\''.format(state=str(machine.session_state)))
raise LockingException(msg)
class LockingException(Exception):
pass