def setUp(self): mock_vm = flexmock(_FakeVM()) flexmock(virt).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 = VMTestRunner(job, None) mock_vm.should_receive('stop').once().ordered() mock_vm.should_receive('restore_snapshot').once().ordered()