From 6f54468dd589d6e5fd336671fe61a034f42550b4 Mon Sep 17 00:00:00 2001
From: Megan Marsh <megan@hashicorp.com>
Date: Thu, 31 Oct 2019 15:36:22 -0700
Subject: [PATCH] fix mocks

---
 builder/virtualbox/common/driver_mock.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/builder/virtualbox/common/driver_mock.go b/builder/virtualbox/common/driver_mock.go
index 68d57fa64..6ff9e80b8 100644
--- a/builder/virtualbox/common/driver_mock.go
+++ b/builder/virtualbox/common/driver_mock.go
@@ -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