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  }