tests(settings): adding tests for FindHome() and SetHome()

settings/home_unix_test.go

@@ -0,0 +1,26 @@
+// +build linux darwin
+
+package settings
+
+import (
+	"os"
+	"testing"
+
+	"github.com/arschles/assert"
+)
+
+// TestFindHome ensures the correct home directory is returned by FindHome().
+func TestFindHome(t *testing.T) {
+	expectedHomeDir := "/d/e/f"
+	os.Setenv("HOME", expectedHomeDir)
+
+	assert.Equal(t, FindHome(), expectedHomeDir, "output")
+}
+
+// TestSetHome ensures the correct env vars are set when SetHome() is called.
+func TestSetHome(t *testing.T) {
+	homeDir := "/a/b/c"
+	SetHome(homeDir)
+
+	assert.Equal(t, os.Getenv("HOME"), homeDir, "output")
+}

settings/home_windows_test.go

@@ -0,0 +1,29 @@
+// +build windows
+
+package settings
+
+import (
+	"os"
+	"testing"
+
+	"github.com/arschles/assert"
+)
+
+// TestFindHome ensures the correct home directory is returned by FindHome().
+func TestFindHome(t *testing.T) {
+	homedrive := "C:"
+	homepath := "/a/b/c"
+	os.Setenv("HOMEDRIVE", homedrive)
+	os.Setenv("HOMEPATH", homepath)
+	assert.Equal(t, FindHome(), homedrive+homepath, "output")
+}
+
+// TestSetHome ensures the correct env vars are set when SetHome() is called.
+func TestSetHome(t *testing.T) {
+	homeDrive := "D:"
+	homePath := "/e/f/g"
+	SetHome(homeDrive + homePath)
+
+	assert.Equal(t, os.Getenv("HOMEDRIVE"), homeDrive, "output")
+	assert.Equal(t, os.Getenv("HOMEPATH"), homePath, "output")
+}

settings/settings_test.go

@@ -91,6 +91,14 @@ func TestLoadSave(t *testing.T) {
 
 	s.Client.ControllerURL = u
 
+	// Create a tempdir and set as HOME.
+	dir, err := ioutil.TempDir("", "deishome")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(dir)
+	SetHome(dir)
+
 	if _, err = s.Save(file); err != nil {
 		t.Fatal(err)
 	}