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