Fix import name for pegomock in generated code

petergtz · May 4, 2023 · c969304 · c969304
1 parent 2af9258
commit c969304
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mockgen/mockgen.go b/mockgen/mockgen.go
@@ -103,6 +103,11 @@ func generateUniquePackageNamesFor(importPaths map[string]bool) (packageMap, non
 			packageName = sanitizedPackagePathBaseName + strconv.Itoa(i)
 		}
 
+		// hardcode package name for pegomock, because it's hardcoded in the generated code too
+		if importPath == mockFrameworkImportPath {
+			packageName = "pegomock"
+		}
+
 		packageMap[importPath] = packageName
 		packageNamesAlreadyUsed[packageName] = true