refactor file|path (#1409)
Co-authored-by: anqiansong <anqiansong@bytedance.com>
This commit is contained in:
@@ -15,6 +15,7 @@ import (
|
||||
"github.com/tal-tech/go-zero/tools/goctl/model/sql/util"
|
||||
file "github.com/tal-tech/go-zero/tools/goctl/util"
|
||||
"github.com/tal-tech/go-zero/tools/goctl/util/console"
|
||||
"github.com/tal-tech/go-zero/tools/goctl/util/pathx"
|
||||
"github.com/urfave/cli"
|
||||
)
|
||||
|
||||
@@ -50,7 +51,7 @@ func MysqlDDL(ctx *cli.Context) error {
|
||||
}
|
||||
}
|
||||
if len(home) > 0 {
|
||||
file.RegisterGoctlHome(home)
|
||||
pathx.RegisterGoctlHome(home)
|
||||
}
|
||||
cfg, err := config.NewConfig(style)
|
||||
if err != nil {
|
||||
@@ -76,7 +77,7 @@ func MySqlDataSource(ctx *cli.Context) error {
|
||||
}
|
||||
}
|
||||
if len(home) > 0 {
|
||||
file.RegisterGoctlHome(home)
|
||||
pathx.RegisterGoctlHome(home)
|
||||
}
|
||||
|
||||
pattern := strings.TrimSpace(ctx.String(flagTable))
|
||||
@@ -105,7 +106,7 @@ func PostgreSqlDataSource(ctx *cli.Context) error {
|
||||
}
|
||||
}
|
||||
if len(home) > 0 {
|
||||
file.RegisterGoctlHome(home)
|
||||
pathx.RegisterGoctlHome(home)
|
||||
}
|
||||
|
||||
if len(schema) == 0 {
|
||||
|
||||
@@ -9,7 +9,7 @@ import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/tal-tech/go-zero/tools/goctl/config"
|
||||
"github.com/tal-tech/go-zero/tools/goctl/model/sql/gen"
|
||||
"github.com/tal-tech/go-zero/tools/goctl/util"
|
||||
"github.com/tal-tech/go-zero/tools/goctl/util/pathx"
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -23,12 +23,12 @@ func TestFromDDl(t *testing.T) {
|
||||
err := gen.Clean()
|
||||
assert.Nil(t, err)
|
||||
|
||||
err = fromDDL("./user.sql", util.MustTempDir(), cfg, true, false, "go_zero")
|
||||
err = fromDDL("./user.sql", pathx.MustTempDir(), cfg, true, false, "go_zero")
|
||||
assert.Equal(t, errNotMatched, err)
|
||||
|
||||
// case dir is not exists
|
||||
unknownDir := filepath.Join(util.MustTempDir(), "test", "user.sql")
|
||||
err = fromDDL(unknownDir, util.MustTempDir(), cfg, true, false, "go_zero")
|
||||
unknownDir := filepath.Join(pathx.MustTempDir(), "test", "user.sql")
|
||||
err = fromDDL(unknownDir, pathx.MustTempDir(), cfg, true, false, "go_zero")
|
||||
assert.True(t, func() bool {
|
||||
switch err.(type) {
|
||||
case *os.PathError:
|
||||
@@ -39,13 +39,13 @@ func TestFromDDl(t *testing.T) {
|
||||
}())
|
||||
|
||||
// case empty src
|
||||
err = fromDDL("", util.MustTempDir(), cfg, true, false, "go_zero")
|
||||
err = fromDDL("", pathx.MustTempDir(), cfg, true, false, "go_zero")
|
||||
if err != nil {
|
||||
assert.Equal(t, "expected path or path globbing patterns, but nothing found", err.Error())
|
||||
}
|
||||
|
||||
tempDir := filepath.Join(util.MustTempDir(), "test")
|
||||
err = util.MkdirIfNotExist(tempDir)
|
||||
tempDir := filepath.Join(pathx.MustTempDir(), "test")
|
||||
err = pathx.MkdirIfNotExist(tempDir)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user