package command import ( "io/ioutil" "os" "path/filepath" "testing" "github.com/stretchr/testify/assert" "github.com/tal-tech/go-zero/tools/goctl/config" "github.com/tal-tech/go-zero/tools/goctl/util" ) var ( sql = "-- 用户表 --\nCREATE TABLE `user` (\n `id` bigint(10) NOT NULL AUTO_INCREMENT,\n `name` varchar(255) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT '用户名称',\n `password` varchar(255) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT '用户密码',\n `mobile` varchar(255) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT '手机号',\n `gender` char(5) COLLATE utf8mb4_general_ci NOT NULL COMMENT '男|女|未公开',\n `nickname` varchar(255) COLLATE utf8mb4_general_ci DEFAULT '' COMMENT '用户昵称',\n `create_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP,\n `update_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,\n PRIMARY KEY (`id`),\n UNIQUE KEY `name_index` (`name`),\n UNIQUE KEY `mobile_index` (`mobile`)\n) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;\n\n" cfg = &config.Config{ NamingFormat: "gozero", } ) func TestFromDDl(t *testing.T) { err := fromDDl("./user.sql", t.TempDir(), cfg, true, false) assert.Equal(t, errNotMatched, err) // case dir is not exists unknownDir := filepath.Join(t.TempDir(), "test", "user.sql") err = fromDDl(unknownDir, t.TempDir(), cfg, true, false) assert.True(t, func() bool { switch err.(type) { case *os.PathError: return true default: return false } }()) // case empty src err = fromDDl("", t.TempDir(), cfg, true, false) if err != nil { assert.Equal(t, "expected path or path globbing patterns, but nothing found", err.Error()) } tempDir := filepath.Join(t.TempDir(), "test") err = util.MkdirIfNotExist(tempDir) if err != nil { return } user1Sql := filepath.Join(tempDir, "user1.sql") user2Sql := filepath.Join(tempDir, "user2.sql") err = ioutil.WriteFile(user1Sql, []byte(sql), os.ModePerm) if err != nil { return } err = ioutil.WriteFile(user2Sql, []byte(sql), os.ModePerm) if err != nil { return } _, err = os.Stat(user1Sql) assert.Nil(t, err) _, err = os.Stat(user2Sql) assert.Nil(t, err) err = fromDDl(filepath.Join(tempDir, "user*.sql"), tempDir, cfg, true, false) assert.Nil(t, err) _, err = os.Stat(filepath.Join(tempDir, "usermodel.go")) assert.Nil(t, err) }