chore!: rename package to envctl

envctl_test.go

@@ -0,0 +1,219 @@
+package envctl
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestWith(t *testing.T) {
+	tests := []struct {
+		name string
+		env  map[string]string
+	}{
+		{
+			name: "empty",
+			env:  map[string]string{},
+		},
+		{
+			name: "new vars",
+			env: map[string]string{
+				"CC_TEST_INSIDE": "new var is here",
+				"CC_TEST_FOO":    "bar",
+			},
+		},
+		{
+			name: "set existing",
+			env: map[string]string{
+				"CC_TEST_INSIDE":  "new var is here",
+				"CC_TEST_OUTSIDE": "this is not the same",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			os.Setenv("CC_TEST_OUTSIDE", t.Name())
+			outside := parseEnviron(os.Environ())
+
+			With(tt.env, func() {
+				os.Setenv("CC_TEST_SET_INSIDE", t.Name()+"-manual")
+
+				for k, v := range tt.env {
+					got := os.Getenv(k)
+					assert.Equal(t, v, got)
+				}
+
+				for k, v := range outside {
+					if _, ok := tt.env[k]; !ok {
+						assert.Equal(t, v, os.Getenv(k))
+					}
+				}
+			})
+
+			_, exists := os.LookupEnv("CC_TEST_SET_INSIDE")
+			assert.Equal(t, false, exists)
+
+			for k := range tt.env {
+				if _, ok := outside[k]; !ok {
+					_, exists := os.LookupEnv(k)
+					assert.Equal(t, false, exists)
+				}
+			}
+
+			for k, v := range outside {
+				assert.Equal(t, v, os.Getenv(k))
+			}
+
+			os.Unsetenv("CC_TEST_OUTSIDE")
+		})
+	}
+}
+
+func TestWithClean(t *testing.T) {
+	tests := []struct {
+		name string
+		env  map[string]string
+	}{
+		{
+			name: "empty",
+			env:  map[string]string{},
+		},
+		{
+			name: "new vars",
+			env: map[string]string{
+				"CC_TEST_INSIDE": "new var is here",
+				"CC_TEST_FOO":    "bar",
+			},
+		},
+		{
+			name: "set existing",
+			env: map[string]string{
+				"CC_TEST_INSIDE":  "new var is here",
+				"CC_TEST_OUTSIDE": "is is not the same",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			os.Setenv("CC_TEST_OUTSIDE", t.Name())
+			outside := parseEnviron(os.Environ())
+
+			WithClean(tt.env, func() {
+				os.Setenv("CC_TEST_SET_INSIDE", t.Name()+"-manual")
+
+				for k, v := range tt.env {
+					got := os.Getenv(k)
+					assert.Equal(t, v, got)
+				}
+
+				for k := range outside {
+					if _, ok := tt.env[k]; !ok {
+						_, exists := os.LookupEnv(k)
+						assert.Equal(t, false, exists)
+					}
+				}
+			})
+
+			_, exists := os.LookupEnv("CC_TEST_SET_INSIDE")
+			assert.Equal(t, false, exists)
+
+			for k := range tt.env {
+				if _, ok := outside[k]; !ok {
+					_, exists := os.LookupEnv(k)
+					assert.Equal(t, false, exists)
+				}
+			}
+
+			for k, v := range outside {
+				assert.Equal(t, v, os.Getenv(k))
+			}
+
+			os.Unsetenv("CC_TEST_OUTSIDE")
+		})
+	}
+}
+
+func Test_parseEnviron(t *testing.T) {
+	tests := []struct {
+		name string
+		vars []string
+		want map[string]string
+	}{
+		{
+			name: "empty",
+			vars: []string{},
+			want: map[string]string{},
+		},
+		{
+			name: "various",
+			vars: []string{
+				"USER=john",
+				"NAME=John Doe",
+				"SHELL=/bin/bash",
+				"GOPRIVATE=",
+				"LESS= -R",
+				"LESSOPEN=| src-hilite-lesspipe.sh %s",
+				"X=11",
+				"TAGS=foo=bar,hello=world",
+				"_=go",
+				"INVALID-var",
+			},
+			want: map[string]string{
+				"USER":      "john",
+				"NAME":      "John Doe",
+				"SHELL":     "/bin/bash",
+				"GOPRIVATE": "",
+				"LESS":      " -R",
+				"LESSOPEN":  "| src-hilite-lesspipe.sh %s",
+				"X":         "11",
+				"TAGS":      "foo=bar,hello=world",
+				"_":         "go",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := parseEnviron(tt.vars)
+
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func Benchmark_parseEnviron(b *testing.B) {
+	env := []string{
+		`EDITOR=emacsclient-wrapper`,
+		`EMACS=/Applications/Emacs.app/Contents/MacOS/Emacs`,
+		`GEM_EDITOR=emacsclient-wrapper`,
+		`GOPATH=/Users/jimeh/.go`,
+		`HOME=/Users/jimeh`,
+		`HOMEBREW_NO_ANALYTICS=1`,
+		`LANG=en_US.UTF-8`,
+		`LC_ALL=en_US.UTF-8`,
+		`LC_TERMINAL=iTerm2`,
+		`LC_TERMINAL_VERSION=3.4.3`,
+		`LESS= -R`,
+		`LESSOPEN=| src-hilite-lesspipe.sh %s`,
+		`NODENV_SHELL=zsh`,
+		`PWD=/Users/jimeh/Projects/envctl`,
+		`PYENV_SHELL=zsh`,
+		`RBENV_SHELL=zsh`,
+		`TERM=screen-256color`,
+		`TERM_PROGRAM=iTerm.app`,
+		`TERM_PROGRAM_VERSION=3.4.3`,
+		`TMPDIR=/tmp/user-jimeh`,
+		`TMUX=/private/tmp/tmux-501/default,4148,2`,
+		`TMUX_PANE=%29`,
+		`TMUX_PLUGIN_MANAGER_PATH=/Users/jimeh/.tmux/plugins/`,
+		`USER=jimeh`,
+		`ZPFX=/Users/jimeh/.local/zsh/zinit/polaris`,
+		`ZSH_CACHE_DIR=/Users/jimeh/.cache/zinit`,
+		`_=/usr/bin/env`,
+		`__CFBundleIdentifier=com.googlecode.iterm2`,
+	}
+
+	for i := 0; i < b.N; i++ {
+		parseEnviron(env)
+	}
+}