From 258804106bff27a6b9d8fd8efed727fb2ce289f4 Mon Sep 17 00:00:00 2001
From: Ali Rizvi-Santiago <arizvisa@gmail.com>
Date: Thu, 13 Apr 2017 19:19:43 -0500
Subject: [PATCH] Added missing GetVmwareDriver() method to VMware Builder's
 DriverMock.

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

diff --git a/builder/vmware/common/driver_mock.go b/builder/vmware/common/driver_mock.go
index 562eab382..8b533db88 100644
--- a/builder/vmware/common/driver_mock.go
+++ b/builder/vmware/common/driver_mock.go
@@ -206,3 +206,20 @@ func (d *DriverMock) Verify() error {
 	d.VerifyCalled = true
 	return d.VerifyErr
 }
+
+func (d *DriverMock) GetVmwareDriver() VmwareDriver {
+	var state VmwareDriver
+	state.DhcpLeasesPath = func(string) string {
+		return "/path/to/dhcp.leases"
+	}
+	state.DhcpConfPath = func(string) string {
+		return "/path/to/dhcp.conf"
+	}
+	state.VmnetnatConfPath = func(string) string {
+		return "/path/to/vmnetnat.conf"
+	}
+	state.NetmapConfPath = func() string {
+		return "/path/to/netmap.conf"
+	}
+	return state
+}