Files
go-zero/tools/goctl/util/ctx/gowork_test.go
2025-02-05 14:22:40 +00:00

41 lines
810 B
Go

package ctx
import (
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/assert"
"github.com/zeromicro/go-zero/core/stringx"
"github.com/zeromicro/go-zero/tools/goctl/rpc/execx"
"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
)
func Test_isGoWork(t *testing.T) {
dir := filepath.Join("/tmp", stringx.Rand())
err := pathx.MkdirIfNotExist(dir)
assert.Nil(t, err)
defer os.RemoveAll(dir)
gowork, err := isGoWork(dir)
assert.False(t, gowork)
assert.Nil(t, err)
_, err = execx.Run("go work init", dir)
assert.Nil(t, err)
gowork, err = isGoWork(dir)
assert.True(t, gowork)
assert.Nil(t, err)
subDir := filepath.Join(dir, stringx.Rand())
err = pathx.MkdirIfNotExist(subDir)
assert.Nil(t, err)
gowork, err = isGoWork(subDir)
assert.True(t, gowork)
assert.Nil(t, err)
}