fix mocks

This commit is contained in:
Megan Marsh 2019-10-31 15:36:22 -07:00
parent 65ad3105b3
commit 6f54468dd5
1 changed files with 10 additions and 0 deletions

View File

@ -13,6 +13,10 @@ type DriverMock struct {
CreateSCSIControllerController string
CreateSCSIControllerErr error
CreateNVMeControllerVM string
CreateNVMeControllerController string
CreateNVMeControllerErr error
DeleteCalled bool
DeleteName string
DeleteErr error
@ -70,6 +74,12 @@ func (d *DriverMock) CreateSCSIController(vm string, controller string) error {
return d.CreateSCSIControllerErr
}
func (d *DriverMock) CreateNVMeController(vm string, controller string, portcount int) error {
d.CreateNVMeControllerVM = vm
d.CreateNVMeControllerController = vm
return d.CreateNVMeControllerErr
}
func (d *DriverMock) Delete(name string) error {
d.DeleteCalled = true
d.DeleteName = name