github.phpd.cn/hashicorp/packer@v1.3.2/builder/hyperv/common/driver_mock.go (about) 1 package common 2 3 import ( 4 "context" 5 ) 6 7 type DriverMock struct { 8 IsRunning_Called bool 9 IsRunning_VmName string 10 IsRunning_Return bool 11 IsRunning_Err error 12 13 IsOff_Called bool 14 IsOff_VmName string 15 IsOff_Return bool 16 IsOff_Err error 17 18 Uptime_Called bool 19 Uptime_VmName string 20 Uptime_Return uint64 21 Uptime_Err error 22 23 Start_Called bool 24 Start_VmName string 25 Start_Err error 26 27 Stop_Called bool 28 Stop_VmName string 29 Stop_Err error 30 31 Verify_Called bool 32 Verify_Err error 33 34 Mac_Called bool 35 Mac_VmName string 36 Mac_Return string 37 Mac_Err error 38 39 IpAddress_Called bool 40 IpAddress_Mac string 41 IpAddress_Return string 42 IpAddress_Err error 43 44 GetHostName_Called bool 45 GetHostName_Ip string 46 GetHostName_Return string 47 GetHostName_Err error 48 49 GetVirtualMachineGeneration_Called bool 50 GetVirtualMachineGeneration_VmName string 51 GetVirtualMachineGeneration_Return uint 52 GetVirtualMachineGeneration_Err error 53 54 GetHostAdapterIpAddressForSwitch_Called bool 55 GetHostAdapterIpAddressForSwitch_SwitchName string 56 GetHostAdapterIpAddressForSwitch_Return string 57 GetHostAdapterIpAddressForSwitch_Err error 58 59 TypeScanCodes_Called bool 60 TypeScanCodes_VmName string 61 TypeScanCodes_ScanCodes string 62 TypeScanCodes_Err error 63 64 GetVirtualMachineNetworkAdapterAddress_Called bool 65 GetVirtualMachineNetworkAdapterAddress_VmName string 66 GetVirtualMachineNetworkAdapterAddress_Return string 67 GetVirtualMachineNetworkAdapterAddress_Err error 68 69 SetNetworkAdapterVlanId_Called bool 70 SetNetworkAdapterVlanId_SwitchName string 71 SetNetworkAdapterVlanId_VlanId string 72 SetNetworkAdapterVlanId_Err error 73 74 SetVmNetworkAdapterMacAddress_Called bool 75 SetVmNetworkAdapterMacAddress_VmName string 76 SetVmNetworkAdapterMacAddress_Mac string 77 SetVmNetworkAdapterMacAddress_Err error 78 79 SetVirtualMachineVlanId_Called bool 80 SetVirtualMachineVlanId_VmName string 81 SetVirtualMachineVlanId_VlanId string 82 SetVirtualMachineVlanId_Err error 83 84 UntagVirtualMachineNetworkAdapterVlan_Called bool 85 UntagVirtualMachineNetworkAdapterVlan_VmName string 86 UntagVirtualMachineNetworkAdapterVlan_SwitchName string 87 UntagVirtualMachineNetworkAdapterVlan_Err error 88 89 CreateExternalVirtualSwitch_Called bool 90 CreateExternalVirtualSwitch_VmName string 91 CreateExternalVirtualSwitch_SwitchName string 92 CreateExternalVirtualSwitch_Err error 93 94 GetVirtualMachineSwitchName_Called bool 95 GetVirtualMachineSwitchName_VmName string 96 GetVirtualMachineSwitchName_Return string 97 GetVirtualMachineSwitchName_Err error 98 99 ConnectVirtualMachineNetworkAdapterToSwitch_Called bool 100 ConnectVirtualMachineNetworkAdapterToSwitch_VmName string 101 ConnectVirtualMachineNetworkAdapterToSwitch_SwitchName string 102 ConnectVirtualMachineNetworkAdapterToSwitch_Err error 103 104 DeleteVirtualSwitch_Called bool 105 DeleteVirtualSwitch_SwitchName string 106 DeleteVirtualSwitch_Err error 107 108 CreateVirtualSwitch_Called bool 109 CreateVirtualSwitch_SwitchName string 110 CreateVirtualSwitch_SwitchType string 111 CreateVirtualSwitch_Return bool 112 CreateVirtualSwitch_Err error 113 114 AddVirtualMachineHardDrive_Called bool 115 AddVirtualMachineHardDrive_VmName string 116 AddVirtualMachineHardDrive_VhdFile string 117 AddVirtualMachineHardDrive_VhdName string 118 AddVirtualMachineHardDrive_VhdSizeBytes int64 119 AddVirtualMachineHardDrive_VhdBlockSize int64 120 AddVirtualMachineHardDrive_ControllerType string 121 AddVirtualMachineHardDrive_Err error 122 123 CreateVirtualMachine_Called bool 124 CreateVirtualMachine_VmName string 125 CreateVirtualMachine_Path string 126 CreateVirtualMachine_HarddrivePath string 127 CreateVirtualMachine_Ram int64 128 CreateVirtualMachine_DiskSize int64 129 CreateVirtualMachine_DiskBlockSize int64 130 CreateVirtualMachine_SwitchName string 131 CreateVirtualMachine_Generation uint 132 CreateVirtualMachine_DifferentialDisk bool 133 CreateVirtualMachine_FixedVHD bool 134 CreateVirtualMachine_Err error 135 136 CloneVirtualMachine_Called bool 137 CloneVirtualMachine_CloneFromVmcxPath string 138 CloneVirtualMachine_CloneFromVmName string 139 CloneVirtualMachine_CloneFromSnapshotName string 140 CloneVirtualMachine_CloneAllSnapshots bool 141 CloneVirtualMachine_VmName string 142 CloneVirtualMachine_Path string 143 CloneVirtualMachine_HarddrivePath string 144 CloneVirtualMachine_Ram int64 145 CloneVirtualMachine_SwitchName string 146 CloneVirtualMachine_Err error 147 148 DeleteVirtualMachine_Called bool 149 DeleteVirtualMachine_VmName string 150 DeleteVirtualMachine_Err error 151 152 SetVirtualMachineCpuCount_Called bool 153 SetVirtualMachineCpuCount_VmName string 154 SetVirtualMachineCpuCount_Cpu uint 155 SetVirtualMachineCpuCount_Err error 156 157 SetVirtualMachineMacSpoofing_Called bool 158 SetVirtualMachineMacSpoofing_VmName string 159 SetVirtualMachineMacSpoofing_Enable bool 160 SetVirtualMachineMacSpoofing_Err error 161 162 SetVirtualMachineDynamicMemory_Called bool 163 SetVirtualMachineDynamicMemory_VmName string 164 SetVirtualMachineDynamicMemory_Enable bool 165 SetVirtualMachineDynamicMemory_Err error 166 167 SetVirtualMachineSecureBoot_Called bool 168 SetVirtualMachineSecureBoot_VmName string 169 SetVirtualMachineSecureBoot_TemplateName string 170 SetVirtualMachineSecureBoot_Enable bool 171 SetVirtualMachineSecureBoot_Err error 172 173 SetVirtualMachineVirtualizationExtensions_Called bool 174 SetVirtualMachineVirtualizationExtensions_VmName string 175 SetVirtualMachineVirtualizationExtensions_Enable bool 176 SetVirtualMachineVirtualizationExtensions_Err error 177 178 EnableVirtualMachineIntegrationService_Called bool 179 EnableVirtualMachineIntegrationService_VmName string 180 EnableVirtualMachineIntegrationService_IntegrationServiceName string 181 EnableVirtualMachineIntegrationService_Err error 182 183 ExportVirtualMachine_Called bool 184 ExportVirtualMachine_VmName string 185 ExportVirtualMachine_Path string 186 ExportVirtualMachine_Err error 187 188 PreserveLegacyExportBehaviour_Called bool 189 PreserveLegacyExportBehaviour_SrcPath string 190 PreserveLegacyExportBehaviour_DstPath string 191 PreserveLegacyExportBehaviour_Err error 192 193 MoveCreatedVHDsToOutputDir_Called bool 194 MoveCreatedVHDsToOutputDir_SrcPath string 195 MoveCreatedVHDsToOutputDir_DstPath string 196 MoveCreatedVHDsToOutputDir_Err error 197 198 CompactDisks_Called bool 199 CompactDisks_Path string 200 CompactDisks_Result string 201 CompactDisks_Err error 202 203 RestartVirtualMachine_Called bool 204 RestartVirtualMachine_VmName string 205 RestartVirtualMachine_Err error 206 207 CreateDvdDrive_Called bool 208 CreateDvdDrive_VmName string 209 CreateDvdDrive_IsoPath string 210 CreateDvdDrive_Generation uint 211 CreateDvdDrive_ControllerNumber uint 212 CreateDvdDrive_ControllerLocation uint 213 CreateDvdDrive_Err error 214 215 MountDvdDrive_Called bool 216 MountDvdDrive_VmName string 217 MountDvdDrive_Path string 218 MountDvdDrive_ControllerNumber uint 219 MountDvdDrive_ControllerLocation uint 220 MountDvdDrive_Err error 221 222 SetBootDvdDrive_Called bool 223 SetBootDvdDrive_VmName string 224 SetBootDvdDrive_ControllerNumber uint 225 SetBootDvdDrive_ControllerLocation uint 226 SetBootDvdDrive_Generation uint 227 SetBootDvdDrive_Err error 228 229 UnmountDvdDrive_Called bool 230 UnmountDvdDrive_VmName string 231 UnmountDvdDrive_ControllerNumber uint 232 UnmountDvdDrive_ControllerLocation uint 233 UnmountDvdDrive_Err error 234 235 DeleteDvdDrive_Called bool 236 DeleteDvdDrive_VmName string 237 DeleteDvdDrive_ControllerNumber uint 238 DeleteDvdDrive_ControllerLocation uint 239 DeleteDvdDrive_Err error 240 241 MountFloppyDrive_Called bool 242 MountFloppyDrive_VmName string 243 MountFloppyDrive_Path string 244 MountFloppyDrive_Err error 245 246 UnmountFloppyDrive_Called bool 247 UnmountFloppyDrive_VmName string 248 UnmountFloppyDrive_Err error 249 250 Connect_Called bool 251 Connect_VmName string 252 Connect_Cancel context.CancelFunc 253 Connect_Err error 254 255 Disconnect_Called bool 256 Disconnect_Cancel context.CancelFunc 257 } 258 259 func (d *DriverMock) IsRunning(vmName string) (bool, error) { 260 d.IsRunning_Called = true 261 d.IsRunning_VmName = vmName 262 return d.IsRunning_Return, d.IsRunning_Err 263 } 264 265 func (d *DriverMock) IsOff(vmName string) (bool, error) { 266 d.IsOff_Called = true 267 d.IsOff_VmName = vmName 268 return d.IsOff_Return, d.IsOff_Err 269 } 270 271 func (d *DriverMock) Uptime(vmName string) (uint64, error) { 272 d.Uptime_Called = true 273 d.Uptime_VmName = vmName 274 return d.Uptime_Return, d.Uptime_Err 275 } 276 277 func (d *DriverMock) Start(vmName string) error { 278 d.Start_Called = true 279 d.Start_VmName = vmName 280 return d.Start_Err 281 } 282 283 func (d *DriverMock) Stop(vmName string) error { 284 d.Stop_Called = true 285 d.Stop_VmName = vmName 286 return d.Stop_Err 287 } 288 289 func (d *DriverMock) Verify() error { 290 d.Verify_Called = true 291 return d.Verify_Err 292 } 293 294 func (d *DriverMock) Mac(vmName string) (string, error) { 295 d.Mac_Called = true 296 d.Mac_VmName = vmName 297 return d.Mac_Return, d.Mac_Err 298 } 299 300 func (d *DriverMock) IpAddress(mac string) (string, error) { 301 d.IpAddress_Called = true 302 d.IpAddress_Mac = mac 303 return d.IpAddress_Return, d.IpAddress_Err 304 } 305 306 func (d *DriverMock) GetHostName(ip string) (string, error) { 307 d.GetHostName_Called = true 308 d.GetHostName_Ip = ip 309 return d.GetHostName_Return, d.GetHostName_Err 310 } 311 312 func (d *DriverMock) GetVirtualMachineGeneration(vmName string) (uint, error) { 313 d.GetVirtualMachineGeneration_Called = true 314 d.GetVirtualMachineGeneration_VmName = vmName 315 return d.GetVirtualMachineGeneration_Return, d.GetVirtualMachineGeneration_Err 316 } 317 318 func (d *DriverMock) GetHostAdapterIpAddressForSwitch(switchName string) (string, error) { 319 d.GetHostAdapterIpAddressForSwitch_Called = true 320 d.GetHostAdapterIpAddressForSwitch_SwitchName = switchName 321 return d.GetHostAdapterIpAddressForSwitch_Return, d.GetHostAdapterIpAddressForSwitch_Err 322 } 323 324 func (d *DriverMock) TypeScanCodes(vmName string, scanCodes string) error { 325 d.TypeScanCodes_Called = true 326 d.TypeScanCodes_VmName = vmName 327 d.TypeScanCodes_ScanCodes = scanCodes 328 return d.TypeScanCodes_Err 329 } 330 331 func (d *DriverMock) GetVirtualMachineNetworkAdapterAddress(vmName string) (string, error) { 332 d.GetVirtualMachineNetworkAdapterAddress_Called = true 333 d.GetVirtualMachineNetworkAdapterAddress_VmName = vmName 334 return d.GetVirtualMachineNetworkAdapterAddress_Return, d.GetVirtualMachineNetworkAdapterAddress_Err 335 } 336 337 func (d *DriverMock) SetNetworkAdapterVlanId(switchName string, vlanId string) error { 338 d.SetNetworkAdapterVlanId_Called = true 339 d.SetNetworkAdapterVlanId_SwitchName = switchName 340 d.SetNetworkAdapterVlanId_VlanId = vlanId 341 return d.SetNetworkAdapterVlanId_Err 342 } 343 344 func (d *DriverMock) SetVmNetworkAdapterMacAddress(vmName string, mac string) error { 345 d.SetVmNetworkAdapterMacAddress_Called = true 346 d.SetVmNetworkAdapterMacAddress_VmName = vmName 347 d.SetVmNetworkAdapterMacAddress_Mac = mac 348 return d.SetVmNetworkAdapterMacAddress_Err 349 } 350 351 func (d *DriverMock) SetVirtualMachineVlanId(vmName string, vlanId string) error { 352 d.SetVirtualMachineVlanId_Called = true 353 d.SetVirtualMachineVlanId_VmName = vmName 354 d.SetVirtualMachineVlanId_VlanId = vlanId 355 return d.SetVirtualMachineVlanId_Err 356 } 357 358 func (d *DriverMock) UntagVirtualMachineNetworkAdapterVlan(vmName string, switchName string) error { 359 d.UntagVirtualMachineNetworkAdapterVlan_Called = true 360 d.UntagVirtualMachineNetworkAdapterVlan_VmName = vmName 361 d.UntagVirtualMachineNetworkAdapterVlan_SwitchName = switchName 362 return d.UntagVirtualMachineNetworkAdapterVlan_Err 363 } 364 365 func (d *DriverMock) CreateExternalVirtualSwitch(vmName string, switchName string) error { 366 d.CreateExternalVirtualSwitch_Called = true 367 d.CreateExternalVirtualSwitch_VmName = vmName 368 d.CreateExternalVirtualSwitch_SwitchName = switchName 369 return d.CreateExternalVirtualSwitch_Err 370 } 371 372 func (d *DriverMock) GetVirtualMachineSwitchName(vmName string) (string, error) { 373 d.GetVirtualMachineSwitchName_Called = true 374 d.GetVirtualMachineSwitchName_VmName = vmName 375 return d.GetVirtualMachineSwitchName_Return, d.GetVirtualMachineSwitchName_Err 376 } 377 378 func (d *DriverMock) ConnectVirtualMachineNetworkAdapterToSwitch(vmName string, switchName string) error { 379 d.ConnectVirtualMachineNetworkAdapterToSwitch_Called = true 380 d.ConnectVirtualMachineNetworkAdapterToSwitch_VmName = vmName 381 d.ConnectVirtualMachineNetworkAdapterToSwitch_SwitchName = switchName 382 return d.ConnectVirtualMachineNetworkAdapterToSwitch_Err 383 } 384 385 func (d *DriverMock) DeleteVirtualSwitch(switchName string) error { 386 d.DeleteVirtualSwitch_Called = true 387 d.DeleteVirtualSwitch_SwitchName = switchName 388 return d.DeleteVirtualSwitch_Err 389 } 390 391 func (d *DriverMock) CreateVirtualSwitch(switchName string, switchType string) (bool, error) { 392 d.CreateVirtualSwitch_Called = true 393 d.CreateVirtualSwitch_SwitchName = switchName 394 d.CreateVirtualSwitch_SwitchType = switchType 395 return d.CreateVirtualSwitch_Return, d.CreateVirtualSwitch_Err 396 } 397 398 func (d *DriverMock) AddVirtualMachineHardDrive(vmName string, vhdFile string, vhdName string, 399 vhdSizeBytes int64, vhdDiskBlockSize int64, controllerType string) error { 400 d.AddVirtualMachineHardDrive_Called = true 401 d.AddVirtualMachineHardDrive_VmName = vmName 402 d.AddVirtualMachineHardDrive_VhdFile = vhdFile 403 d.AddVirtualMachineHardDrive_VhdName = vhdName 404 d.AddVirtualMachineHardDrive_VhdSizeBytes = vhdSizeBytes 405 d.AddVirtualMachineHardDrive_VhdSizeBytes = vhdDiskBlockSize 406 d.AddVirtualMachineHardDrive_ControllerType = controllerType 407 return d.AddVirtualMachineHardDrive_Err 408 } 409 410 func (d *DriverMock) CreateVirtualMachine(vmName string, path string, harddrivePath string, 411 ram int64, diskSize int64, diskBlockSize int64, switchName string, generation uint, 412 diffDisks bool, fixedVHD bool) error { 413 d.CreateVirtualMachine_Called = true 414 d.CreateVirtualMachine_VmName = vmName 415 d.CreateVirtualMachine_Path = path 416 d.CreateVirtualMachine_HarddrivePath = harddrivePath 417 d.CreateVirtualMachine_Ram = ram 418 d.CreateVirtualMachine_DiskSize = diskSize 419 d.CreateVirtualMachine_DiskBlockSize = diskBlockSize 420 d.CreateVirtualMachine_SwitchName = switchName 421 d.CreateVirtualMachine_Generation = generation 422 d.CreateVirtualMachine_DifferentialDisk = diffDisks 423 return d.CreateVirtualMachine_Err 424 } 425 426 func (d *DriverMock) CloneVirtualMachine(cloneFromVmcxPath string, cloneFromVmName string, 427 cloneFromSnapshotName string, cloneAllSnapshots bool, vmName string, path string, 428 harddrivePath string, ram int64, switchName string) error { 429 d.CloneVirtualMachine_Called = true 430 d.CloneVirtualMachine_CloneFromVmcxPath = cloneFromVmcxPath 431 d.CloneVirtualMachine_CloneFromVmName = cloneFromVmName 432 d.CloneVirtualMachine_CloneFromSnapshotName = cloneFromSnapshotName 433 d.CloneVirtualMachine_CloneAllSnapshots = cloneAllSnapshots 434 d.CloneVirtualMachine_VmName = vmName 435 d.CloneVirtualMachine_Path = path 436 d.CloneVirtualMachine_HarddrivePath = harddrivePath 437 d.CloneVirtualMachine_Ram = ram 438 d.CloneVirtualMachine_SwitchName = switchName 439 return d.CloneVirtualMachine_Err 440 } 441 442 func (d *DriverMock) DeleteVirtualMachine(vmName string) error { 443 d.DeleteVirtualMachine_Called = true 444 d.DeleteVirtualMachine_VmName = vmName 445 return d.DeleteVirtualMachine_Err 446 } 447 448 func (d *DriverMock) SetVirtualMachineCpuCount(vmName string, cpu uint) error { 449 d.SetVirtualMachineCpuCount_Called = true 450 d.SetVirtualMachineCpuCount_VmName = vmName 451 d.SetVirtualMachineCpuCount_Cpu = cpu 452 return d.SetVirtualMachineCpuCount_Err 453 } 454 455 func (d *DriverMock) SetVirtualMachineMacSpoofing(vmName string, enable bool) error { 456 d.SetVirtualMachineMacSpoofing_Called = true 457 d.SetVirtualMachineMacSpoofing_VmName = vmName 458 d.SetVirtualMachineMacSpoofing_Enable = enable 459 return d.SetVirtualMachineMacSpoofing_Err 460 } 461 462 func (d *DriverMock) SetVirtualMachineDynamicMemory(vmName string, enable bool) error { 463 d.SetVirtualMachineDynamicMemory_Called = true 464 d.SetVirtualMachineDynamicMemory_VmName = vmName 465 d.SetVirtualMachineDynamicMemory_Enable = enable 466 return d.SetVirtualMachineDynamicMemory_Err 467 } 468 469 func (d *DriverMock) SetVirtualMachineSecureBoot(vmName string, enable bool, templateName string) error { 470 d.SetVirtualMachineSecureBoot_Called = true 471 d.SetVirtualMachineSecureBoot_VmName = vmName 472 d.SetVirtualMachineSecureBoot_Enable = enable 473 d.SetVirtualMachineSecureBoot_TemplateName = templateName 474 return d.SetVirtualMachineSecureBoot_Err 475 } 476 477 func (d *DriverMock) SetVirtualMachineVirtualizationExtensions(vmName string, enable bool) error { 478 d.SetVirtualMachineVirtualizationExtensions_Called = true 479 d.SetVirtualMachineVirtualizationExtensions_VmName = vmName 480 d.SetVirtualMachineVirtualizationExtensions_Enable = enable 481 return d.SetVirtualMachineVirtualizationExtensions_Err 482 } 483 484 func (d *DriverMock) EnableVirtualMachineIntegrationService(vmName string, integrationServiceName string) error { 485 d.EnableVirtualMachineIntegrationService_Called = true 486 d.EnableVirtualMachineIntegrationService_VmName = vmName 487 d.EnableVirtualMachineIntegrationService_IntegrationServiceName = integrationServiceName 488 return d.EnableVirtualMachineIntegrationService_Err 489 } 490 491 func (d *DriverMock) ExportVirtualMachine(vmName string, path string) error { 492 d.ExportVirtualMachine_Called = true 493 d.ExportVirtualMachine_VmName = vmName 494 d.ExportVirtualMachine_Path = path 495 return d.ExportVirtualMachine_Err 496 } 497 498 func (d *DriverMock) PreserveLegacyExportBehaviour(srcPath string, dstPath string) error { 499 d.PreserveLegacyExportBehaviour_Called = true 500 d.PreserveLegacyExportBehaviour_SrcPath = srcPath 501 d.PreserveLegacyExportBehaviour_DstPath = dstPath 502 return d.PreserveLegacyExportBehaviour_Err 503 } 504 505 func (d *DriverMock) MoveCreatedVHDsToOutputDir(srcPath string, dstPath string) error { 506 d.MoveCreatedVHDsToOutputDir_Called = true 507 d.MoveCreatedVHDsToOutputDir_SrcPath = srcPath 508 d.MoveCreatedVHDsToOutputDir_DstPath = dstPath 509 return d.MoveCreatedVHDsToOutputDir_Err 510 } 511 512 func (d *DriverMock) CompactDisks(path string) (result string, err error) { 513 d.CompactDisks_Called = true 514 d.CompactDisks_Path = path 515 d.CompactDisks_Result = "Mock compact result msg: mockdisk.vhdx. Disk size reduced by 20%" 516 return d.CompactDisks_Result, d.CompactDisks_Err 517 } 518 519 func (d *DriverMock) RestartVirtualMachine(vmName string) error { 520 d.RestartVirtualMachine_Called = true 521 d.RestartVirtualMachine_VmName = vmName 522 return d.RestartVirtualMachine_Err 523 } 524 525 func (d *DriverMock) CreateDvdDrive(vmName string, isoPath string, generation uint) (uint, uint, error) { 526 d.CreateDvdDrive_Called = true 527 d.CreateDvdDrive_VmName = vmName 528 d.CreateDvdDrive_IsoPath = isoPath 529 d.CreateDvdDrive_Generation = generation 530 return d.CreateDvdDrive_ControllerNumber, d.CreateDvdDrive_ControllerLocation, d.CreateDvdDrive_Err 531 } 532 533 func (d *DriverMock) MountDvdDrive(vmName string, path string, controllerNumber uint, 534 controllerLocation uint) error { 535 d.MountDvdDrive_Called = true 536 d.MountDvdDrive_VmName = vmName 537 d.MountDvdDrive_Path = path 538 d.MountDvdDrive_ControllerNumber = controllerNumber 539 d.MountDvdDrive_ControllerLocation = controllerLocation 540 return d.MountDvdDrive_Err 541 } 542 543 func (d *DriverMock) SetBootDvdDrive(vmName string, controllerNumber uint, controllerLocation uint, 544 generation uint) error { 545 d.SetBootDvdDrive_Called = true 546 d.SetBootDvdDrive_VmName = vmName 547 d.SetBootDvdDrive_ControllerNumber = controllerNumber 548 d.SetBootDvdDrive_ControllerLocation = controllerLocation 549 d.SetBootDvdDrive_Generation = generation 550 return d.SetBootDvdDrive_Err 551 } 552 553 func (d *DriverMock) UnmountDvdDrive(vmName string, controllerNumber uint, controllerLocation uint) error { 554 d.UnmountDvdDrive_Called = true 555 d.UnmountDvdDrive_VmName = vmName 556 d.UnmountDvdDrive_ControllerNumber = controllerNumber 557 d.UnmountDvdDrive_ControllerLocation = controllerLocation 558 return d.UnmountDvdDrive_Err 559 } 560 561 func (d *DriverMock) DeleteDvdDrive(vmName string, controllerNumber uint, controllerLocation uint) error { 562 d.DeleteDvdDrive_Called = true 563 d.DeleteDvdDrive_VmName = vmName 564 d.DeleteDvdDrive_ControllerNumber = controllerNumber 565 d.DeleteDvdDrive_ControllerLocation = controllerLocation 566 return d.DeleteDvdDrive_Err 567 } 568 569 func (d *DriverMock) MountFloppyDrive(vmName string, path string) error { 570 d.MountFloppyDrive_Called = true 571 d.MountFloppyDrive_VmName = vmName 572 d.MountFloppyDrive_Path = path 573 return d.MountFloppyDrive_Err 574 } 575 576 func (d *DriverMock) UnmountFloppyDrive(vmName string) error { 577 d.UnmountFloppyDrive_Called = true 578 d.UnmountFloppyDrive_VmName = vmName 579 return d.UnmountFloppyDrive_Err 580 } 581 582 func (d *DriverMock) Connect(vmName string) (context.CancelFunc, error) { 583 d.Connect_Called = true 584 d.Connect_VmName = vmName 585 return d.Connect_Cancel, d.Connect_Err 586 } 587 588 func (d *DriverMock) Disconnect(cancel context.CancelFunc) { 589 d.Disconnect_Called = true 590 d.Disconnect_Cancel = cancel 591 }