73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package environment
|
|
|
|
import (
|
|
"reflect"
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
func TestEnvironment_AsList(t *testing.T) {
|
|
env := NewTaskEnivornment()
|
|
env.SetTaskIp("127.0.0.1")
|
|
env.SetPorts(map[string]int{"http": 80})
|
|
env.SetMeta(map[string]string{"foo": "baz"})
|
|
|
|
act := env.List()
|
|
exp := []string{"NOMAD_IP=127.0.0.1", "NOMAD_PORT_http=80", "NOMAD_META_FOO=baz"}
|
|
sort.Strings(act)
|
|
sort.Strings(exp)
|
|
if !reflect.DeepEqual(act, exp) {
|
|
t.Fatalf("env.List() returned %v; want %v", act, exp)
|
|
}
|
|
}
|
|
|
|
func TestEnvironment_ParseFromList(t *testing.T) {
|
|
input := []string{"foo=bar", "BAZ=baM"}
|
|
env, err := ParseFromList(input)
|
|
if err != nil {
|
|
t.Fatalf("ParseFromList(%#v) failed: %v", input, err)
|
|
}
|
|
|
|
exp := map[string]string{
|
|
"foo": "bar",
|
|
"BAZ": "baM",
|
|
}
|
|
|
|
if len(env) != len(exp) {
|
|
t.Fatalf("ParseFromList(%#v) has length %v; want %v", input, len(env), len(exp))
|
|
}
|
|
|
|
for k, v := range exp {
|
|
if actV, ok := env[k]; !ok {
|
|
t.Fatalf("ParseFromList(%#v) doesn't contain expected %v", input, k)
|
|
} else if actV != v {
|
|
t.Fatalf("ParseFromList(%#v) has incorrect value for %v; got %v; want %v", input, k, actV, v)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestEnvironment_ClearEnvvars(t *testing.T) {
|
|
env := NewTaskEnivornment()
|
|
env.SetTaskIp("127.0.0.1")
|
|
env.SetEnvvars(map[string]string{"foo": "baz", "bar": "bang"})
|
|
|
|
act := env.List()
|
|
exp := []string{"NOMAD_IP=127.0.0.1", "bar=bang", "foo=baz"}
|
|
sort.Strings(act)
|
|
sort.Strings(exp)
|
|
if !reflect.DeepEqual(act, exp) {
|
|
t.Fatalf("env.List() returned %v; want %v", act, exp)
|
|
}
|
|
|
|
// Clear the environent variables.
|
|
env.ClearEnvvars()
|
|
|
|
act = env.List()
|
|
exp = []string{"NOMAD_IP=127.0.0.1"}
|
|
sort.Strings(act)
|
|
sort.Strings(exp)
|
|
if !reflect.DeepEqual(act, exp) {
|
|
t.Fatalf("env.List() returned %v; want %v", act, exp)
|
|
}
|
|
}
|