fix(goctl): fix the unit test bug of goctl (#2458)

zeromicro · Sep 27, 2022 · 4f6a900 · 4f6a900
1 parent 63cfe60
commit 4f6a900
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/goctl/model/sql/gen/gen_test.go b/tools/goctl/model/sql/gen/gen_test.go
@@ -119,7 +119,7 @@ func TestFolderName(t *testing.T) {
 
 	pkg := g.pkg
 
-	err = g.StartFromDDL(sqlFile, true, "go_zero")
+	err = g.StartFromDDL(sqlFile, true, true, "go_zero")
 	assert.Nil(t, err)
 	assert.True(t, func() bool {
 		_, err := os.Stat(filepath.Join(camelDir, "TestUserModel.go"))
@@ -132,7 +132,7 @@ func TestFolderName(t *testing.T) {
 	})
 	assert.Nil(t, err)
 
-	err = g.StartFromDDL(sqlFile, true, "go_zero")
+	err = g.StartFromDDL(sqlFile, true, true, "go_zero")
 	assert.Nil(t, err)
 	assert.True(t, func() bool {
 		_, err := os.Stat(filepath.Join(snakeDir, "test_user_model.go"))