609 lines
21 KiB
Go
609 lines
21 KiB
Go
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
|
|
|
|
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
|
|
|
|
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) 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) 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
|
|
}
|