Fix putenv (check key prefix properly)

djhworld · Jul 7, 2012 · f8ef66b · f8ef66b
1 parent 6ed339b
commit f8ef66b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/gocov/instrument.go b/gocov/instrument.go
@@ -159,8 +159,8 @@ func (v *funcVisitor) Visit(n ast.Node) ast.Visitor {
 		sv.VisitStmt(n.Body)
 		// TODO function coverage (insert "function.Enter", "function.Leave").
 
-	// TODO come up with naming scheme for function literals.
-	// case *ast.FuncLit:
+		// TODO come up with naming scheme for function literals.
+		// case *ast.FuncLit:
 	}
 	return v
 }

diff --git a/gocov/main.go b/gocov/main.go
@@ -59,7 +59,7 @@ type instrumenter struct {
 
 func putenv(env []string, key, value string) []string {
 	for i, s := range env {
-		if strings.HasPrefix(s, "GOPATH=") {
+		if strings.HasPrefix(s, key+"=") {
 			env[i] = key + "=" + value
 			return env
 		}