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

2 years ago · 4f6a900fd4
parent 63cfe60f1a
commit 4f6a900fd4
1 changed files with 2 additions and 2 deletions
--- 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"))