Search vendor directories when using import path

Fixes maxbrunsfeld#75
nkovacs · Jun 23, 2017 · ee3dc58 · ee3dc58
1 parent c9ddd61
commit ee3dc58
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/locator/locator.go b/locator/locator.go
@@ -26,7 +26,7 @@ func GetInterfaceFromFilePath(interfaceName, filePath string) (*model.InterfaceT
 		return nil, err
 	}
 
-	vendorPaths, err := vendorPathsForDirPath(dirPath)
+	vendorPaths, err := VendorPathsForDirPath(dirPath)
 	if err != nil {
 		return nil, err
 	}
@@ -145,7 +145,7 @@ func importPathForDirPath(sourcePath string) (string, error) {
 	return "", fmt.Errorf("Path '%s' is not on GOPATH", sourcePath)
 }
 
-func vendorPathsForDirPath(dirPath string) ([]string, error) {
+func VendorPathsForDirPath(dirPath string) ([]string, error) {
 	dirPath, err := filepath.Abs(dirPath)
 	if err != nil {
 		return nil, err

diff --git a/main.go b/main.go
@@ -48,7 +48,11 @@ func generateFake(interfaceName, sourcePackageDir, importPath, outputPath, desti
 	var err error
 	var iface *model.InterfaceToFake
 	if sourcePackageDir == "" {
-		iface, err = locator.GetInterfaceFromImportPath(interfaceName, importPath)
+		var vendorPaths []string
+		vendorPaths, err = locator.VendorPathsForDirPath(".")
+		if err == nil {
+			iface, err = locator.GetInterfaceFromImportPath(interfaceName, importPath, vendorPaths...)
+		}
 	} else {
 		iface, err = locator.GetInterfaceFromFilePath(interfaceName, sourcePackageDir)
 	}