From ecf45d47404d0ac106da7ee90f7aa18d4c8f3c32 Mon Sep 17 00:00:00 2001
From: Ali Rizvi-Santiago <arizvisa@gmail.com>
Date: Mon, 3 Dec 2018 21:11:40 -0600
Subject: [PATCH] Added tests for packer.ExpandUser in
 packer/config_file_test.go.

---
 packer/config_file_test.go | 161 +++++++++++++++++++++++++++++++++++++
 1 file changed, 161 insertions(+)
 create mode 100644 packer/config_file_test.go

diff --git a/packer/config_file_test.go b/packer/config_file_test.go
new file mode 100644
index 000000000..ba275793b
--- /dev/null
+++ b/packer/config_file_test.go
@@ -0,0 +1,161 @@
+package packer
+
+import (
+	"fmt"
+	"os/user"
+	"path/filepath"
+	"runtime"
+	"testing"
+)
+
+// Depending on the platform, find a valid username to use
+func platform_user() string {
+	// XXX: We make an assumption here that there's an Administrator user
+	//      on the windows platform, whereas the correct way is to use
+	//		the api or to scrape `net user`.
+	if runtime.GOOS == "windows" {
+		return "Administrator"
+	}
+	return "root"
+}
+
+func homedir_current() (string, error) {
+	u, err := user.Current()
+	if err != nil {
+		return "", err
+	}
+
+	return u.HomeDir, nil
+}
+
+func homedir_user(username string) (string, error) {
+	u, err := user.Lookup(username)
+	if err != nil {
+		return "", err
+	}
+
+	return u.HomeDir, nil
+}
+
+// Begin the actual tests and stuff
+func TestExpandUser_Empty(t *testing.T) {
+	var path, expected string
+
+	// Try an invalid user
+	path, err := ExpandUser("~invalid-user-that-should-not-exist")
+	if err == nil {
+		t.Fatalf("expected failure")
+	}
+
+	// Try an empty string
+	expected = ""
+	if path, err = ExpandUser(""); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+
+	// Try an absolute path
+	expected = "/etc/shadow"
+	if path, err = ExpandUser("/etc/shadow"); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+
+	// Try a relative path
+	expected = "tmp/foo"
+	if path, err = ExpandUser("tmp/foo"); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+}
+
+func TestExpandUser_Current(t *testing.T) {
+	var path, expected string
+
+	// Grab the current user's home directory to verify ExpandUser works
+	homedir, err := homedir_current()
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	// Try just a tilde
+	expected = homedir
+	if path, err = ExpandUser("~"); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+
+	// Try as a directory
+	expected = filepath.Join(homedir, "")
+	if path, err = ExpandUser("~/"); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+
+	// Try as a file
+	expected = filepath.Join(homedir, "foo")
+	if path, err = ExpandUser("~/foo"); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+}
+
+func TestExpandUser_User(t *testing.T) {
+	var path, expected string
+
+	username := platform_user()
+
+	// Grab the current user's home directory to verify ExpandUser works
+	homedir, err := homedir_user(username)
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	// Try just a tilde
+	expected = homedir
+	if path, err = ExpandUser(fmt.Sprintf("~%s", username)); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+
+	// Try as a directory
+	expected = filepath.Join(homedir, "")
+	if path, err = ExpandUser(fmt.Sprintf("~%s/", username)); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+
+	// Try as a file
+	expected = filepath.Join(homedir, "foo")
+	if path, err = ExpandUser(fmt.Sprintf("~%s/foo", username)); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	if path != expected {
+		t.Fatalf("err: %v != %v", path, expected)
+	}
+}