package common import ( "context" ) type DriverMock struct { IsRunning_Called bool IsRunning_VmName string IsRunning_Return bool IsRunning_Err error IsOff_Called bool IsOff_VmName string IsOff_Return bool IsOff_Err error Uptime_Called bool Uptime_VmName string Uptime_Return uint64 Uptime_Err error Start_Called bool Start_VmName string Start_Err error Stop_Called bool Stop_VmName string Stop_Err error Verify_Called bool Verify_Err error Mac_Called bool Mac_VmName string Mac_Return string Mac_Err error IpAddress_Called bool IpAddress_Mac string IpAddress_Return string IpAddress_Err error GetHostName_Called bool GetHostName_Ip string GetHostName_Return string GetHostName_Err error GetVirtualMachineGeneration_Called bool GetVirtualMachineGeneration_VmName string GetVirtualMachineGeneration_Return uint GetVirtualMachineGeneration_Err error GetHostAdapterIpAddressForSwitch_Called bool GetHostAdapterIpAddressForSwitch_SwitchName string GetHostAdapterIpAddressForSwitch_Return string GetHostAdapterIpAddressForSwitch_Err error TypeScanCodes_Called bool TypeScanCodes_VmName string TypeScanCodes_ScanCodes string TypeScanCodes_Err error GetVirtualMachineNetworkAdapterAddress_Called bool GetVirtualMachineNetworkAdapterAddress_VmName string GetVirtualMachineNetworkAdapterAddress_Return string GetVirtualMachineNetworkAdapterAddress_Err error ReplaceVirtualMachineNetworkAdapter_Called bool ReplaceVirtualMachineNetworkAdapter_VmName string ReplaceVirtualMachineNetworkAdapter_Replace bool ReplaceVirtualMachineNetworkAdapter_Err error SetNetworkAdapterVlanId_Called bool SetNetworkAdapterVlanId_SwitchName string SetNetworkAdapterVlanId_VlanId string SetNetworkAdapterVlanId_Err error SetVmNetworkAdapterMacAddress_Called bool SetVmNetworkAdapterMacAddress_VmName string SetVmNetworkAdapterMacAddress_Mac string SetVmNetworkAdapterMacAddress_Err error SetVirtualMachineVlanId_Called bool SetVirtualMachineVlanId_VmName string SetVirtualMachineVlanId_VlanId string SetVirtualMachineVlanId_Err error UntagVirtualMachineNetworkAdapterVlan_Called bool UntagVirtualMachineNetworkAdapterVlan_VmName string UntagVirtualMachineNetworkAdapterVlan_SwitchName string UntagVirtualMachineNetworkAdapterVlan_Err error CreateExternalVirtualSwitch_Called bool CreateExternalVirtualSwitch_VmName string CreateExternalVirtualSwitch_SwitchName string CreateExternalVirtualSwitch_Err error GetVirtualMachineSwitchName_Called bool GetVirtualMachineSwitchName_VmName string GetVirtualMachineSwitchName_Return string GetVirtualMachineSwitchName_Err error ConnectVirtualMachineNetworkAdapterToSwitch_Called bool ConnectVirtualMachineNetworkAdapterToSwitch_VmName string ConnectVirtualMachineNetworkAdapterToSwitch_SwitchName string ConnectVirtualMachineNetworkAdapterToSwitch_Err error DeleteVirtualSwitch_Called bool DeleteVirtualSwitch_SwitchName string DeleteVirtualSwitch_Err error CheckVMName_Called bool CheckVMName_Err error CreateVirtualSwitch_Called bool CreateVirtualSwitch_SwitchName string CreateVirtualSwitch_SwitchType string CreateVirtualSwitch_Return bool CreateVirtualSwitch_Err error AddVirtualMachineHardDrive_Called bool AddVirtualMachineHardDrive_VmName string AddVirtualMachineHardDrive_VhdFile string AddVirtualMachineHardDrive_VhdName string AddVirtualMachineHardDrive_VhdSizeBytes int64 AddVirtualMachineHardDrive_VhdBlockSize int64 AddVirtualMachineHardDrive_ControllerType string AddVirtualMachineHardDrive_Err error CreateVirtualMachine_Called bool CreateVirtualMachine_VmName string CreateVirtualMachine_Path string CreateVirtualMachine_HarddrivePath string CreateVirtualMachine_Ram int64 CreateVirtualMachine_DiskSize int64 CreateVirtualMachine_DiskBlockSize int64 CreateVirtualMachine_SwitchName string CreateVirtualMachine_Generation uint CreateVirtualMachine_DifferentialDisk bool CreateVirtualMachine_FixedVHD bool CreateVirtualMachine_Version string CreateVirtualMachine_Err error CloneVirtualMachine_Called bool CloneVirtualMachine_CloneFromVmcxPath string CloneVirtualMachine_CloneFromVmName string CloneVirtualMachine_CloneFromSnapshotName string CloneVirtualMachine_CloneAllSnapshots bool CloneVirtualMachine_VmName string CloneVirtualMachine_Path string CloneVirtualMachine_HarddrivePath string CloneVirtualMachine_Ram int64 CloneVirtualMachine_SwitchName string CloneVirtualMachine_Copy bool CloneVirtualMachine_Err error DeleteVirtualMachine_Called bool DeleteVirtualMachine_VmName string DeleteVirtualMachine_Err error SetVirtualMachineCpuCount_Called bool SetVirtualMachineCpuCount_VmName string SetVirtualMachineCpuCount_Cpu uint SetVirtualMachineCpuCount_Err error SetVirtualMachineMacSpoofing_Called bool SetVirtualMachineMacSpoofing_VmName string SetVirtualMachineMacSpoofing_Enable bool SetVirtualMachineMacSpoofing_Err error SetVirtualMachineDynamicMemory_Called bool SetVirtualMachineDynamicMemory_VmName string SetVirtualMachineDynamicMemory_Enable bool SetVirtualMachineDynamicMemory_Err error SetVirtualMachineSecureBoot_Called bool SetVirtualMachineSecureBoot_VmName string SetVirtualMachineSecureBoot_TemplateName string SetVirtualMachineSecureBoot_Enable bool SetVirtualMachineSecureBoot_Err error SetVirtualMachineVirtualizationExtensions_Called bool SetVirtualMachineVirtualizationExtensions_VmName string SetVirtualMachineVirtualizationExtensions_Enable bool SetVirtualMachineVirtualizationExtensions_Err error EnableVirtualMachineIntegrationService_Called bool EnableVirtualMachineIntegrationService_VmName string EnableVirtualMachineIntegrationService_IntegrationServiceName string EnableVirtualMachineIntegrationService_Err error ExportVirtualMachine_Called bool ExportVirtualMachine_VmName string ExportVirtualMachine_Path string ExportVirtualMachine_Err error PreserveLegacyExportBehaviour_Called bool PreserveLegacyExportBehaviour_SrcPath string PreserveLegacyExportBehaviour_DstPath string PreserveLegacyExportBehaviour_Err error MoveCreatedVHDsToOutputDir_Called bool MoveCreatedVHDsToOutputDir_SrcPath string MoveCreatedVHDsToOutputDir_DstPath string MoveCreatedVHDsToOutputDir_Err error CompactDisks_Called bool CompactDisks_Path string CompactDisks_Result string CompactDisks_Err error RestartVirtualMachine_Called bool RestartVirtualMachine_VmName string RestartVirtualMachine_Err error CreateDvdDrive_Called bool CreateDvdDrive_VmName string CreateDvdDrive_IsoPath string CreateDvdDrive_Generation uint CreateDvdDrive_ControllerNumber uint CreateDvdDrive_ControllerLocation uint CreateDvdDrive_Err error MountDvdDrive_Called bool MountDvdDrive_VmName string MountDvdDrive_Path string MountDvdDrive_ControllerNumber uint MountDvdDrive_ControllerLocation uint MountDvdDrive_Err error SetBootDvdDrive_Called bool SetBootDvdDrive_VmName string SetBootDvdDrive_ControllerNumber uint SetBootDvdDrive_ControllerLocation uint SetBootDvdDrive_Generation uint SetBootDvdDrive_Err error SetFirstBootDevice_Called bool SetFirstBootDevice_VmName string SetFirstBootDevice_ControllerType string SetFirstBootDevice_ControllerNumber uint SetFirstBootDevice_ControllerLocation uint SetFirstBootDevice_Generation uint SetFirstBootDevice_Err error SetBootOrder_Called bool SetBootOrder_VmName string SetBootOrder_BootOrder []string SetBootOrder_Err error UnmountDvdDrive_Called bool UnmountDvdDrive_VmName string UnmountDvdDrive_ControllerNumber uint UnmountDvdDrive_ControllerLocation uint UnmountDvdDrive_Err error DeleteDvdDrive_Called bool DeleteDvdDrive_VmName string DeleteDvdDrive_ControllerNumber uint DeleteDvdDrive_ControllerLocation uint DeleteDvdDrive_Err error MountFloppyDrive_Called bool MountFloppyDrive_VmName string MountFloppyDrive_Path string MountFloppyDrive_Err error UnmountFloppyDrive_Called bool UnmountFloppyDrive_VmName string UnmountFloppyDrive_Err error Connect_Called bool Connect_VmName string Connect_Cancel context.CancelFunc Connect_Err error Disconnect_Called bool Disconnect_Cancel context.CancelFunc } func (d *DriverMock) IsRunning(vmName string) (bool, error) { d.IsRunning_Called = true d.IsRunning_VmName = vmName return d.IsRunning_Return, d.IsRunning_Err } func (d *DriverMock) IsOff(vmName string) (bool, error) { d.IsOff_Called = true d.IsOff_VmName = vmName return d.IsOff_Return, d.IsOff_Err } func (d *DriverMock) Uptime(vmName string) (uint64, error) { d.Uptime_Called = true d.Uptime_VmName = vmName return d.Uptime_Return, d.Uptime_Err } func (d *DriverMock) Start(vmName string) error { d.Start_Called = true d.Start_VmName = vmName return d.Start_Err } func (d *DriverMock) Stop(vmName string) error { d.Stop_Called = true d.Stop_VmName = vmName return d.Stop_Err } func (d *DriverMock) Verify() error { d.Verify_Called = true return d.Verify_Err } func (d *DriverMock) Mac(vmName string) (string, error) { d.Mac_Called = true d.Mac_VmName = vmName return d.Mac_Return, d.Mac_Err } func (d *DriverMock) IpAddress(mac string) (string, error) { d.IpAddress_Called = true d.IpAddress_Mac = mac return d.IpAddress_Return, d.IpAddress_Err } func (d *DriverMock) GetHostName(ip string) (string, error) { d.GetHostName_Called = true d.GetHostName_Ip = ip return d.GetHostName_Return, d.GetHostName_Err } func (d *DriverMock) GetVirtualMachineGeneration(vmName string) (uint, error) { d.GetVirtualMachineGeneration_Called = true d.GetVirtualMachineGeneration_VmName = vmName return d.GetVirtualMachineGeneration_Return, d.GetVirtualMachineGeneration_Err } func (d *DriverMock) GetHostAdapterIpAddressForSwitch(switchName string) (string, error) { d.GetHostAdapterIpAddressForSwitch_Called = true d.GetHostAdapterIpAddressForSwitch_SwitchName = switchName return d.GetHostAdapterIpAddressForSwitch_Return, d.GetHostAdapterIpAddressForSwitch_Err } func (d *DriverMock) TypeScanCodes(vmName string, scanCodes string) error { d.TypeScanCodes_Called = true d.TypeScanCodes_VmName = vmName d.TypeScanCodes_ScanCodes = scanCodes return d.TypeScanCodes_Err } func (d *DriverMock) GetVirtualMachineNetworkAdapterAddress(vmName string) (string, error) { d.GetVirtualMachineNetworkAdapterAddress_Called = true d.GetVirtualMachineNetworkAdapterAddress_VmName = vmName return d.GetVirtualMachineNetworkAdapterAddress_Return, d.GetVirtualMachineNetworkAdapterAddress_Err } func (d *DriverMock) ReplaceVirtualMachineNetworkAdapter(vmName string, replace bool) error { d.ReplaceVirtualMachineNetworkAdapter_Called = true d.ReplaceVirtualMachineNetworkAdapter_VmName = vmName d.ReplaceVirtualMachineNetworkAdapter_Replace = replace return d.ReplaceVirtualMachineNetworkAdapter_Err } func (d *DriverMock) SetNetworkAdapterVlanId(switchName string, vlanId string) error { d.SetNetworkAdapterVlanId_Called = true d.SetNetworkAdapterVlanId_SwitchName = switchName d.SetNetworkAdapterVlanId_VlanId = vlanId return d.SetNetworkAdapterVlanId_Err } func (d *DriverMock) SetVmNetworkAdapterMacAddress(vmName string, mac string) error { d.SetVmNetworkAdapterMacAddress_Called = true d.SetVmNetworkAdapterMacAddress_VmName = vmName d.SetVmNetworkAdapterMacAddress_Mac = mac return d.SetVmNetworkAdapterMacAddress_Err } func (d *DriverMock) SetVirtualMachineVlanId(vmName string, vlanId string) error { d.SetVirtualMachineVlanId_Called = true d.SetVirtualMachineVlanId_VmName = vmName d.SetVirtualMachineVlanId_VlanId = vlanId return d.SetVirtualMachineVlanId_Err } func (d *DriverMock) UntagVirtualMachineNetworkAdapterVlan(vmName string, switchName string) error { d.UntagVirtualMachineNetworkAdapterVlan_Called = true d.UntagVirtualMachineNetworkAdapterVlan_VmName = vmName d.UntagVirtualMachineNetworkAdapterVlan_SwitchName = switchName return d.UntagVirtualMachineNetworkAdapterVlan_Err } func (d *DriverMock) CreateExternalVirtualSwitch(vmName string, switchName string) error { d.CreateExternalVirtualSwitch_Called = true d.CreateExternalVirtualSwitch_VmName = vmName d.CreateExternalVirtualSwitch_SwitchName = switchName return d.CreateExternalVirtualSwitch_Err } func (d *DriverMock) GetVirtualMachineSwitchName(vmName string) (string, error) { d.GetVirtualMachineSwitchName_Called = true d.GetVirtualMachineSwitchName_VmName = vmName return d.GetVirtualMachineSwitchName_Return, d.GetVirtualMachineSwitchName_Err } func (d *DriverMock) ConnectVirtualMachineNetworkAdapterToSwitch(vmName string, switchName string) error { d.ConnectVirtualMachineNetworkAdapterToSwitch_Called = true d.ConnectVirtualMachineNetworkAdapterToSwitch_VmName = vmName d.ConnectVirtualMachineNetworkAdapterToSwitch_SwitchName = switchName return d.ConnectVirtualMachineNetworkAdapterToSwitch_Err } func (d *DriverMock) DeleteVirtualSwitch(switchName string) error { d.DeleteVirtualSwitch_Called = true d.DeleteVirtualSwitch_SwitchName = switchName return d.DeleteVirtualSwitch_Err } func (d *DriverMock) CreateVirtualSwitch(switchName string, switchType string) (bool, error) { d.CreateVirtualSwitch_Called = true d.CreateVirtualSwitch_SwitchName = switchName d.CreateVirtualSwitch_SwitchType = switchType return d.CreateVirtualSwitch_Return, d.CreateVirtualSwitch_Err } func (d *DriverMock) AddVirtualMachineHardDrive(vmName string, vhdFile string, vhdName string, vhdSizeBytes int64, vhdDiskBlockSize int64, controllerType string) error { d.AddVirtualMachineHardDrive_Called = true d.AddVirtualMachineHardDrive_VmName = vmName d.AddVirtualMachineHardDrive_VhdFile = vhdFile d.AddVirtualMachineHardDrive_VhdName = vhdName d.AddVirtualMachineHardDrive_VhdSizeBytes = vhdSizeBytes d.AddVirtualMachineHardDrive_VhdSizeBytes = vhdDiskBlockSize d.AddVirtualMachineHardDrive_ControllerType = controllerType return d.AddVirtualMachineHardDrive_Err } func (d *DriverMock) CheckVMName(vmName string) error { d.CheckVMName_Called = true return d.CheckVMName_Err } func (d *DriverMock) CreateVirtualMachine(vmName string, path string, harddrivePath string, ram int64, diskSize int64, diskBlockSize int64, switchName string, generation uint, diffDisks bool, fixedVHD bool, version string) error { d.CreateVirtualMachine_Called = true d.CreateVirtualMachine_VmName = vmName d.CreateVirtualMachine_Path = path d.CreateVirtualMachine_HarddrivePath = harddrivePath d.CreateVirtualMachine_Ram = ram d.CreateVirtualMachine_DiskSize = diskSize d.CreateVirtualMachine_DiskBlockSize = diskBlockSize d.CreateVirtualMachine_SwitchName = switchName d.CreateVirtualMachine_Generation = generation d.CreateVirtualMachine_DifferentialDisk = diffDisks d.CreateVirtualMachine_Version = version return d.CreateVirtualMachine_Err } func (d *DriverMock) CloneVirtualMachine(cloneFromVmcxPath string, cloneFromVmName string, cloneFromSnapshotName string, cloneAllSnapshots bool, vmName string, path string, harddrivePath string, ram int64, switchName string, copyTF bool) error { d.CloneVirtualMachine_Called = true d.CloneVirtualMachine_CloneFromVmcxPath = cloneFromVmcxPath d.CloneVirtualMachine_CloneFromVmName = cloneFromVmName d.CloneVirtualMachine_CloneFromSnapshotName = cloneFromSnapshotName d.CloneVirtualMachine_CloneAllSnapshots = cloneAllSnapshots d.CloneVirtualMachine_VmName = vmName d.CloneVirtualMachine_Path = path d.CloneVirtualMachine_HarddrivePath = harddrivePath d.CloneVirtualMachine_Ram = ram d.CloneVirtualMachine_SwitchName = switchName d.CloneVirtualMachine_Copy = copyTF return d.CloneVirtualMachine_Err } func (d *DriverMock) DeleteVirtualMachine(vmName string) error { d.DeleteVirtualMachine_Called = true d.DeleteVirtualMachine_VmName = vmName return d.DeleteVirtualMachine_Err } func (d *DriverMock) SetVirtualMachineCpuCount(vmName string, cpu uint) error { d.SetVirtualMachineCpuCount_Called = true d.SetVirtualMachineCpuCount_VmName = vmName d.SetVirtualMachineCpuCount_Cpu = cpu return d.SetVirtualMachineCpuCount_Err } func (d *DriverMock) SetVirtualMachineMacSpoofing(vmName string, enable bool) error { d.SetVirtualMachineMacSpoofing_Called = true d.SetVirtualMachineMacSpoofing_VmName = vmName d.SetVirtualMachineMacSpoofing_Enable = enable return d.SetVirtualMachineMacSpoofing_Err } func (d *DriverMock) SetVirtualMachineDynamicMemory(vmName string, enable bool) error { d.SetVirtualMachineDynamicMemory_Called = true d.SetVirtualMachineDynamicMemory_VmName = vmName d.SetVirtualMachineDynamicMemory_Enable = enable return d.SetVirtualMachineDynamicMemory_Err } func (d *DriverMock) SetVirtualMachineSecureBoot(vmName string, enable bool, templateName string) error { d.SetVirtualMachineSecureBoot_Called = true d.SetVirtualMachineSecureBoot_VmName = vmName d.SetVirtualMachineSecureBoot_Enable = enable d.SetVirtualMachineSecureBoot_TemplateName = templateName return d.SetVirtualMachineSecureBoot_Err } func (d *DriverMock) SetVirtualMachineVirtualizationExtensions(vmName string, enable bool) error { d.SetVirtualMachineVirtualizationExtensions_Called = true d.SetVirtualMachineVirtualizationExtensions_VmName = vmName d.SetVirtualMachineVirtualizationExtensions_Enable = enable return d.SetVirtualMachineVirtualizationExtensions_Err } func (d *DriverMock) EnableVirtualMachineIntegrationService(vmName string, integrationServiceName string) error { d.EnableVirtualMachineIntegrationService_Called = true d.EnableVirtualMachineIntegrationService_VmName = vmName d.EnableVirtualMachineIntegrationService_IntegrationServiceName = integrationServiceName return d.EnableVirtualMachineIntegrationService_Err } func (d *DriverMock) ExportVirtualMachine(vmName string, path string) error { d.ExportVirtualMachine_Called = true d.ExportVirtualMachine_VmName = vmName d.ExportVirtualMachine_Path = path return d.ExportVirtualMachine_Err } func (d *DriverMock) PreserveLegacyExportBehaviour(srcPath string, dstPath string) error { d.PreserveLegacyExportBehaviour_Called = true d.PreserveLegacyExportBehaviour_SrcPath = srcPath d.PreserveLegacyExportBehaviour_DstPath = dstPath return d.PreserveLegacyExportBehaviour_Err } func (d *DriverMock) MoveCreatedVHDsToOutputDir(srcPath string, dstPath string) error { d.MoveCreatedVHDsToOutputDir_Called = true d.MoveCreatedVHDsToOutputDir_SrcPath = srcPath d.MoveCreatedVHDsToOutputDir_DstPath = dstPath return d.MoveCreatedVHDsToOutputDir_Err } func (d *DriverMock) CompactDisks(path string) (result string, err error) { d.CompactDisks_Called = true d.CompactDisks_Path = path d.CompactDisks_Result = "Mock compact result msg: mockdisk.vhdx. Disk size reduced by 20%" return d.CompactDisks_Result, d.CompactDisks_Err } func (d *DriverMock) RestartVirtualMachine(vmName string) error { d.RestartVirtualMachine_Called = true d.RestartVirtualMachine_VmName = vmName return d.RestartVirtualMachine_Err } func (d *DriverMock) CreateDvdDrive(vmName string, isoPath string, generation uint) (uint, uint, error) { d.CreateDvdDrive_Called = true d.CreateDvdDrive_VmName = vmName d.CreateDvdDrive_IsoPath = isoPath d.CreateDvdDrive_Generation = generation return d.CreateDvdDrive_ControllerNumber, d.CreateDvdDrive_ControllerLocation, d.CreateDvdDrive_Err } func (d *DriverMock) MountDvdDrive(vmName string, path string, controllerNumber uint, controllerLocation uint) error { d.MountDvdDrive_Called = true d.MountDvdDrive_VmName = vmName d.MountDvdDrive_Path = path d.MountDvdDrive_ControllerNumber = controllerNumber d.MountDvdDrive_ControllerLocation = controllerLocation return d.MountDvdDrive_Err } func (d *DriverMock) SetBootDvdDrive(vmName string, controllerNumber uint, controllerLocation uint, generation uint) error { d.SetBootDvdDrive_Called = true d.SetBootDvdDrive_VmName = vmName d.SetBootDvdDrive_ControllerNumber = controllerNumber d.SetBootDvdDrive_ControllerLocation = controllerLocation d.SetBootDvdDrive_Generation = generation return d.SetBootDvdDrive_Err } func (d *DriverMock) SetFirstBootDevice(vmName string, controllerType string, controllerNumber uint, controllerLocation uint, generation uint) error { d.SetFirstBootDevice_Called = true d.SetFirstBootDevice_VmName = vmName d.SetFirstBootDevice_ControllerType = controllerType d.SetFirstBootDevice_ControllerNumber = controllerNumber d.SetFirstBootDevice_ControllerLocation = controllerLocation d.SetFirstBootDevice_Generation = generation return d.SetFirstBootDevice_Err } func (d *DriverMock) SetBootOrder(vmName string, bootOrder []string) error { d.SetBootOrder_Called = true d.SetBootOrder_VmName = vmName d.SetBootOrder_BootOrder = bootOrder return d.SetBootOrder_Err } func (d *DriverMock) UnmountDvdDrive(vmName string, controllerNumber uint, controllerLocation uint) error { d.UnmountDvdDrive_Called = true d.UnmountDvdDrive_VmName = vmName d.UnmountDvdDrive_ControllerNumber = controllerNumber d.UnmountDvdDrive_ControllerLocation = controllerLocation return d.UnmountDvdDrive_Err } func (d *DriverMock) DeleteDvdDrive(vmName string, controllerNumber uint, controllerLocation uint) error { d.DeleteDvdDrive_Called = true d.DeleteDvdDrive_VmName = vmName d.DeleteDvdDrive_ControllerNumber = controllerNumber d.DeleteDvdDrive_ControllerLocation = controllerLocation return d.DeleteDvdDrive_Err } func (d *DriverMock) MountFloppyDrive(vmName string, path string) error { d.MountFloppyDrive_Called = true d.MountFloppyDrive_VmName = vmName d.MountFloppyDrive_Path = path return d.MountFloppyDrive_Err } func (d *DriverMock) UnmountFloppyDrive(vmName string) error { d.UnmountFloppyDrive_Called = true d.UnmountFloppyDrive_VmName = vmName return d.UnmountFloppyDrive_Err } func (d *DriverMock) Connect(vmName string) (context.CancelFunc, error) { d.Connect_Called = true d.Connect_VmName = vmName return d.Connect_Cancel, d.Connect_Err } func (d *DriverMock) Disconnect(cancel context.CancelFunc) { d.Disconnect_Called = true d.Disconnect_Cancel = cancel }