mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
619b63d5f7
Reviewed By: idoshveki Differential Revision: D16687892 fbshipit-source-id: e8cfaaf1241e94c2de0a9fe9077326339d593716
88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package field_test
|
|
|
|
import (
|
|
"regexp"
|
|
"testing"
|
|
|
|
"fbc/ent"
|
|
"fbc/ent/field"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestInt(t *testing.T) {
|
|
f := field.Int("age").Positive()
|
|
assert.Equal(t, "age", f.Name())
|
|
assert.Equal(t, field.TypeInt, f.Type())
|
|
assert.Len(t, f.Validators(), 1)
|
|
|
|
f = field.Int("age").Default(10).Min(10).Max(20)
|
|
assert.True(t, f.HasDefault())
|
|
assert.Equal(t, 10, f.Value())
|
|
assert.Len(t, f.Validators(), 2)
|
|
|
|
f = field.Int("age").Range(20, 40).Nillable()
|
|
assert.False(t, f.HasDefault())
|
|
assert.True(t, f.IsNillable())
|
|
assert.Len(t, f.Validators(), 1)
|
|
|
|
assert.Equal(t, field.TypeInt8, field.Int8("age").Type())
|
|
assert.Equal(t, field.TypeInt16, field.Int16("age").Type())
|
|
assert.Equal(t, field.TypeInt32, field.Int32("age").Type())
|
|
assert.Equal(t, field.TypeInt64, field.Int64("age").Type())
|
|
}
|
|
|
|
func TestFloat(t *testing.T) {
|
|
f := field.Float("age").Positive()
|
|
assert.Equal(t, "age", f.Name())
|
|
assert.Equal(t, field.TypeFloat64, f.Type())
|
|
assert.Len(t, f.Validators(), 1)
|
|
|
|
f = field.Float("age").Min(2.5).Max(5)
|
|
assert.Len(t, f.Validators(), 2)
|
|
}
|
|
|
|
func TestBool(t *testing.T) {
|
|
f := field.Bool("active").Default(true)
|
|
assert.Equal(t, "active", f.Name())
|
|
assert.Equal(t, field.TypeBool, f.Type())
|
|
assert.True(t, f.HasDefault())
|
|
assert.Equal(t, true, f.Value())
|
|
}
|
|
|
|
func TestBytes(t *testing.T) {
|
|
f := field.Bytes("active").Default([]byte("{}"))
|
|
assert.Equal(t, "active", f.Name())
|
|
assert.Equal(t, field.TypeBytes, f.Type())
|
|
assert.True(t, f.HasDefault())
|
|
assert.Equal(t, []byte("{}"), f.Value())
|
|
}
|
|
|
|
func TestString(t *testing.T) {
|
|
re := regexp.MustCompile("[a-zA-Z0-9]")
|
|
f := field.String("name").Unique().Match(re).Validate(func(string) error { return nil })
|
|
assert.Equal(t, field.TypeString, f.Type())
|
|
assert.Equal(t, "name", f.Name())
|
|
assert.True(t, f.IsUnique())
|
|
assert.Len(t, f.Validators(), 2)
|
|
}
|
|
|
|
func TestCharset(t *testing.T) {
|
|
var f ent.Field = field.String("name").SetCharset("utf8")
|
|
cs, ok := f.(field.Charseter)
|
|
assert.True(t, ok, "string field implements the Charseter interface")
|
|
assert.Equal(t, "utf8", cs.Charset())
|
|
}
|
|
|
|
func TestTime(t *testing.T) {
|
|
f := field.Time("created_at")
|
|
assert.Equal(t, "created_at", f.Name())
|
|
assert.Equal(t, field.TypeTime, f.Type())
|
|
assert.Equal(t, "time.Time", f.Type().String())
|
|
}
|
|
|
|
func TestField_Tag(t *testing.T) {
|
|
f := field.Bool("expired").StructTag(`json:"expired,omitempty"`)
|
|
assert.Equal(t, `json:"expired,omitempty"`, f.Tag())
|
|
}
|