ResiLien
/
terraform
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve readability and purpose of multi-env default test

This commit is contained in:
Seth Vargo 2015-01-22 15:34:48 -05:00
parent 038d241043
commit 0a7dea5532
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
helper/schema/schema_test.go
View File

@ -51,8 +51,9 @@ func TestMultiEnvDefaultFunc(t *testing.T) {
} }
}() }()
// Test that the first key is returned first
f := MultiEnvDefaultFunc(keys, "42") f := MultiEnvDefaultFunc(keys, "42")
if err := os.Setenv(keys[1], "foo"); err != nil { if err := os.Setenv(keys[0], "foo"); err != nil {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
} }
@ -64,10 +65,29 @@ func TestMultiEnvDefaultFunc(t *testing.T) {
t.Fatalf("bad: %#v", actual) t.Fatalf("bad: %#v", actual)
} }
if err := os.Unsetenv(keys[0]); err != nil {
t.Fatalf("err: %s", err)
}
// Test that the second key is returned if the first one is empty
f = MultiEnvDefaultFunc(keys, "42")
if err := os.Setenv(keys[1], "foo"); err != nil {
t.Fatalf("err: %s", err)
}
actual, err = f()
if err != nil {
t.Fatalf("err: %s", err)
}
if actual != "foo" {
t.Fatalf("bad: %#v", actual)
}
if err := os.Unsetenv(keys[1]); err != nil { if err := os.Unsetenv(keys[1]); err != nil {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
} }
// Test that the default value is returned when no keys are set
actual, err = f() actual, err = f()
if err != nil { if err != nil {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)