46 lines
952 B
Go
46 lines
952 B
Go
|
package template
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/go-jet/jet/v2/generator/metadata"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_TableModelField(t *testing.T) {
|
||
|
|
require.Equal(t, DefaultTableModelField(metadata.Column{
|
||
|
|
Name: "col_name",
|
||
|
|
IsPrimaryKey: true,
|
||
|
|
IsNullable: true,
|
||
|
|
DataType: metadata.DataType{
|
||
|
|
Name: "smallint",
|
||
|
|
Kind: "base",
|
||
|
|
IsUnsigned: true,
|
||
|
|
},
|
||
|
|
}), TableModelField{
|
||
|
|
Name: "ColName",
|
||
|
|
Type: Type{
|
||
|
|
ImportPath: "",
|
||
|
|
Name: "*uint16",
|
||
|
|
},
|
||
|
|
Tags: []string{"sql:\"primary_key\""},
|
||
|
|
})
|
||
|
|
|
||
|
|
require.Equal(t, DefaultTableModelField(metadata.Column{
|
||
|
|
Name: "time_column_1",
|
||
|
|
IsPrimaryKey: false,
|
||
|
|
IsNullable: true,
|
||
|
|
DataType: metadata.DataType{
|
||
|
|
Name: "timestamp with time zone",
|
||
|
|
Kind: "base",
|
||
|
|
IsUnsigned: false,
|
||
|
|
},
|
||
|
|
}), TableModelField{
|
||
|
|
Name: "TimeColumn1",
|
||
|
|
Type: Type{
|
||
|
|
ImportPath: "time",
|
||
|
|
Name: "*time.Time",
|
||
|
|
},
|
||
|
|
Tags: nil,
|
||
|
|
})
|
||
|
|
}
|