def test_setup(self):
mock_vm = _FakeVM()
mock_vm.start = unittest.mock.Mock(return_value=True)
mock_vm.create_snapshot = unittest.mock.Mock()
mock_vm.stop = unittest.mock.Mock()
mock_vm.restore_snapshot = unittest.mock.Mock()
job_args = {'test_result_total': 1,
'vm_domain': 'domain',
'vm_username': '******',
'vm_hostname': 'hostname',
'vm_port': 22,
'vm_password': '******',
'vm_key_file': None,
'vm_cleanup': True,
'vm_no_copy': False,
'vm_timeout': 120,
'vm_hypervisor_uri': 'my_hypervisor_uri',
'reference': ['/tests/sleeptest.py',
'/tests/other/test',
'passtest.py'],
'dry_run': True,
'env_keep': None,
'keep_tmp': 'on',
'base_logdir': self.tmpdir.name}
with Job(job_args) as job:
with unittest.mock.patch('avocado_runner_vm.vm_connect',
return_value=mock_vm):
# VMTestRunner()
runner = avocado_runner_vm.VMTestRunner(job, None)
runner.setup()
runner.tear_down()
mock_vm.start.assert_called_once_with()
mock_vm.create_snapshot.assert_called_once_with()
mock_vm.stop.assert_called_once_with()
mock_vm.restore_snapshot.assert_called_once_with()
def setUp(self):
mock_vm = flexmock(_FakeVM())
flexmock(avocado_runner_vm).should_receive('vm_connect').and_return(
mock_vm).once().ordered()
mock_vm.should_receive('start').and_return(True).once().ordered()
mock_vm.should_receive('create_snapshot').once().ordered()
# VMTestRunner()
Args = flexmock(
test_result_total=1,
url=['/tests/sleeptest', '/tests/other/test', 'passtest'],
vm_domain='domain',
vm_username='******',
vm_hostname='hostname',
vm_port=22,
vm_password='******',
vm_key_file=None,
vm_cleanup=True,
vm_no_copy=False,
vm_timeout=120,
vm_hypervisor_uri='my_hypervisor_uri',
env_keep=None)
log = flexmock()
log.should_receive("info")
job = flexmock(args=Args, log=log)
self.runner = avocado_runner_vm.VMTestRunner(job, None)
mock_vm.should_receive('stop').once().ordered()
mock_vm.should_receive('restore_snapshot').once().ordered()
def test_setup(self):
mock_vm = _FakeVM()
mock_vm.start = mock.Mock(return_value=True)
mock_vm.create_snapshot = mock.Mock()
mock_vm.stop = mock.Mock()
mock_vm.restore_snapshot = mock.Mock()
job_args = argparse.Namespace(test_result_total=1,
vm_domain='domain',
vm_username='******',
vm_hostname='hostname',
vm_port=22,
vm_password='******',
vm_key_file=None,
vm_cleanup=True,
vm_no_copy=False,
vm_timeout=120,
vm_hypervisor_uri='my_hypervisor_uri',
reference=[
'/tests/sleeptest.py',
'/tests/other/test', 'passtest.py'
],
dry_run=True,
env_keep=None)
job = None
try:
job = Job(job_args)
job.setup()
with mock.patch('avocado_runner_vm.vm_connect',
return_value=mock_vm):
# VMTestRunner()
runner = avocado_runner_vm.VMTestRunner(job, None)
runner.setup()
runner.tear_down()
mock_vm.start.assert_called_once_with()
mock_vm.create_snapshot.assert_called_once_with()
mock_vm.stop.assert_called_once_with()
mock_vm.restore_snapshot.assert_called_once_with()
finally:
if job:
shutil.rmtree(job.args.base_logdir)