github.com/StackPointCloud/packer@v0.10.2-0.20180716202532-b28098e0f79b/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_VhdPath string 128 CreateVirtualMachine_Ram int64 129 CreateVirtualMachine_DiskSize int64 130 CreateVirtualMachine_DiskBlockSize int64 131 CreateVirtualMachine_SwitchName string 132 CreateVirtualMachine_Generation uint 133 CreateVirtualMachine_DifferentialDisk bool 134 CreateVirtualMachine_FixedVHD bool 135 CreateVirtualMachine_Err error 136 137 CloneVirtualMachine_Called bool 138 CloneVirtualMachine_CloneFromVmxcPath string 139 CloneVirtualMachine_CloneFromVmName string 140 CloneVirtualMachine_CloneFromSnapshotName string 141 CloneVirtualMachine_CloneAllSnapshots bool 142 CloneVirtualMachine_VmName string 143 CloneVirtualMachine_Path string 144 CloneVirtualMachine_HarddrivePath string 145 CloneVirtualMachine_Ram int64 146 CloneVirtualMachine_SwitchName string 147 CloneVirtualMachine_Err error 148 149 DeleteVirtualMachine_Called bool 150 DeleteVirtualMachine_VmName string 151 DeleteVirtualMachine_Err error 152 153 SetVirtualMachineCpuCount_Called bool 154 SetVirtualMachineCpuCount_VmName string 155 SetVirtualMachineCpuCount_Cpu uint 156 SetVirtualMachineCpuCount_Err error 157 158 SetVirtualMachineMacSpoofing_Called bool 159 SetVirtualMachineMacSpoofing_VmName string 160 SetVirtualMachineMacSpoofing_Enable bool 161 SetVirtualMachineMacSpoofing_Err error 162 163 SetVirtualMachineDynamicMemory_Called bool 164 SetVirtualMachineDynamicMemory_VmName string 165 SetVirtualMachineDynamicMemory_Enable bool 166 SetVirtualMachineDynamicMemory_Err error 167 168 SetVirtualMachineSecureBoot_Called bool 169 SetVirtualMachineSecureBoot_VmName string 170 SetVirtualMachineSecureBoot_TemplateName string 171 SetVirtualMachineSecureBoot_Enable bool 172 SetVirtualMachineSecureBoot_Err error 173 174 SetVirtualMachineVirtualizationExtensions_Called bool 175 SetVirtualMachineVirtualizationExtensions_VmName string 176 SetVirtualMachineVirtualizationExtensions_Enable bool 177 SetVirtualMachineVirtualizationExtensions_Err error 178 179 EnableVirtualMachineIntegrationService_Called bool 180 EnableVirtualMachineIntegrationService_VmName string 181 EnableVirtualMachineIntegrationService_IntegrationServiceName string 182 EnableVirtualMachineIntegrationService_Err error 183 184 ExportVirtualMachine_Called bool 185 ExportVirtualMachine_VmName string 186 ExportVirtualMachine_Path string 187 ExportVirtualMachine_Err error 188 189 CompactDisks_Called bool 190 CompactDisks_ExpPath string 191 CompactDisks_VhdDir string 192 CompactDisks_Err error 193 194 CopyExportedVirtualMachine_Called bool 195 CopyExportedVirtualMachine_ExpPath string 196 CopyExportedVirtualMachine_OutputPath string 197 CopyExportedVirtualMachine_VhdDir string 198 CopyExportedVirtualMachine_VmDir string 199 CopyExportedVirtualMachine_Err error 200 201 RestartVirtualMachine_Called bool 202 RestartVirtualMachine_VmName string 203 RestartVirtualMachine_Err error 204 205 CreateDvdDrive_Called bool 206 CreateDvdDrive_VmName string 207 CreateDvdDrive_IsoPath string 208 CreateDvdDrive_Generation uint 209 CreateDvdDrive_ControllerNumber uint 210 CreateDvdDrive_ControllerLocation uint 211 CreateDvdDrive_Err error 212 213 MountDvdDrive_Called bool 214 MountDvdDrive_VmName string 215 MountDvdDrive_Path string 216 MountDvdDrive_ControllerNumber uint 217 MountDvdDrive_ControllerLocation uint 218 MountDvdDrive_Err error 219 220 SetBootDvdDrive_Called bool 221 SetBootDvdDrive_VmName string 222 SetBootDvdDrive_ControllerNumber uint 223 SetBootDvdDrive_ControllerLocation uint 224 SetBootDvdDrive_Generation uint 225 SetBootDvdDrive_Err error 226 227 UnmountDvdDrive_Called bool 228 UnmountDvdDrive_VmName string 229 UnmountDvdDrive_ControllerNumber uint 230 UnmountDvdDrive_ControllerLocation uint 231 UnmountDvdDrive_Err error 232 233 DeleteDvdDrive_Called bool 234 DeleteDvdDrive_VmName string 235 DeleteDvdDrive_ControllerNumber uint 236 DeleteDvdDrive_ControllerLocation uint 237 DeleteDvdDrive_Err error 238 239 MountFloppyDrive_Called bool 240 MountFloppyDrive_VmName string 241 MountFloppyDrive_Path string 242 MountFloppyDrive_Err error 243 244 UnmountFloppyDrive_Called bool 245 UnmountFloppyDrive_VmName string 246 UnmountFloppyDrive_Err error 247 248 Connect_Called bool 249 Connect_VmName string 250 Connect_Cancel context.CancelFunc 251 Connect_Err error 252 253 Disconnect_Called bool 254 Disconnect_Cancel context.CancelFunc 255 } 256 257 func (d *DriverMock) IsRunning(vmName string) (bool, error) { 258 d.IsRunning_Called = true 259 d.IsRunning_VmName = vmName 260 return d.IsRunning_Return, d.IsRunning_Err 261 } 262 263 func (d *DriverMock) IsOff(vmName string) (bool, error) { 264 d.IsOff_Called = true 265 d.IsOff_VmName = vmName 266 return d.IsOff_Return, d.IsOff_Err 267 } 268 269 func (d *DriverMock) Uptime(vmName string) (uint64, error) { 270 d.Uptime_Called = true 271 d.Uptime_VmName = vmName 272 return d.Uptime_Return, d.Uptime_Err 273 } 274 275 func (d *DriverMock) Start(vmName string) error { 276 d.Start_Called = true 277 d.Start_VmName = vmName 278 return d.Start_Err 279 } 280 281 func (d *DriverMock) Stop(vmName string) error { 282 d.Stop_Called = true 283 d.Stop_VmName = vmName 284 return d.Stop_Err 285 } 286 287 func (d *DriverMock) Verify() error { 288 d.Verify_Called = true 289 return d.Verify_Err 290 } 291 292 func (d *DriverMock) Mac(vmName string) (string, error) { 293 d.Mac_Called = true 294 d.Mac_VmName = vmName 295 return d.Mac_Return, d.Mac_Err 296 } 297 298 func (d *DriverMock) IpAddress(mac string) (string, error) { 299 d.IpAddress_Called = true 300 d.IpAddress_Mac = mac 301 return d.IpAddress_Return, d.IpAddress_Err 302 } 303 304 func (d *DriverMock) GetHostName(ip string) (string, error) { 305 d.GetHostName_Called = true 306 d.GetHostName_Ip = ip 307 return d.GetHostName_Return, d.GetHostName_Err 308 } 309 310 func (d *DriverMock) GetVirtualMachineGeneration(vmName string) (uint, error) { 311 d.GetVirtualMachineGeneration_Called = true 312 d.GetVirtualMachineGeneration_VmName = vmName 313 return d.GetVirtualMachineGeneration_Return, d.GetVirtualMachineGeneration_Err 314 } 315 316 func (d *DriverMock) GetHostAdapterIpAddressForSwitch(switchName string) (string, error) { 317 d.GetHostAdapterIpAddressForSwitch_Called = true 318 d.GetHostAdapterIpAddressForSwitch_SwitchName = switchName 319 return d.GetHostAdapterIpAddressForSwitch_Return, d.GetHostAdapterIpAddressForSwitch_Err 320 } 321 322 func (d *DriverMock) TypeScanCodes(vmName string, scanCodes string) error { 323 d.TypeScanCodes_Called = true 324 d.TypeScanCodes_VmName = vmName 325 d.TypeScanCodes_ScanCodes = scanCodes 326 return d.TypeScanCodes_Err 327 } 328 329 func (d *DriverMock) GetVirtualMachineNetworkAdapterAddress(vmName string) (string, error) { 330 d.GetVirtualMachineNetworkAdapterAddress_Called = true 331 d.GetVirtualMachineNetworkAdapterAddress_VmName = vmName 332 return d.GetVirtualMachineNetworkAdapterAddress_Return, d.GetVirtualMachineNetworkAdapterAddress_Err 333 } 334 335 func (d *DriverMock) SetNetworkAdapterVlanId(switchName string, vlanId string) error { 336 d.SetNetworkAdapterVlanId_Called = true 337 d.SetNetworkAdapterVlanId_SwitchName = switchName 338 d.SetNetworkAdapterVlanId_VlanId = vlanId 339 return d.SetNetworkAdapterVlanId_Err 340 } 341 342 func (d *DriverMock) SetVmNetworkAdapterMacAddress(vmName string, mac string) error { 343 d.SetVmNetworkAdapterMacAddress_Called = true 344 d.SetVmNetworkAdapterMacAddress_VmName = vmName 345 d.SetVmNetworkAdapterMacAddress_Mac = mac 346 return d.SetVmNetworkAdapterMacAddress_Err 347 } 348 349 func (d *DriverMock) SetVirtualMachineVlanId(vmName string, vlanId string) error { 350 d.SetVirtualMachineVlanId_Called = true 351 d.SetVirtualMachineVlanId_VmName = vmName 352 d.SetVirtualMachineVlanId_VlanId = vlanId 353 return d.SetVirtualMachineVlanId_Err 354 } 355 356 func (d *DriverMock) UntagVirtualMachineNetworkAdapterVlan(vmName string, switchName string) error { 357 d.UntagVirtualMachineNetworkAdapterVlan_Called = true 358 d.UntagVirtualMachineNetworkAdapterVlan_VmName = vmName 359 d.UntagVirtualMachineNetworkAdapterVlan_SwitchName = switchName 360 return d.UntagVirtualMachineNetworkAdapterVlan_Err 361 } 362 363 func (d *DriverMock) CreateExternalVirtualSwitch(vmName string, switchName string) error { 364 d.CreateExternalVirtualSwitch_Called = true 365 d.CreateExternalVirtualSwitch_VmName = vmName 366 d.CreateExternalVirtualSwitch_SwitchName = switchName 367 return d.CreateExternalVirtualSwitch_Err 368 } 369 370 func (d *DriverMock) GetVirtualMachineSwitchName(vmName string) (string, error) { 371 d.GetVirtualMachineSwitchName_Called = true 372 d.GetVirtualMachineSwitchName_VmName = vmName 373 return d.GetVirtualMachineSwitchName_Return, d.GetVirtualMachineSwitchName_Err 374 } 375 376 func (d *DriverMock) ConnectVirtualMachineNetworkAdapterToSwitch(vmName string, switchName string) error { 377 d.ConnectVirtualMachineNetworkAdapterToSwitch_Called = true 378 d.ConnectVirtualMachineNetworkAdapterToSwitch_VmName = vmName 379 d.ConnectVirtualMachineNetworkAdapterToSwitch_SwitchName = switchName 380 return d.ConnectVirtualMachineNetworkAdapterToSwitch_Err 381 } 382 383 func (d *DriverMock) DeleteVirtualSwitch(switchName string) error { 384 d.DeleteVirtualSwitch_Called = true 385 d.DeleteVirtualSwitch_SwitchName = switchName 386 return d.DeleteVirtualSwitch_Err 387 } 388 389 func (d *DriverMock) CreateVirtualSwitch(switchName string, switchType string) (bool, error) { 390 d.CreateVirtualSwitch_Called = true 391 d.CreateVirtualSwitch_SwitchName = switchName 392 d.CreateVirtualSwitch_SwitchType = switchType 393 return d.CreateVirtualSwitch_Return, d.CreateVirtualSwitch_Err 394 } 395 396 func (d *DriverMock) AddVirtualMachineHardDrive(vmName string, vhdFile string, vhdName string, vhdSizeBytes int64, vhdDiskBlockSize int64, controllerType string) error { 397 d.AddVirtualMachineHardDrive_Called = true 398 d.AddVirtualMachineHardDrive_VmName = vmName 399 d.AddVirtualMachineHardDrive_VhdFile = vhdFile 400 d.AddVirtualMachineHardDrive_VhdName = vhdName 401 d.AddVirtualMachineHardDrive_VhdSizeBytes = vhdSizeBytes 402 d.AddVirtualMachineHardDrive_VhdSizeBytes = vhdDiskBlockSize 403 d.AddVirtualMachineHardDrive_ControllerType = controllerType 404 return d.AddVirtualMachineHardDrive_Err 405 } 406 407 func (d *DriverMock) CreateVirtualMachine(vmName string, path string, harddrivePath string, vhdPath string, ram int64, diskSize int64, diskBlockSize int64, switchName string, generation uint, diffDisks bool, fixedVHD bool) error { 408 d.CreateVirtualMachine_Called = true 409 d.CreateVirtualMachine_VmName = vmName 410 d.CreateVirtualMachine_Path = path 411 d.CreateVirtualMachine_HarddrivePath = harddrivePath 412 d.CreateVirtualMachine_VhdPath = vhdPath 413 d.CreateVirtualMachine_Ram = ram 414 d.CreateVirtualMachine_DiskSize = diskSize 415 d.CreateVirtualMachine_DiskBlockSize = diskBlockSize 416 d.CreateVirtualMachine_SwitchName = switchName 417 d.CreateVirtualMachine_Generation = generation 418 d.CreateVirtualMachine_DifferentialDisk = diffDisks 419 return d.CreateVirtualMachine_Err 420 } 421 422 func (d *DriverMock) CloneVirtualMachine(cloneFromVmxcPath string, cloneFromVmName string, cloneFromSnapshotName string, cloneAllSnapshots bool, vmName string, path string, harddrivePath string, ram int64, switchName string) error { 423 d.CloneVirtualMachine_Called = true 424 d.CloneVirtualMachine_CloneFromVmxcPath = cloneFromVmxcPath 425 d.CloneVirtualMachine_CloneFromVmName = cloneFromVmName 426 d.CloneVirtualMachine_CloneFromSnapshotName = cloneFromSnapshotName 427 d.CloneVirtualMachine_CloneAllSnapshots = cloneAllSnapshots 428 d.CloneVirtualMachine_VmName = vmName 429 d.CloneVirtualMachine_Path = path 430 d.CloneVirtualMachine_HarddrivePath = harddrivePath 431 d.CloneVirtualMachine_Ram = ram 432 d.CloneVirtualMachine_SwitchName = switchName 433 return d.CloneVirtualMachine_Err 434 } 435 436 func (d *DriverMock) DeleteVirtualMachine(vmName string) error { 437 d.DeleteVirtualMachine_Called = true 438 d.DeleteVirtualMachine_VmName = vmName 439 return d.DeleteVirtualMachine_Err 440 } 441 442 func (d *DriverMock) SetVirtualMachineCpuCount(vmName string, cpu uint) error { 443 d.SetVirtualMachineCpuCount_Called = true 444 d.SetVirtualMachineCpuCount_VmName = vmName 445 d.SetVirtualMachineCpuCount_Cpu = cpu 446 return d.SetVirtualMachineCpuCount_Err 447 } 448 449 func (d *DriverMock) SetVirtualMachineMacSpoofing(vmName string, enable bool) error { 450 d.SetVirtualMachineMacSpoofing_Called = true 451 d.SetVirtualMachineMacSpoofing_VmName = vmName 452 d.SetVirtualMachineMacSpoofing_Enable = enable 453 return d.SetVirtualMachineMacSpoofing_Err 454 } 455 456 func (d *DriverMock) SetVirtualMachineDynamicMemory(vmName string, enable bool) error { 457 d.SetVirtualMachineDynamicMemory_Called = true 458 d.SetVirtualMachineDynamicMemory_VmName = vmName 459 d.SetVirtualMachineDynamicMemory_Enable = enable 460 return d.SetVirtualMachineDynamicMemory_Err 461 } 462 463 func (d *DriverMock) SetVirtualMachineSecureBoot(vmName string, enable bool, templateName string) error { 464 d.SetVirtualMachineSecureBoot_Called = true 465 d.SetVirtualMachineSecureBoot_VmName = vmName 466 d.SetVirtualMachineSecureBoot_Enable = enable 467 d.SetVirtualMachineSecureBoot_TemplateName = templateName 468 return d.SetVirtualMachineSecureBoot_Err 469 } 470 471 func (d *DriverMock) SetVirtualMachineVirtualizationExtensions(vmName string, enable bool) error { 472 d.SetVirtualMachineVirtualizationExtensions_Called = true 473 d.SetVirtualMachineVirtualizationExtensions_VmName = vmName 474 d.SetVirtualMachineVirtualizationExtensions_Enable = enable 475 return d.SetVirtualMachineVirtualizationExtensions_Err 476 } 477 478 func (d *DriverMock) EnableVirtualMachineIntegrationService(vmName string, integrationServiceName string) error { 479 d.EnableVirtualMachineIntegrationService_Called = true 480 d.EnableVirtualMachineIntegrationService_VmName = vmName 481 d.EnableVirtualMachineIntegrationService_IntegrationServiceName = integrationServiceName 482 return d.EnableVirtualMachineIntegrationService_Err 483 } 484 485 func (d *DriverMock) ExportVirtualMachine(vmName string, path string) error { 486 d.ExportVirtualMachine_Called = true 487 d.ExportVirtualMachine_VmName = vmName 488 d.ExportVirtualMachine_Path = path 489 return d.ExportVirtualMachine_Err 490 } 491 492 func (d *DriverMock) CompactDisks(expPath string, vhdDir string) error { 493 d.CompactDisks_Called = true 494 d.CompactDisks_ExpPath = expPath 495 d.CompactDisks_VhdDir = vhdDir 496 return d.CompactDisks_Err 497 } 498 499 func (d *DriverMock) CopyExportedVirtualMachine(expPath string, outputPath string, vhdDir string, vmDir string) error { 500 d.CopyExportedVirtualMachine_Called = true 501 d.CopyExportedVirtualMachine_ExpPath = expPath 502 d.CopyExportedVirtualMachine_OutputPath = outputPath 503 d.CopyExportedVirtualMachine_VhdDir = vhdDir 504 d.CopyExportedVirtualMachine_VmDir = vmDir 505 return d.CopyExportedVirtualMachine_Err 506 } 507 508 func (d *DriverMock) RestartVirtualMachine(vmName string) error { 509 d.RestartVirtualMachine_Called = true 510 d.RestartVirtualMachine_VmName = vmName 511 return d.RestartVirtualMachine_Err 512 } 513 514 func (d *DriverMock) CreateDvdDrive(vmName string, isoPath string, generation uint) (uint, uint, error) { 515 d.CreateDvdDrive_Called = true 516 d.CreateDvdDrive_VmName = vmName 517 d.CreateDvdDrive_IsoPath = isoPath 518 d.CreateDvdDrive_Generation = generation 519 return d.CreateDvdDrive_ControllerNumber, d.CreateDvdDrive_ControllerLocation, d.CreateDvdDrive_Err 520 } 521 522 func (d *DriverMock) MountDvdDrive(vmName string, path string, controllerNumber uint, controllerLocation uint) error { 523 d.MountDvdDrive_Called = true 524 d.MountDvdDrive_VmName = vmName 525 d.MountDvdDrive_Path = path 526 d.MountDvdDrive_ControllerNumber = controllerNumber 527 d.MountDvdDrive_ControllerLocation = controllerLocation 528 return d.MountDvdDrive_Err 529 } 530 531 func (d *DriverMock) SetBootDvdDrive(vmName string, controllerNumber uint, controllerLocation uint, generation uint) error { 532 d.SetBootDvdDrive_Called = true 533 d.SetBootDvdDrive_VmName = vmName 534 d.SetBootDvdDrive_ControllerNumber = controllerNumber 535 d.SetBootDvdDrive_ControllerLocation = controllerLocation 536 d.SetBootDvdDrive_Generation = generation 537 return d.SetBootDvdDrive_Err 538 } 539 540 func (d *DriverMock) UnmountDvdDrive(vmName string, controllerNumber uint, controllerLocation uint) error { 541 d.UnmountDvdDrive_Called = true 542 d.UnmountDvdDrive_VmName = vmName 543 d.UnmountDvdDrive_ControllerNumber = controllerNumber 544 d.UnmountDvdDrive_ControllerLocation = controllerLocation 545 return d.UnmountDvdDrive_Err 546 } 547 548 func (d *DriverMock) DeleteDvdDrive(vmName string, controllerNumber uint, controllerLocation uint) error { 549 d.DeleteDvdDrive_Called = true 550 d.DeleteDvdDrive_VmName = vmName 551 d.DeleteDvdDrive_ControllerNumber = controllerNumber 552 d.DeleteDvdDrive_ControllerLocation = controllerLocation 553 return d.DeleteDvdDrive_Err 554 } 555 556 func (d *DriverMock) MountFloppyDrive(vmName string, path string) error { 557 d.MountFloppyDrive_Called = true 558 d.MountFloppyDrive_VmName = vmName 559 d.MountFloppyDrive_Path = path 560 return d.MountFloppyDrive_Err 561 } 562 563 func (d *DriverMock) UnmountFloppyDrive(vmName string) error { 564 d.UnmountFloppyDrive_Called = true 565 d.UnmountFloppyDrive_VmName = vmName 566 return d.UnmountFloppyDrive_Err 567 } 568 569 func (d *DriverMock) Connect(vmName string) (context.CancelFunc, error) { 570 d.Connect_Called = true 571 d.Connect_VmName = vmName 572 return d.Connect_Cancel, d.Connect_Err 573 } 574 575 func (d *DriverMock) Disconnect(cancel context.CancelFunc) { 576 d.Disconnect_Called = true 577 d.Disconnect_Cancel = cancel 578 }