github.com/jmitchell/nomad@v0.1.3-0.20151007230021-7ab84c2862d8/client/driver/environment/vars_test.go (about) 1 package environment 2 3 import ( 4 "reflect" 5 "sort" 6 "testing" 7 ) 8 9 func TestEnvironment_AsList(t *testing.T) { 10 env := NewTaskEnivornment() 11 env.SetTaskIp("127.0.0.1") 12 env.SetPorts(map[string]int{"http": 80}) 13 env.SetMeta(map[string]string{"foo": "baz"}) 14 15 act := env.List() 16 exp := []string{"NOMAD_IP=127.0.0.1", "NOMAD_PORT_http=80", "NOMAD_META_FOO=baz"} 17 sort.Strings(act) 18 sort.Strings(exp) 19 if !reflect.DeepEqual(act, exp) { 20 t.Fatalf("env.List() returned %v; want %v", act, exp) 21 } 22 } 23 24 func TastEnvironment_ParseFromList(t *testing.T) { 25 input := []string{"foo=bar", "BAZ=baM"} 26 env, err := ParseFromList(input) 27 if err != nil { 28 t.Fatalf("ParseFromList(%#v) failed: %v", input, err) 29 } 30 31 exp := map[string]string{ 32 "foo": "baz", 33 "BAZ": "baM", 34 } 35 if !reflect.DeepEqual(env, exp) { 36 t.Fatalf("ParseFromList(%#v) returned %v; want %v", input, env, exp) 37 } 38 }