diff --git a/builder/hyperv/common/driver_mock.go b/builder/hyperv/common/driver_mock.go index 71b7a0750..7a3e6bcab 100644 --- a/builder/hyperv/common/driver_mock.go +++ b/builder/hyperv/common/driver_mock.go @@ -66,6 +66,11 @@ type DriverMock struct { 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 @@ -335,6 +340,13 @@ func (d *DriverMock) GetVirtualMachineNetworkAdapterAddress(vmName string) (stri 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 = vmName + return d.ReplaceVirtualMachineNetworkAdapter_Err +} + func (d *DriverMock) SetNetworkAdapterVlanId(switchName string, vlanId string) error { d.SetNetworkAdapterVlanId_Called = true d.SetNetworkAdapterVlanId_SwitchName = switchName