From 80ebcea5e1cfc0ba29816283482c7aa71ab52859 Mon Sep 17 00:00:00 2001
From: DanHam <DanHam@users.noreply.github.com>
Date: Thu, 19 Jan 2017 13:33:59 +0000
Subject: [PATCH] Mirror cleaner test code back from windows-shell to
 powershell

---
 provisioner/powershell/provisioner_test.go | 40 +++++++---------------
 1 file changed, 13 insertions(+), 27 deletions(-)

diff --git a/provisioner/powershell/provisioner_test.go b/provisioner/powershell/provisioner_test.go
index a3c29ed24..e0a44a008 100644
--- a/provisioner/powershell/provisioner_test.go
+++ b/provisioner/powershell/provisioner_test.go
@@ -310,37 +310,23 @@ func TestProvisionerQuote_EnvironmentVars(t *testing.T) {
 		"keyfive='value=five'",
 		"keysix='=six'",
 	}
-	p := new(Provisioner)
-	p.Prepare(config)
 
-	expectedValue := "keyone=valueone"
-	if p.config.Vars[0] != expectedValue {
-		t.Fatalf("%s should be equal to %s", p.config.Vars[0], expectedValue)
-	}
-
-	expectedValue = "keytwo=value\ntwo"
-	if p.config.Vars[1] != expectedValue {
-		t.Fatalf("%s should be equal to %s", p.config.Vars[1], expectedValue)
-	}
-
-	expectedValue = "keythree='valuethree'"
-	if p.config.Vars[2] != expectedValue {
-		t.Fatalf("%s should be equal to %s", p.config.Vars[2], expectedValue)
-	}
-
-	expectedValue = "keyfour='value\nfour'"
-	if p.config.Vars[3] != expectedValue {
-		t.Fatalf("%s should be equal to %s", p.config.Vars[3], expectedValue)
+	expected := []string{
+		"keyone=valueone",
+		"keytwo=value\ntwo",
+		"keythree='valuethree'",
+		"keyfour='value\nfour'",
+		"keyfive='value=five'",
+		"keysix='=six'",
 	}
 
-	expectedValue = "keyfive='value=five'"
-	if p.config.Vars[4] != expectedValue {
-		t.Fatalf("%s should be equal to %s", p.config.Vars[4], expectedValue)
-	}
+	p := new(Provisioner)
+	p.Prepare(config)
 
-	expectedValue = "keysix='=six'"
-	if p.config.Vars[5] != expectedValue {
-		t.Fatalf("%s should be equal to %s", p.config.Vars[5], expectedValue)
+	for i, expectedValue := range expected {
+		if p.config.Vars[i] != expectedValue {
+			t.Fatalf("%s should be equal to %s", p.config.Vars[i], expectedValue)
+		}
 	}
 }