40 lines
705 B
Go
40 lines
705 B
Go
package orm
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/beego/beego/v2/client/orm/internal/models"
|
|
)
|
|
|
|
func Test_getColumnTyp(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
fi *models.FieldInfo
|
|
al *alias
|
|
|
|
wantCol string
|
|
}{
|
|
{
|
|
// https://github.com/beego/beego/issues/5254
|
|
name: "issue 5254",
|
|
fi: &models.FieldInfo{
|
|
FieldType: TypePositiveIntegerField,
|
|
Column: "my_col",
|
|
},
|
|
al: &alias{
|
|
DbBaser: newdbBasePostgres(),
|
|
},
|
|
wantCol: `bigint CHECK("my_col" >= 0)`,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
col := getColumnTyp(tc.al, tc.fi)
|
|
assert.Equal(t, tc.wantCol, col)
|
|
})
|
|
}
|
|
}
|