From c4cee75b9372a62bee0d7904e656253b488e99db Mon Sep 17 00:00:00 2001
From: Nathan Hartwell <njhartwell@gmail.com>
Date: Wed, 13 May 2015 08:32:09 -0500
Subject: [PATCH] Adding tests for disable sudo

---
 .../salt-masterless/provisioner_test.go       | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/provisioner/salt-masterless/provisioner_test.go b/provisioner/salt-masterless/provisioner_test.go
index c59ee9463..c15053b21 100644
--- a/provisioner/salt-masterless/provisioner_test.go
+++ b/provisioner/salt-masterless/provisioner_test.go
@@ -103,3 +103,28 @@ func TestProvisionerPrepare_LocalPillarRoots(t *testing.T) {
 		t.Fatalf("err: %s", err)
 	}
 }
+
+func TestProvisionerSudo(t *testing.T) {
+	var p Provisioner
+	config := testConfig()
+
+	err := p.Prepare(config)
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+    withSudo := p.sudo("echo hello")
+    if withSudo != "sudo echo hello" {
+        t.Fatalf("sudo command not generated correctly")
+    }
+
+    config["disable_sudo"] = true
+	err = p.Prepare(config)
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+    withoutSudo := p.sudo("echo hello")
+    if withoutSudo != "echo hello" {
+        t.Fatalf("sudo-less command not generated correctly")
+    }
+}