From daddb65da8f746df9e17c7a78eab9dfb06983a93 Mon Sep 17 00:00:00 2001
From: Megan Marsh <megan@hashicorp.com>
Date: Thu, 13 Jun 2019 14:29:22 -0700
Subject: [PATCH] add tests

---
 builder/hyperv/common/step_create_vm_test.go | 58 ++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 builder/hyperv/common/step_create_vm_test.go

diff --git a/builder/hyperv/common/step_create_vm_test.go b/builder/hyperv/common/step_create_vm_test.go
new file mode 100644
index 000000000..93725d9cb
--- /dev/null
+++ b/builder/hyperv/common/step_create_vm_test.go
@@ -0,0 +1,58 @@
+package common
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/packer/helper/multistep"
+)
+
+func TestStepCreateVM_impl(t *testing.T) {
+	var _ multistep.Step = new(StepCreateVM)
+}
+
+func TestStepCreateVM(t *testing.T) {
+	state := testState(t)
+	step := new(StepCreateVM)
+
+	step.VMName = "test-VM-Name"
+	driver := state.Get("driver").(*DriverMock)
+
+	// Test the run
+	if action := step.Run(context.Background(), state); action != multistep.ActionContinue {
+		t.Fatalf("Bad action: %v", action)
+	}
+	if _, ok := state.GetOk("error"); ok {
+		t.Fatal("Should NOT have error")
+	}
+
+	// Test the driver
+	if !driver.CheckVMName_Called {
+		t.Fatal("Should have called CheckVMName")
+	}
+}
+
+func TestStepCreateVM_CheckVMNameErr(t *testing.T) {
+	state := testState(t)
+	step := new(StepCreateVM)
+
+	step.VMName = "test-VM-Name"
+	driver := state.Get("driver").(*DriverMock)
+	driver.CheckVMName_Err = fmt.Errorf("A virtual machine with the name is already" +
+		" defined in Hyper-V. To avoid a name collision, please set your " +
+		"vm_name to a unique value")
+
+	// Test the run
+	if action := step.Run(context.Background(), state); action != multistep.ActionHalt {
+		t.Fatalf("Bad action: %v", action)
+	}
+	if _, ok := state.GetOk("error"); !ok {
+		t.Fatal("Should have error")
+	}
+
+	// Test the driver
+	if !driver.CheckVMName_Called {
+		t.Fatal("Should have called CheckVMName")
+	}
+}