Fix a bug introduced in bdab667 that broke mocking interfaces with ve…

…ndored dependencies in source mode.
golang · Apr 4, 2018 · 42764bf · 42764bf
1 parent c4148b8
commit 42764bf
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 2 deletions.
diff --git a/mockgen/parse.go b/mockgen/parse.go
@@ -444,13 +444,13 @@ func importsOfFile(file *ast.File) map[string]string {
 			pkg, err := build.Import(importPath, "", 0)
 			if err != nil {
 				// Fallback to import path suffix. Note that this is uncertain.
-				log.Printf("failed to import package by path %s: %s - fallback to import path suffix", importPath, err.Error())
 				_, last := path.Split(importPath)
 				// If the last path component has dots, the first dot-delimited
 				// field is used as the name.
 				pkgName = strings.SplitN(last, ".", 2)[0]
+			} else {
+				pkgName = pkg.Name
 			}
-			pkgName = pkg.Name
 		}
 
 		if _, ok := m[pkgName]; ok {

diff --git a/mockgen/tests/vendor_dep/source_mock_package/mock.go b/mockgen/tests/vendor_dep/source_mock_package/mock.go