fix: fix query unit test import list

go-gorm · May 8, 2024 · d7a0969 · d7a0969
1 parent 9932bbb
commit d7a0969
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/generator.go b/generator.go
@@ -462,7 +462,7 @@ func (g *Generator) generateQueryUnitTestFile(data *genInfo) (err error) {
 	}
 	err = render(tmpl.Header, &buf, map[string]interface{}{
 		"Package":        g.queryPkgName,
-		"ImportPkgPaths": unitTestImportList.Add(structPkgPath).Add(data.ImportPkgPaths...).Paths(),
+		"ImportPkgPaths": unitTestQueryImportList.Add(structPkgPath).Add(data.ImportPkgPaths...).Paths(),
 	})
 	if err != nil {
 		return err

diff --git a/import.go b/import.go
@@ -27,6 +27,16 @@ var (
 		"gorm.io/driver/sqlite",
 		"gorm.io/gorm",
 	)
+	unitTestQueryImportList = new(importPkgS).Add(
+		"context",
+		"fmt",
+		"strconv",
+		"testing",
+		"",
+		"gorm.io/gen",
+		"gorm.io/gen/field",
+		"gorm.io/gorm/clause",
+	)
 )
 
 type importPkgS struct {