From 5022db8d551403ae55c479d0ebe2be0b2b4df2ee Mon Sep 17 00:00:00 2001
From: Brad Rydzewski <brad.rydzewski@gmail.com>
Date: Sat, 1 Nov 2014 11:41:02 -0700
Subject: [PATCH] updated unit tests to account for quoted env vars

---
 plugin/deploy/modulus/modulus_test.go     | 2 +-
 plugin/deploy/nodejitsu/nodejitsu_test.go | 4 ++--
 plugin/deploy/ssh_test.go                 | 8 ++++----
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/plugin/deploy/modulus/modulus_test.go b/plugin/deploy/modulus/modulus_test.go
index 59c8e6bf..b8f56ad3 100644
--- a/plugin/deploy/modulus/modulus_test.go
+++ b/plugin/deploy/modulus/modulus_test.go
@@ -40,7 +40,7 @@ func Test_Modulus(t *testing.T) {
 			}
 
 			m.Write(b)
-			g.Assert(b.String()).Equal(`export MODULUS_TOKEN=bar
+			g.Assert(b.String()).Equal(`export MODULUS_TOKEN="bar"
 [ -f /usr/bin/npm ] || echo ERROR: npm is required for modulus.io deployments
 [ -f /usr/bin/npm ] || exit 1
 [ -f /usr/bin/sudo ] || npm install -g modulus
diff --git a/plugin/deploy/nodejitsu/nodejitsu_test.go b/plugin/deploy/nodejitsu/nodejitsu_test.go
index f1424d94..9acbf8de 100644
--- a/plugin/deploy/nodejitsu/nodejitsu_test.go
+++ b/plugin/deploy/nodejitsu/nodejitsu_test.go
@@ -40,8 +40,8 @@ func Test_Modulus(t *testing.T) {
 			}
 
 			n.Write(b)
-			g.Assert(b.String()).Equal(`export username=foo
-export apiToken=bar
+			g.Assert(b.String()).Equal(`export username="foo"
+export apiToken="bar"
 [ -f /usr/bin/sudo ] || npm install -g jitsu
 [ -f /usr/bin/sudo ] && sudo npm install -g jitsu
 echo '#DRONE:6a69747375206465706c6f79'
diff --git a/plugin/deploy/ssh_test.go b/plugin/deploy/ssh_test.go
index 037d3ab3..54aa1a1c 100644
--- a/plugin/deploy/ssh_test.go
+++ b/plugin/deploy/ssh_test.go
@@ -81,7 +81,7 @@ func TestSSHOneArtifact(t *testing.T) {
 		t.Fatalf("Can't unmarshal deploy script: %s", err)
 	}
 
-	if !strings.Contains(bscr, "ARTIFACT=build.result") {
+	if !strings.Contains(bscr, `ARTIFACT="build.result"`) {
 		t.Error("Expect script to contains artifact")
 	}
 
@@ -96,7 +96,7 @@ func TestSSHMultiArtifact(t *testing.T) {
 		t.Fatalf("Can't unmarshal deploy script: %s", err)
 	}
 
-	if !strings.Contains(bscr, "ARTIFACT=${PWD##*/}.tar.gz") {
+	if !strings.Contains(bscr, `ARTIFACT="${PWD##*/}.tar.gz"`) {
 		t.Errorf("Expect script to contains artifact")
 	}
 
@@ -111,11 +111,11 @@ func TestSSHGitArchive(t *testing.T) {
 		t.Fatalf("Can't unmarshal deploy script: %s", err)
 	}
 
-	if !strings.Contains(bscr, "COMMIT=$(git rev-parse HEAD)") {
+	if !strings.Contains(bscr, `COMMIT="$(git rev-parse HEAD)"`) {
 		t.Errorf("Expect script to contains commit ref")
 	}
 
-	if !strings.Contains(bscr, "ARTIFACT=${PWD##*/}-${COMMIT}.tar.gz") {
+	if !strings.Contains(bscr, `ARTIFACT="${PWD##*/}-${COMMIT}.tar.gz"`) {
 		t.Errorf("Expect script to contains artifact")
 	}