def test_virtual_machine_download_fail(helpers_run, vm_run, get, session):
os.environ['vcdriver_vm_ssh_username'] = '******'
os.environ['vcdriver_vm_ssh_password'] = '******'
load()
vm = VirtualMachine()
vm_object_mock = mock.MagicMock()
vm_object_mock.summary.guest.ipAddress = '127.0.0.1'
vm.__setattr__('_vm_object', vm_object_mock)
with pytest.raises(DownloadError):
vm.download('from', 'to')
def test_virtual_machine_download_success(helpers_run, vm_run, get, session):
os.environ['vcdriver_vm_ssh_username'] = '******'
os.environ['vcdriver_vm_ssh_password'] = '******'
load()
vm = VirtualMachine()
assert vm.download('from', 'to') is None
vm_object_mock = mock.MagicMock()
vm_object_mock.summary.guest.ipAddress = '127.0.0.1'
vm.__setattr__('_vm_object', vm_object_mock)
result_mock = mock.MagicMock()
result_mock.failed = False
get.return_value = result_mock
assert vm.download('from', 'to') == result_mock
