Environment variables may not have a value, currently env.Map blows u…

…p as it's expecting this to always be the case
codeship · Sep 21, 2017 · cd1c311 · cd1c311
1 parent 98edf3e
commit cd1c311
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/env.go b/env.go
@@ -162,7 +162,11 @@ func (env *Env) Map() map[string]string {
 	m := make(map[string]string)
 	for _, kv := range *env {
 		parts := strings.SplitN(kv, "=", 2)
-		m[parts[0]] = parts[1]
+		if len(parts) == 1 {
+			m[parts[0]] = ""
+		} else {
+			m[parts[0]] = parts[1]
+		}
 	}
 	return m
 }
diff --git a/env_test.go b/env_test.go
@@ -351,6 +351,7 @@ func TestMap(t *testing.T) {
 		expected map[string]string
 	}{
 		{[]string{"PATH=/usr/bin:/bin", "PYTHONPATH=/usr/local"}, map[string]string{"PATH": "/usr/bin:/bin", "PYTHONPATH": "/usr/local"}},
+		{[]string{"ENABLE_LOGGING", "PYTHONPATH=/usr/local"}, map[string]string{"ENABLE_LOGGING": "", "PYTHONPATH": "/usr/local"}},
 		{nil, nil},
 	}
 	for _, tt := range tests {