Damnit again
Former-commit-id: b394f26caf0df7d113ac4cc7dacc9c544af6897f
This commit is contained in:
40
config/new_test.go
Normal file
40
config/new_test.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestNew(t *testing.T) {
|
||||
invalidEnv := "&&&&&&"
|
||||
keys := []string{
|
||||
"DB",
|
||||
"DATABASE",
|
||||
"ADDR",
|
||||
"PATH",
|
||||
"USER",
|
||||
"USERNAME",
|
||||
"PASS",
|
||||
"PASSWORD",
|
||||
}
|
||||
was := make(map[string]string)
|
||||
for _, k := range keys {
|
||||
v, ok := os.LookupEnv(k)
|
||||
if !ok {
|
||||
v = invalidEnv
|
||||
}
|
||||
was[k] = v
|
||||
}
|
||||
defer func() {
|
||||
for k, v := range was {
|
||||
if v != invalidEnv {
|
||||
os.Setenv(k, v)
|
||||
} else {
|
||||
os.Unsetenv(k)
|
||||
}
|
||||
}
|
||||
}()
|
||||
if err := New(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user