fix vendor imports

travisjeffery · Nov 15, 2017 · b81056f · b81056f
1 parent 5fbb656
commit b81056f
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pkg/mocker/mocker.go b/pkg/mocker/mocker.go
@@ -183,12 +183,14 @@ func (m *mocker) params(sig *types.Signature, tuple *types.Tuple, format string)
 			return ""
 		}
 		path := pkg.Path()
+		wd, err := os.Getwd()
+		if err != nil {
+			return ""
+		}
 		if path == "." {
-			wd, err := os.Getwd()
-			if err != nil {
-				return ""
-			}
 			path = strings.TrimPrefix(wd, os.Getenv("GOPATH")+"/src/")
+		} else {
+			path = strings.TrimPrefix(path, strings.TrimPrefix(wd, os.Getenv("GOPATH")+"/src/")+"/vendor/")
 		}
 		m.imports[path] = true
 		return pkg.Name()