123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561 |
- package test
- import (
- "git.sxidc.com/go-framework/baize/framework/core/tag/assign"
- "git.sxidc.com/go-tools/utils/strutils"
- "github.com/pkg/errors"
- "math/rand"
- "strings"
- "testing"
- "time"
- )
- type AssignTagFrom struct {
- StringField string `assign:"toField:StringField"`
- BoolField bool `assign:"toField:BoolField"`
- IntField int `assign:"toField:IntField"`
- Int8Field int8 `assign:"toField:Int8Field"`
- Int16Field int16 `assign:"toField:Int16Field"`
- Int32Field int32 `assign:"toField:Int32Field"`
- Int64Field int64 `assign:"toField:Int64Field"`
- UintField uint `assign:"toField:UintField"`
- Uint8Field uint8 `assign:"toField:Uint8Field"`
- Uint16Field uint16 `assign:"toField:Uint16Field"`
- Uint32Field uint32 `assign:"toField:Uint32Field"`
- Uint64Field uint64 `assign:"toField:Uint64Field"`
- Float32Field float32 `assign:"toField:Float32Field"`
- Float64Field float64 `assign:"toField:Float64Field"`
- SliceField []string `assign:"toField:SliceField"`
- MapField map[string]string `assign:"toField:MapField"`
- ChanField chan any `assign:"toField:ChanField"`
- FuncField func() string `assign:"toField:FuncField"`
- StructField struct{ ID string } `assign:"toField:StructField"`
- FromStringSliceToStringField []string `assign:"toField:FromStringSliceToStringField"`
- FromTimeToStringField time.Time `assign:"toField:FromTimeToStringField"`
- FromTimeToTimeField time.Time `assign:"toField:FromTimeToTimeField"`
- FromStringToTimeField string `assign:"toField:FromStringToTimeField"`
- }
- type AssignTagFromPointerField struct {
- StringField *string `assign:"toField:StringField"`
- BoolField *bool `assign:"toField:BoolField"`
- IntField *int `assign:"toField:IntField"`
- Int8Field *int8 `assign:"toField:Int8Field"`
- Int16Field *int16 `assign:"toField:Int16Field"`
- Int32Field *int32 `assign:"toField:Int32Field"`
- Int64Field *int64 `assign:"toField:Int64Field"`
- UintField *uint `assign:"toField:UintField"`
- Uint8Field *uint8 `assign:"toField:Uint8Field"`
- Uint16Field *uint16 `assign:"toField:Uint16Field"`
- Uint32Field *uint32 `assign:"toField:Uint32Field"`
- Uint64Field *uint64 `assign:"toField:Uint64Field"`
- Float32Field *float32 `assign:"toField:Float32Field"`
- Float64Field *float64 `assign:"toField:Float64Field"`
- SliceField *[]string `assign:"toField:SliceField"`
- MapField *map[string]string `assign:"toField:MapField"`
- ChanField *chan any `assign:"toField:ChanField"`
- FuncField *func() string `assign:"toField:FuncField"`
- StructField *struct{ ID string } `assign:"toField:StructField"`
- FromStringSliceToStringField *[]string `assign:"toField:FromStringSliceToStringField"`
- FromTimeToStringField *time.Time `assign:"toField:FromTimeToStringField"`
- FromTimeToTimeField *time.Time `assign:"toField:FromTimeToTimeField"`
- FromStringToTimeField *string `assign:"toField:FromStringToTimeField"`
- }
- type AssignTagTo struct {
- StringField string
- BoolField bool
- IntField int
- Int8Field int8
- Int16Field int16
- Int32Field int32
- Int64Field int64
- UintField uint
- Uint8Field uint8
- Uint16Field uint16
- Uint32Field uint32
- Uint64Field uint64
- Float32Field float32
- Float64Field float64
- SliceField []string
- MapField map[string]string
- ChanField chan any
- FuncField func() string
- StructField struct{ ID string }
- FromStringSliceToStringField string
- FromTimeToStringField string
- FromTimeToTimeField time.Time
- FromStringToTimeField time.Time
- }
- func (to AssignTagTo) checkFields(t *testing.T, from AssignTagFrom) {
- if from.StringField != to.StringField {
- t.Fatalf("%+v\n", errors.Errorf("StringField not equal: from %v, to %v",
- from.StringField, to.StringField))
- }
- if from.BoolField != to.BoolField {
- t.Fatalf("%+v\n", errors.Errorf("BoolField not equal: from %v, to %v",
- from.BoolField, to.BoolField))
- }
- if from.IntField != to.IntField {
- t.Fatalf("%+v\n", errors.Errorf("IntField not equal: from %v, to %v",
- from.IntField, to.IntField))
- }
- if from.Int8Field != to.Int8Field {
- t.Fatalf("%+v\n", errors.Errorf("Int8Field not equal: from %v, to %v",
- from.Int8Field, to.Int8Field))
- }
- if from.Int16Field != to.Int16Field {
- t.Fatalf("%+v\n", errors.Errorf("Int16Field not equal: from %v, to %v",
- from.Int16Field, to.Int16Field))
- }
- if from.Int32Field != to.Int32Field {
- t.Fatalf("%+v\n", errors.Errorf("Int32Field not equal: from %v, to %v",
- from.Int32Field, to.Int32Field))
- }
- if from.Int64Field != to.Int64Field {
- t.Fatalf("%+v\n", errors.Errorf("Int64Field not equal: from %v, to %v",
- from.Int64Field, to.Int64Field))
- }
- if from.UintField != to.UintField {
- t.Fatalf("%+v\n", errors.Errorf("UintField not equal: from %v, to %v",
- from.UintField, to.UintField))
- }
- if from.Uint8Field != to.Uint8Field {
- t.Fatalf("%+v\n", errors.Errorf("Uint8Field not equal: from %v, to %v",
- from.Uint8Field, to.Uint8Field))
- }
- if from.Uint16Field != to.Uint16Field {
- t.Fatalf("%+v\n", errors.Errorf("Uint16Field not equal: from %v, to %v",
- from.Uint16Field, to.Uint16Field))
- }
- if from.Uint32Field != to.Uint32Field {
- t.Fatalf("%+v\n", errors.Errorf("Uint32Field not equal: from %v, to %v",
- from.Uint32Field, to.Uint32Field))
- }
- if from.Uint64Field != to.Uint64Field {
- t.Fatalf("%+v\n", errors.Errorf("Uint64Field not equal: from %v, to %v",
- from.Uint64Field, to.Uint64Field))
- }
- if from.Float32Field != to.Float32Field {
- t.Fatalf("%+v\n", errors.Errorf("Float32Field not equal: from %v, to %v",
- from.Float32Field, to.Float32Field))
- }
- if from.Float64Field != to.Float64Field {
- t.Fatalf("%+v\n", errors.Errorf("Float64Field not equal: from %v, to %v",
- from.Float64Field, to.Float64Field))
- }
- for i, value := range from.SliceField {
- if value != to.SliceField[i] {
- t.Fatalf("%+v\n", errors.Errorf("SliceField not equal: from %v, to %v",
- value, to.SliceField[i]))
- }
- }
- for key, value := range from.MapField {
- if value != to.MapField[key] {
- t.Fatalf("%+v\n", errors.Errorf("MapField not equal: from %v, to %v",
- value, to.MapField[key]))
- }
- }
- if from.ChanField != to.ChanField {
- t.Fatalf("%+v\n", errors.Errorf("ChanField not equal: from %v, to %v",
- from.ChanField, to.ChanField))
- }
- if from.FuncField() != to.FuncField() {
- t.Fatalf("%+v\n", errors.Errorf("FuncField not equal: from %v, to %v",
- from.FuncField(), to.FuncField()))
- }
- if from.StructField.ID != to.StructField.ID {
- t.Fatalf("%+v\n", errors.Errorf("StructField not equal: from %v, to %v",
- from.StructField.ID, to.StructField.ID))
- }
- if strings.Join(from.FromStringSliceToStringField, "::") != to.FromStringSliceToStringField {
- t.Fatalf("%+v\n", errors.Errorf("FromStringSliceToStringField not equal: from %v, to %v",
- strings.Join(from.FromStringSliceToStringField, "::"), to.FromStringSliceToStringField))
- }
- if from.FromTimeToStringField.Format(time.DateTime) != to.FromTimeToStringField {
- t.Fatalf("%+v\n", errors.Errorf("FromTimeToStringField not equal: from %v, to %v",
- from.FromTimeToStringField.Format(time.DateTime), to.FromTimeToStringField))
- }
- if from.FromTimeToTimeField.Format(time.DateTime) != to.FromTimeToTimeField.Format(time.DateTime) {
- t.Fatalf("%+v\n", errors.Errorf("FromTimeToTimeField not equal: from %v, to %v",
- from.FromTimeToTimeField.Format(time.DateTime), to.FromTimeToTimeField.Format(time.DateTime)))
- }
- if from.FromStringToTimeField != to.FromStringToTimeField.Format(time.DateTime) {
- t.Fatalf("%+v\n", errors.Errorf("FromStringToTimeField not equal: from %v, to %v",
- from.FromStringToTimeField, to.FromStringToTimeField.Format(time.DateTime)))
- }
- }
- type AssignTagToPointerField struct {
- StringField *string
- BoolField *bool
- IntField *int
- Int8Field *int8
- Int16Field *int16
- Int32Field *int32
- Int64Field *int64
- UintField *uint
- Uint8Field *uint8
- Uint16Field *uint16
- Uint32Field *uint32
- Uint64Field *uint64
- Float32Field *float32
- Float64Field *float64
- SliceField *[]string
- MapField *map[string]string
- ChanField *chan any
- FuncField *func() string
- StructField *struct{ ID string }
- FromStringSliceToStringField *string
- FromTimeToStringField *string
- FromTimeToTimeField *time.Time
- FromStringToTimeField *time.Time
- }
- func (to AssignTagToPointerField) checkFields(t *testing.T, from AssignTagFrom) {
- if from.StringField != *to.StringField {
- t.Fatalf("%+v\n", errors.Errorf("StringField not equal: from %v, to %v",
- from.StringField, *to.StringField))
- }
- if from.BoolField != *to.BoolField {
- t.Fatalf("%+v\n", errors.Errorf("BoolField not equal: from %v, to %v",
- from.BoolField, *to.BoolField))
- }
- if from.IntField != *to.IntField {
- t.Fatalf("%+v\n", errors.Errorf("IntField not equal: from %v, to %v",
- from.IntField, *to.IntField))
- }
- if from.Int8Field != *to.Int8Field {
- t.Fatalf("%+v\n", errors.Errorf("Int8Field not equal: from %v, to %v",
- from.Int8Field, *to.Int8Field))
- }
- if from.Int16Field != *to.Int16Field {
- t.Fatalf("%+v\n", errors.Errorf("Int16Field not equal: from %v, to %v",
- from.Int16Field, *to.Int16Field))
- }
- if from.Int32Field != *to.Int32Field {
- t.Fatalf("%+v\n", errors.Errorf("Int32Field not equal: from %v, to %v",
- from.Int32Field, *to.Int32Field))
- }
- if from.Int64Field != *to.Int64Field {
- t.Fatalf("%+v\n", errors.Errorf("Int64Field not equal: from %v, to %v",
- from.Int64Field, *to.Int64Field))
- }
- if from.UintField != *to.UintField {
- t.Fatalf("%+v\n", errors.Errorf("UintField not equal: from %v, to %v",
- from.UintField, *to.UintField))
- }
- if from.Uint8Field != *to.Uint8Field {
- t.Fatalf("%+v\n", errors.Errorf("Uint8Field not equal: from %v, to %v",
- from.Uint8Field, *to.Uint8Field))
- }
- if from.Uint16Field != *to.Uint16Field {
- t.Fatalf("%+v\n", errors.Errorf("Uint16Field not equal: from %v, to %v",
- from.Uint16Field, *to.Uint16Field))
- }
- if from.Uint32Field != *to.Uint32Field {
- t.Fatalf("%+v\n", errors.Errorf("Uint32Field not equal: from %v, to %v",
- from.Uint32Field, *to.Uint32Field))
- }
- if from.Uint64Field != *to.Uint64Field {
- t.Fatalf("%+v\n", errors.Errorf("Uint64Field not equal: from %v, to %v",
- from.Uint64Field, *to.Uint64Field))
- }
- if from.Float32Field != *to.Float32Field {
- t.Fatalf("%+v\n", errors.Errorf("Float32Field not equal: from %v, to %v",
- from.Float32Field, *to.Float32Field))
- }
- if from.Float64Field != *to.Float64Field {
- t.Fatalf("%+v\n", errors.Errorf("Float64Field not equal: from %v, to %v",
- from.Float64Field, *to.Float64Field))
- }
- for i, value := range from.SliceField {
- if value != (*to.SliceField)[i] {
- t.Fatalf("%+v\n", errors.Errorf("SliceField not equal: from %v, to %v",
- value, (*to.SliceField)[i]))
- }
- }
- for key, value := range from.MapField {
- if value != (*to.MapField)[key] {
- t.Fatalf("%+v\n", errors.Errorf("MapField not equal: from %v, to %v",
- value, (*to.MapField)[key]))
- }
- }
- if from.ChanField != *to.ChanField {
- t.Fatalf("%+v\n", errors.Errorf("ChanField not equal: from %v, to %v",
- from.ChanField, *to.ChanField))
- }
- if from.FuncField() != (*to.FuncField)() {
- t.Fatalf("%+v\n", errors.Errorf("FuncField not equal: from %v, to %v",
- from.FuncField(), (*to.FuncField)()))
- }
- if from.StructField.ID != (*to.StructField).ID {
- t.Fatalf("%+v\n", errors.Errorf("StructField not equal: from %v, to %v",
- from.StructField.ID, (*to.StructField).ID))
- }
- if strings.Join(from.FromStringSliceToStringField, "::") != *to.FromStringSliceToStringField {
- t.Fatalf("%+v\n", errors.Errorf("FromStringSliceToStringField not equal: from %v, to %v",
- strings.Join(from.FromStringSliceToStringField, "::"), to.FromStringSliceToStringField))
- }
- if from.FromTimeToStringField.Format(time.DateTime) != *to.FromTimeToStringField {
- t.Fatalf("%+v\n", errors.Errorf("FromTimeToStringField not equal: from %v, to %v",
- from.FromTimeToStringField.Format(time.DateTime), to.FromTimeToStringField))
- }
- if from.FromTimeToTimeField.Format(time.DateTime) != (*to.FromTimeToTimeField).Format(time.DateTime) {
- t.Fatalf("%+v\n", errors.Errorf("FromTimeToTimeField not equal: from %v, to %v",
- from.FromTimeToTimeField.Format(time.DateTime), (*to.FromTimeToTimeField).Format(time.DateTime)))
- }
- if from.FromStringToTimeField != (*to.FromStringToTimeField).Format(time.DateTime) {
- t.Fatalf("%+v\n", errors.Errorf("FromStringToTimeField not equal: from %v, to %v",
- from.FromStringToTimeField, (*to.FromStringToTimeField).Format(time.DateTime)))
- }
- }
- func TestAssignTagDefaultUsage(t *testing.T) {
- stringField := strutils.SimpleUUID()
- boolField := rand.Intn(2) == 0
- intField := rand.Int()
- int8Field := int8(rand.Int())
- int16Field := int16(rand.Int())
- int32Field := int32(rand.Int())
- int64Field := int64(rand.Int())
- uintField := uint(rand.Int())
- uint8Field := uint8(rand.Int())
- uint16Field := uint16(rand.Int())
- uint32Field := uint32(rand.Int())
- uint64Field := uint64(rand.Int())
- float32Field := rand.Float32()
- float64Field := rand.Float64()
- sliceField := []string{strutils.SimpleUUID(), strutils.SimpleUUID()}
- mapField := map[string]string{strutils.SimpleUUID(): strutils.SimpleUUID()}
- chanField := make(chan any)
- returnValue := strutils.SimpleUUID()
- funcField := func() string { return returnValue }
- structField := struct{ ID string }{ID: strutils.SimpleUUID()}
- fromStringSliceToStringField := []string{strutils.SimpleUUID(), strutils.SimpleUUID()}
- fromTimeToStringField := time.Now().Local()
- fromTimeToTimeField := time.Now().Local()
- fromStringToTimeField := time.Now().Local().Format(time.DateTime)
- from := AssignTagFrom{
- StringField: stringField,
- BoolField: boolField,
- IntField: intField,
- Int8Field: int8Field,
- Int16Field: int16Field,
- Int32Field: int32Field,
- Int64Field: int64Field,
- UintField: uintField,
- Uint8Field: uint8Field,
- Uint16Field: uint16Field,
- Uint32Field: uint32Field,
- Uint64Field: uint64Field,
- Float32Field: float32Field,
- Float64Field: float64Field,
- SliceField: sliceField,
- MapField: mapField,
- ChanField: chanField,
- FuncField: funcField,
- StructField: structField,
- FromStringSliceToStringField: fromStringSliceToStringField,
- FromTimeToStringField: fromTimeToStringField,
- FromTimeToTimeField: fromTimeToTimeField,
- FromStringToTimeField: fromStringToTimeField,
- }
- fromPointer := AssignTagFromPointerField{
- StringField: &stringField,
- BoolField: &boolField,
- IntField: &intField,
- Int8Field: &int8Field,
- Int16Field: &int16Field,
- Int32Field: &int32Field,
- Int64Field: &int64Field,
- UintField: &uintField,
- Uint8Field: &uint8Field,
- Uint16Field: &uint16Field,
- Uint32Field: &uint32Field,
- Uint64Field: &uint64Field,
- Float32Field: &float32Field,
- Float64Field: &float64Field,
- SliceField: &sliceField,
- MapField: &mapField,
- ChanField: &chanField,
- FuncField: &funcField,
- StructField: &structField,
- FromStringSliceToStringField: &fromStringSliceToStringField,
- FromTimeToStringField: &fromTimeToStringField,
- FromTimeToTimeField: &fromTimeToTimeField,
- FromStringToTimeField: &fromStringToTimeField,
- }
- fromPointerNil := AssignTagFromPointerField{}
- to := AssignTagTo{}
- toPointer := AssignTagToPointerField{}
- err := assign.DefaultUsage(from, to)
- if err == nil || err.Error() != "返回类型不是结构指针" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出to参数必须是指针类型"))
- }
- err = assign.DefaultUsage(from, toPointer)
- if err == nil || err.Error() != "返回类型不是结构指针" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出to参数必须是指针类型"))
- }
- err = assign.DefaultUsage(fromPointer, to)
- if err == nil || err.Error() != "返回类型不是结构指针" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出to参数必须是指针类型"))
- }
- err = assign.DefaultUsage(fromPointer, toPointer)
- if err == nil || err.Error() != "返回类型不是结构指针" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出to参数必须是指针类型"))
- }
- err = assign.DefaultUsage(from, &to)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- to.checkFields(t, from)
- err = assign.DefaultUsage(from, &toPointer)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- toPointer.checkFields(t, from)
- err = assign.DefaultUsage(fromPointer, &to)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- to.checkFields(t, from)
- err = assign.DefaultUsage(fromPointer, &toPointer)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- toPointer.checkFields(t, from)
- err = assign.DefaultUsage(&from, to)
- if err == nil || err.Error() != "返回类型不是结构指针" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出to参数必须是指针类型"))
- }
- err = assign.DefaultUsage(&from, toPointer)
- if err == nil || err.Error() != "返回类型不是结构指针" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出to参数必须是指针类型"))
- }
- err = assign.DefaultUsage(&fromPointer, to)
- if err == nil || err.Error() != "返回类型不是结构指针" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出to参数必须是指针类型"))
- }
- err = assign.DefaultUsage(&fromPointer, toPointer)
- if err == nil || err.Error() != "返回类型不是结构指针" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出to参数必须是指针类型"))
- }
- err = assign.DefaultUsage(&from, &to)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- to.checkFields(t, from)
- err = assign.DefaultUsage(&from, &toPointer)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- toPointer.checkFields(t, from)
- err = assign.DefaultUsage(&fromPointer, &to)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- to.checkFields(t, from)
- err = assign.DefaultUsage(&fromPointer, &toPointer)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- toPointer.checkFields(t, from)
- err = assign.DefaultUsage(fromPointerNil, &to)
- if err == nil || err.Error() != "无法设置值,请检查是否传递的是非指针结构变量且字段为nil" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出from如果有nil字段,必须使用指针结构变量"))
- }
- err = assign.DefaultUsage(fromPointerNil, &toPointer)
- if err == nil || err.Error() != "无法设置值,请检查是否传递的是非指针结构变量且字段为nil" {
- t.Fatalf("%+v\n", errors.Errorf("没有检测出from如果有nil字段,必须使用指针结构变量"))
- }
- err = assign.DefaultUsage(&fromPointerNil, &to)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- to.checkFields(t, from)
- err = assign.DefaultUsage(&fromPointerNil, &toPointer)
- if err != nil {
- t.Fatalf("%+v\n", err)
- }
- toPointer.checkFields(t, from)
- }
|