mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-27 20:38:41 +08:00
36 lines
677 B
Go
36 lines
677 B
Go
package generate
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/tools/goctl/config"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
|
|
)
|
|
|
|
var testTypes = `
|
|
type User struct{}
|
|
type Class struct{}
|
|
`
|
|
|
|
func TestDo(t *testing.T) {
|
|
cfg, err := config.NewConfig(config.DefaultFormat)
|
|
assert.Nil(t, err)
|
|
|
|
tempDir := pathx.MustTempDir()
|
|
typesfile := filepath.Join(tempDir, "types.go")
|
|
err = ioutil.WriteFile(typesfile, []byte(testTypes), 0o666)
|
|
assert.Nil(t, err)
|
|
|
|
err = Do(&Context{
|
|
Types: []string{"User", "Class"},
|
|
Cache: false,
|
|
Output: tempDir,
|
|
Cfg: cfg,
|
|
})
|
|
|
|
assert.Nil(t, err)
|
|
}
|