def test_virtual_machine_upload_fail(helpers_run, vm_run, put, connection):
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(UploadError):
vm.upload('from', 'to')
def test_virtual_machine_upload_success(helpers_run, vm_run, put, connection):
os.environ['vcdriver_vm_ssh_username'] = '******'
os.environ['vcdriver_vm_ssh_password'] = '******'
load()
vm = VirtualMachine()
assert vm.upload('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
put.return_value = result_mock
assert vm.upload('from', 'to') == result_mock
