-
Notifications
You must be signed in to change notification settings - Fork 14
/
dbstruct_test.go
36 lines (33 loc) · 916 Bytes
/
dbstruct_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
package dbstruct
import (
"fmt"
"testing"
)
func TestNewDBStruct(t *testing.T) {
dbStruct := NewDBStruct()
err := dbStruct.
Dsn("root:root@tcp(127.0.0.1:3306)/wp-puock?charset=utf8").
StructNameFmt(FmtUnderlineToStartUpHump).
FieldNameFmt(FmtUnderlineToStartUpHump).
FileNameFmt(FmtUnderline).
SingleFile(true).
GenTableNameFunc(true).
GenTableName("TableName").
TagJson(true).
PackageName("model").
TagOrm(true).
AppendTag(NewTag("xml", FmtDefault)).
Generate()
if err != nil {
t.Error(err)
} else {
fmt.Println("ok.")
}
}
func TestGetNameFormat(t *testing.T) {
dbStruct := NewDBStruct()
fmt.Println(dbStruct.getFormatName("user_name", FmtUnderlineToStartLowHump))
fmt.Println(dbStruct.getFormatName("user_all_name", FmtUnderlineToStartUpHump))
fmt.Println(dbStruct.getFormatName("PlayerInfo", FmtUnderline))
fmt.Println(dbStruct.getFormatName("user_name", FmtDefault))
}