You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.1 KiB
Go
35 lines
1.1 KiB
Go
package gen
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/tal-tech/go-zero/core/logx"
|
|
)
|
|
|
|
var (
|
|
source = "CREATE TABLE `test_user_info` (\n `id` bigint NOT NULL AUTO_INCREMENT,\n `nanosecond` bigint NOT NULL DEFAULT '0',\n `data` varchar(255) DEFAULT '',\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 `nanosecond_unique` (`nanosecond`)\n) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;"
|
|
)
|
|
|
|
func TestCacheModel(t *testing.T) {
|
|
logx.Disable()
|
|
_ = Clean()
|
|
g := NewDefaultGenerator(source, "./testmodel/cache", NamingLower)
|
|
err := g.Start(true)
|
|
assert.Nil(t, err)
|
|
g = NewDefaultGenerator(source, "./testmodel/nocache", NamingLower)
|
|
err = g.Start(false)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestNamingModel(t *testing.T) {
|
|
logx.Disable()
|
|
_ = Clean()
|
|
g := NewDefaultGenerator(source, "./testmodel/camel", NamingCamel)
|
|
err := g.Start(true)
|
|
assert.Nil(t, err)
|
|
g = NewDefaultGenerator(source, "./testmodel/snake", NamingUnderline)
|
|
err = g.Start(true)
|
|
assert.Nil(t, err)
|
|
}
|