diff --git a/pkg/loader/compose/compose_test.go b/pkg/loader/compose/compose_test.go index d9d081291a..763714eb68 100644 --- a/pkg/loader/compose/compose_test.go +++ b/pkg/loader/compose/compose_test.go @@ -16,7 +16,12 @@ limitations under the License. package compose -import "testing" +import ( + "os" + "testing" + + "github.com/kubernetes-incubator/kompose/pkg/kobject" +) // Test if service types are parsed properly on user input // give a service type and expect correct input @@ -41,3 +46,45 @@ func TestHandleServiceType(t *testing.T) { } } } + +func TestLoadEnvVar(t *testing.T) { + ev1 := []string{"foo=bar"} + rs1 := kobject.EnvVar{ + Name: "foo", + Value: "bar", + } + ev2 := []string{"foo:bar"} + rs2 := kobject.EnvVar{ + Name: "foo", + Value: "bar", + } + ev3 := []string{"foo"} + rs3 := kobject.EnvVar{ + Name: "foo", + Value: "", + } + ev4 := []string{"osfoo"} + rs4 := kobject.EnvVar{ + Name: "osfoo", + Value: "osbar", + } + + tests := []struct { + envvars []string + results kobject.EnvVar + }{ + {ev1, rs1}, + {ev2, rs2}, + {ev3, rs3}, + {ev4, rs4}, + } + + os.Setenv("osfoo", "osbar") + + for _, tt := range tests { + result := loadEnvVars(tt.envvars) + if result[0] != tt.results { + t.Errorf("Expected %q, got %q", tt.results, result[0]) + } + } +}