feat(goctl): support go work (#4332) (#4344)

tools/goctl/util/ctx/gowork_test.go

@@ -0,0 +1,40 @@
+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)
+}