From 2de731453f4f1bdb6941b46766aca02f1d7d5434 Mon Sep 17 00:00:00 2001 From: William Brooks Date: Fri, 21 Feb 2020 01:43:03 -0600 Subject: [PATCH] really fix driver_mock this time --- builder/hyperv/common/driver_mock.go | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/builder/hyperv/common/driver_mock.go b/builder/hyperv/common/driver_mock.go index 64cb14fa5..eb633bcd3 100644 --- a/builder/hyperv/common/driver_mock.go +++ b/builder/hyperv/common/driver_mock.go @@ -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