really fix driver_mock this time

This commit is contained in:
William Brooks 2020-02-21 01:43:03 -06:00
parent 0a0fbfc33b
commit 2de731453f
1 changed files with 11 additions and 1 deletions

View File

@ -580,10 +580,20 @@ func (d *DriverMock) SetBootDvdDrive(vmName string, controllerNumber uint, contr
d.SetBootDvdDrive_ControllerNumber = controllerNumber
d.SetBootDvdDrive_ControllerLocation = controllerLocation
d.SetBootDvdDrive_Generation = generation
d.SetBootDvdDrive_LegacyGen1BootOrder = legacyGen1BootOrder
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) UnmountDvdDrive(vmName string, controllerNumber uint, controllerLocation uint) error {
d.UnmountDvdDrive_Called = true
d.UnmountDvdDrive_VmName = vmName