| // Copyright 2018 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package impl |
| |
| import ( |
| "reflect" |
| "testing" |
| |
| "github.com/google/go-cmp/cmp" |
| |
| pref "google.golang.org/proto/reflect/protoreflect" |
| ptype "google.golang.org/proto/reflect/prototype" |
| ) |
| |
| type ( |
| MyBool bool |
| MyInt32 int32 |
| MyInt64 int64 |
| MyUint32 uint32 |
| MyUint64 uint64 |
| MyFloat32 float32 |
| MyFloat64 float64 |
| MyString string |
| MyBytes []byte |
| ) |
| |
| type ScalarProto2 struct { |
| Bool *bool `protobuf:"1"` |
| Int32 *int32 `protobuf:"2"` |
| Int64 *int64 `protobuf:"3"` |
| Uint32 *uint32 `protobuf:"4"` |
| Uint64 *uint64 `protobuf:"5"` |
| Float32 *float32 `protobuf:"6"` |
| Float64 *float64 `protobuf:"7"` |
| String *string `protobuf:"8"` |
| StringA []byte `protobuf:"9"` |
| Bytes []byte `protobuf:"10"` |
| BytesA *string `protobuf:"11"` |
| |
| MyBool *MyBool `protobuf:"12"` |
| MyInt32 *MyInt32 `protobuf:"13"` |
| MyInt64 *MyInt64 `protobuf:"14"` |
| MyUint32 *MyUint32 `protobuf:"15"` |
| MyUint64 *MyUint64 `protobuf:"16"` |
| MyFloat32 *MyFloat32 `protobuf:"17"` |
| MyFloat64 *MyFloat64 `protobuf:"18"` |
| MyString *MyString `protobuf:"19"` |
| MyStringA MyBytes `protobuf:"20"` |
| MyBytes MyBytes `protobuf:"21"` |
| MyBytesA *MyString `protobuf:"22"` |
| } |
| |
| type ScalarProto3 struct { |
| Bool bool `protobuf:"1"` |
| Int32 int32 `protobuf:"2"` |
| Int64 int64 `protobuf:"3"` |
| Uint32 uint32 `protobuf:"4"` |
| Uint64 uint64 `protobuf:"5"` |
| Float32 float32 `protobuf:"6"` |
| Float64 float64 `protobuf:"7"` |
| String string `protobuf:"8"` |
| StringA []byte `protobuf:"9"` |
| Bytes []byte `protobuf:"10"` |
| BytesA string `protobuf:"11"` |
| |
| MyBool MyBool `protobuf:"12"` |
| MyInt32 MyInt32 `protobuf:"13"` |
| MyInt64 MyInt64 `protobuf:"14"` |
| MyUint32 MyUint32 `protobuf:"15"` |
| MyUint64 MyUint64 `protobuf:"16"` |
| MyFloat32 MyFloat32 `protobuf:"17"` |
| MyFloat64 MyFloat64 `protobuf:"18"` |
| MyString MyString `protobuf:"19"` |
| MyStringA MyBytes `protobuf:"20"` |
| MyBytes MyBytes `protobuf:"21"` |
| MyBytesA MyString `protobuf:"22"` |
| } |
| |
| func TestFieldFuncs(t *testing.T) { |
| V := pref.ValueOf |
| type ( |
| // has checks that each field matches the list. |
| hasOp []bool |
| // get checks that each field returns values matching the list. |
| getOp []pref.Value |
| // set calls set on each field with the given value in the list. |
| setOp []pref.Value |
| // clear calls clear on each field. |
| clearOp []bool |
| // equal checks that the current message equals the provided value. |
| equalOp struct{ want interface{} } |
| |
| testOp interface{} // has | get | set | clear | equal |
| ) |
| |
| tests := []struct { |
| structType reflect.Type |
| messageDesc ptype.StandaloneMessage |
| testOps []testOp |
| }{{ |
| structType: reflect.TypeOf(ScalarProto2{}), |
| messageDesc: ptype.StandaloneMessage{ |
| Syntax: pref.Proto2, |
| FullName: "ScalarProto2", |
| Fields: []ptype.Field{ |
| {Name: "f1", Number: 1, Cardinality: pref.Optional, Kind: pref.BoolKind, Default: V(bool(true))}, |
| {Name: "f2", Number: 2, Cardinality: pref.Optional, Kind: pref.Int32Kind, Default: V(int32(2))}, |
| {Name: "f3", Number: 3, Cardinality: pref.Optional, Kind: pref.Int64Kind, Default: V(int64(3))}, |
| {Name: "f4", Number: 4, Cardinality: pref.Optional, Kind: pref.Uint32Kind, Default: V(uint32(4))}, |
| {Name: "f5", Number: 5, Cardinality: pref.Optional, Kind: pref.Uint64Kind, Default: V(uint64(5))}, |
| {Name: "f6", Number: 6, Cardinality: pref.Optional, Kind: pref.FloatKind, Default: V(float32(6))}, |
| {Name: "f7", Number: 7, Cardinality: pref.Optional, Kind: pref.DoubleKind, Default: V(float64(7))}, |
| {Name: "f8", Number: 8, Cardinality: pref.Optional, Kind: pref.StringKind, Default: V(string("8"))}, |
| {Name: "f9", Number: 9, Cardinality: pref.Optional, Kind: pref.StringKind, Default: V(string("9"))}, |
| {Name: "f10", Number: 10, Cardinality: pref.Optional, Kind: pref.BytesKind, Default: V([]byte("10"))}, |
| {Name: "f11", Number: 11, Cardinality: pref.Optional, Kind: pref.BytesKind, Default: V([]byte("11"))}, |
| |
| {Name: "f12", Number: 12, Cardinality: pref.Optional, Kind: pref.BoolKind, Default: V(bool(true))}, |
| {Name: "f13", Number: 13, Cardinality: pref.Optional, Kind: pref.Int32Kind, Default: V(int32(13))}, |
| {Name: "f14", Number: 14, Cardinality: pref.Optional, Kind: pref.Int64Kind, Default: V(int64(14))}, |
| {Name: "f15", Number: 15, Cardinality: pref.Optional, Kind: pref.Uint32Kind, Default: V(uint32(15))}, |
| {Name: "f16", Number: 16, Cardinality: pref.Optional, Kind: pref.Uint64Kind, Default: V(uint64(16))}, |
| {Name: "f17", Number: 17, Cardinality: pref.Optional, Kind: pref.FloatKind, Default: V(float32(17))}, |
| {Name: "f18", Number: 18, Cardinality: pref.Optional, Kind: pref.DoubleKind, Default: V(float64(18))}, |
| {Name: "f19", Number: 19, Cardinality: pref.Optional, Kind: pref.StringKind, Default: V(string("19"))}, |
| {Name: "f20", Number: 20, Cardinality: pref.Optional, Kind: pref.StringKind, Default: V(string("20"))}, |
| {Name: "f21", Number: 21, Cardinality: pref.Optional, Kind: pref.BytesKind, Default: V([]byte("21"))}, |
| {Name: "f22", Number: 22, Cardinality: pref.Optional, Kind: pref.BytesKind, Default: V([]byte("22"))}, |
| }, |
| }, |
| testOps: []testOp{ |
| hasOp([]bool{ |
| false, false, false, false, false, false, false, false, false, false, false, |
| false, false, false, false, false, false, false, false, false, false, false, |
| }), |
| getOp([]pref.Value{ |
| V(bool(true)), V(int32(2)), V(int64(3)), V(uint32(4)), V(uint64(5)), V(float32(6)), V(float64(7)), V(string("8")), V(string("9")), V([]byte("10")), V([]byte("11")), |
| V(bool(true)), V(int32(13)), V(int64(14)), V(uint32(15)), V(uint64(16)), V(float32(17)), V(float64(18)), V(string("19")), V(string("20")), V([]byte("21")), V([]byte("22")), |
| }), |
| setOp([]pref.Value{ |
| V(bool(false)), V(int32(0)), V(int64(0)), V(uint32(0)), V(uint64(0)), V(float32(0)), V(float64(0)), V(string("")), V(string("")), V([]byte(nil)), V([]byte(nil)), |
| V(bool(false)), V(int32(0)), V(int64(0)), V(uint32(0)), V(uint64(0)), V(float32(0)), V(float64(0)), V(string("")), V(string("")), V([]byte(nil)), V([]byte(nil)), |
| }), |
| hasOp([]bool{ |
| true, true, true, true, true, true, true, true, true, true, true, |
| true, true, true, true, true, true, true, true, true, true, true, |
| }), |
| equalOp{&ScalarProto2{ |
| new(bool), new(int32), new(int64), new(uint32), new(uint64), new(float32), new(float64), new(string), []byte{}, []byte{}, new(string), |
| new(MyBool), new(MyInt32), new(MyInt64), new(MyUint32), new(MyUint64), new(MyFloat32), new(MyFloat64), new(MyString), MyBytes{}, MyBytes{}, new(MyString), |
| }}, |
| clearOp([]bool{ |
| true, true, true, true, true, true, true, true, true, true, true, |
| true, true, true, true, true, true, true, true, true, true, true, |
| }), |
| equalOp{&ScalarProto2{}}, |
| }, |
| }, { |
| structType: reflect.TypeOf(ScalarProto3{}), |
| messageDesc: ptype.StandaloneMessage{ |
| Syntax: pref.Proto3, |
| FullName: "ScalarProto3", |
| Fields: []ptype.Field{ |
| {Name: "f1", Number: 1, Cardinality: pref.Optional, Kind: pref.BoolKind}, |
| {Name: "f2", Number: 2, Cardinality: pref.Optional, Kind: pref.Int32Kind}, |
| {Name: "f3", Number: 3, Cardinality: pref.Optional, Kind: pref.Int64Kind}, |
| {Name: "f4", Number: 4, Cardinality: pref.Optional, Kind: pref.Uint32Kind}, |
| {Name: "f5", Number: 5, Cardinality: pref.Optional, Kind: pref.Uint64Kind}, |
| {Name: "f6", Number: 6, Cardinality: pref.Optional, Kind: pref.FloatKind}, |
| {Name: "f7", Number: 7, Cardinality: pref.Optional, Kind: pref.DoubleKind}, |
| {Name: "f8", Number: 8, Cardinality: pref.Optional, Kind: pref.StringKind}, |
| {Name: "f9", Number: 9, Cardinality: pref.Optional, Kind: pref.StringKind}, |
| {Name: "f10", Number: 10, Cardinality: pref.Optional, Kind: pref.BytesKind}, |
| {Name: "f11", Number: 11, Cardinality: pref.Optional, Kind: pref.BytesKind}, |
| |
| {Name: "f12", Number: 12, Cardinality: pref.Optional, Kind: pref.BoolKind}, |
| {Name: "f13", Number: 13, Cardinality: pref.Optional, Kind: pref.Int32Kind}, |
| {Name: "f14", Number: 14, Cardinality: pref.Optional, Kind: pref.Int64Kind}, |
| {Name: "f15", Number: 15, Cardinality: pref.Optional, Kind: pref.Uint32Kind}, |
| {Name: "f16", Number: 16, Cardinality: pref.Optional, Kind: pref.Uint64Kind}, |
| {Name: "f17", Number: 17, Cardinality: pref.Optional, Kind: pref.FloatKind}, |
| {Name: "f18", Number: 18, Cardinality: pref.Optional, Kind: pref.DoubleKind}, |
| {Name: "f19", Number: 19, Cardinality: pref.Optional, Kind: pref.StringKind}, |
| {Name: "f20", Number: 20, Cardinality: pref.Optional, Kind: pref.StringKind}, |
| {Name: "f21", Number: 21, Cardinality: pref.Optional, Kind: pref.BytesKind}, |
| {Name: "f22", Number: 22, Cardinality: pref.Optional, Kind: pref.BytesKind}, |
| }, |
| }, |
| testOps: []testOp{ |
| hasOp([]bool{ |
| false, false, false, false, false, false, false, false, false, false, false, |
| false, false, false, false, false, false, false, false, false, false, false, |
| }), |
| getOp([]pref.Value{ |
| V(bool(false)), V(int32(0)), V(int64(0)), V(uint32(0)), V(uint64(0)), V(float32(0)), V(float64(0)), V(string("")), V(string("")), V([]byte(nil)), V([]byte(nil)), |
| V(bool(false)), V(int32(0)), V(int64(0)), V(uint32(0)), V(uint64(0)), V(float32(0)), V(float64(0)), V(string("")), V(string("")), V([]byte(nil)), V([]byte(nil)), |
| }), |
| setOp([]pref.Value{ |
| V(bool(false)), V(int32(0)), V(int64(0)), V(uint32(0)), V(uint64(0)), V(float32(0)), V(float64(0)), V(string("")), V(string("")), V([]byte(nil)), V([]byte(nil)), |
| V(bool(false)), V(int32(0)), V(int64(0)), V(uint32(0)), V(uint64(0)), V(float32(0)), V(float64(0)), V(string("")), V(string("")), V([]byte(nil)), V([]byte(nil)), |
| }), |
| hasOp([]bool{ |
| false, false, false, false, false, false, false, false, false, false, false, |
| false, false, false, false, false, false, false, false, false, false, false, |
| }), |
| equalOp{&ScalarProto3{}}, |
| setOp([]pref.Value{ |
| V(bool(true)), V(int32(2)), V(int64(3)), V(uint32(4)), V(uint64(5)), V(float32(6)), V(float64(7)), V(string("8")), V(string("9")), V([]byte("10")), V([]byte("11")), |
| V(bool(true)), V(int32(13)), V(int64(14)), V(uint32(15)), V(uint64(16)), V(float32(17)), V(float64(18)), V(string("19")), V(string("20")), V([]byte("21")), V([]byte("22")), |
| }), |
| hasOp([]bool{ |
| true, true, true, true, true, true, true, true, true, true, true, |
| true, true, true, true, true, true, true, true, true, true, true, |
| }), |
| equalOp{&ScalarProto3{ |
| true, 2, 3, 4, 5, 6, 7, "8", []byte("9"), []byte("10"), "11", |
| true, 13, 14, 15, 16, 17, 18, "19", []byte("20"), []byte("21"), "22", |
| }}, |
| clearOp([]bool{ |
| true, true, true, true, true, true, true, true, true, true, true, |
| true, true, true, true, true, true, true, true, true, true, true, |
| }), |
| equalOp{&ScalarProto3{}}, |
| }, |
| }} |
| |
| for _, tt := range tests { |
| t.Run(tt.structType.Name(), func(t *testing.T) { |
| // Construct the message descriptor. |
| md, err := ptype.NewMessage(&tt.messageDesc) |
| if err != nil { |
| t.Fatalf("NewMessage error: %v", err) |
| } |
| |
| // Generate the field functions from the message descriptor. |
| var mi MessageInfo |
| mi.generateFieldFuncs(tt.structType, md) // must not panic |
| |
| // Test the field functions. |
| m := reflect.New(tt.structType) |
| p := pointerOfValue(m) |
| for i, op := range tt.testOps { |
| switch op := op.(type) { |
| case hasOp: |
| got := map[pref.FieldNumber]bool{} |
| want := map[pref.FieldNumber]bool{} |
| for j, ok := range op { |
| n := pref.FieldNumber(j + 1) |
| got[n] = mi.fields[n].has(p) |
| want[n] = ok |
| } |
| if diff := cmp.Diff(want, got); diff != "" { |
| t.Errorf("operation %d, has mismatch (-want, +got):\n%s", i, diff) |
| } |
| case getOp: |
| got := map[pref.FieldNumber]pref.Value{} |
| want := map[pref.FieldNumber]pref.Value{} |
| for j, v := range op { |
| n := pref.FieldNumber(j + 1) |
| got[n] = mi.fields[n].get(p) |
| want[n] = v |
| } |
| xformValue := cmp.Transformer("", func(v pref.Value) interface{} { |
| return v.Interface() |
| }) |
| if diff := cmp.Diff(want, got, xformValue); diff != "" { |
| t.Errorf("operation %d, get mismatch (-want, +got):\n%s", i, diff) |
| } |
| case setOp: |
| for j, v := range op { |
| n := pref.FieldNumber(j + 1) |
| mi.fields[n].set(p, v) |
| } |
| case clearOp: |
| for j, ok := range op { |
| n := pref.FieldNumber(j + 1) |
| if ok { |
| mi.fields[n].clear(p) |
| } |
| } |
| case equalOp: |
| got := m.Interface() |
| if diff := cmp.Diff(op.want, got); diff != "" { |
| t.Errorf("operation %d, equal mismatch (-want, +got):\n%s", i, diff) |
| } |
| } |
| } |
| }) |
| } |
| } |
