Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 1 | // Copyright 2018 The Go Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
Joe Tsai | 08e0030 | 2018-11-26 22:32:06 -0800 | [diff] [blame] | 5 | package impl_test |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 6 | |
| 7 | import ( |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 8 | "fmt" |
| 9 | "math" |
Damien Neil | 8012b44 | 2019-01-18 09:32:24 -0800 | [diff] [blame] | 10 | "reflect" |
Joe Tsai | 82760ce | 2019-06-20 03:09:57 -0700 | [diff] [blame] | 11 | "runtime" |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 12 | "strings" |
Joe Tsai | 82760ce | 2019-06-20 03:09:57 -0700 | [diff] [blame] | 13 | "sync" |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 14 | "testing" |
| 15 | |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 16 | cmp "github.com/google/go-cmp/cmp" |
| 17 | cmpopts "github.com/google/go-cmp/cmp/cmpopts" |
Joe Tsai | e0daf31 | 2020-02-25 12:51:10 -0800 | [diff] [blame] | 18 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 19 | "google.golang.org/protobuf/encoding/prototext" |
Damien Neil | e89e624 | 2019-05-13 23:55:40 -0700 | [diff] [blame] | 20 | pimpl "google.golang.org/protobuf/internal/impl" |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 21 | "google.golang.org/protobuf/proto" |
| 22 | pdesc "google.golang.org/protobuf/reflect/protodesc" |
Damien Neil | e89e624 | 2019-05-13 23:55:40 -0700 | [diff] [blame] | 23 | pref "google.golang.org/protobuf/reflect/protoreflect" |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 24 | "google.golang.org/protobuf/reflect/protoregistry" |
Joe Tsai | 5d63473 | 2020-07-26 22:37:47 -0700 | [diff] [blame] | 25 | "google.golang.org/protobuf/testing/protopack" |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 26 | |
Joe Tsai | 55f1825 | 2020-01-11 00:25:01 -0800 | [diff] [blame] | 27 | proto2_20180125 "google.golang.org/protobuf/internal/testprotos/legacy/proto2_20180125_92554152" |
Joe Tsai | 82760ce | 2019-06-20 03:09:57 -0700 | [diff] [blame] | 28 | testpb "google.golang.org/protobuf/internal/testprotos/test" |
Joe Tsai | a95b29f | 2019-05-16 12:47:20 -0700 | [diff] [blame] | 29 | "google.golang.org/protobuf/types/descriptorpb" |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 30 | ) |
| 31 | |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 32 | // List of test operations to perform on messages, lists, or maps. |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 33 | type ( |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 34 | messageOp interface{ isMessageOp() } |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 35 | messageOps []messageOp |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 36 | |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 37 | listOp interface{ isListOp() } |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 38 | listOps []listOp |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 39 | |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 40 | mapOp interface{ isMapOp() } |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 41 | mapOps []mapOp |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 42 | ) |
| 43 | |
| 44 | // Test operations performed on a message. |
| 45 | type ( |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 46 | // check that the message contents match |
| 47 | equalMessage struct{ pref.Message } |
| 48 | // check presence for specific fields in the message |
| 49 | hasFields map[pref.FieldNumber]bool |
| 50 | // check that specific message fields match |
| 51 | getFields map[pref.FieldNumber]pref.Value |
| 52 | // set specific message fields |
| 53 | setFields map[pref.FieldNumber]pref.Value |
| 54 | // clear specific fields in the message |
| 55 | clearFields []pref.FieldNumber |
Joe Tsai | 4ec39c7 | 2019-04-03 13:40:53 -0700 | [diff] [blame] | 56 | // check for the presence of specific oneof member fields. |
| 57 | whichOneofs map[pref.Name]pref.FieldNumber |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 58 | // apply messageOps on each specified message field |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 59 | messageFields map[pref.FieldNumber]messageOps |
| 60 | messageFieldsMutable map[pref.FieldNumber]messageOps |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 61 | // apply listOps on each specified list field |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 62 | listFields map[pref.FieldNumber]listOps |
| 63 | listFieldsMutable map[pref.FieldNumber]listOps |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 64 | // apply mapOps on each specified map fields |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 65 | mapFields map[pref.FieldNumber]mapOps |
| 66 | mapFieldsMutable map[pref.FieldNumber]mapOps |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 67 | // range through all fields and check that they match |
| 68 | rangeFields map[pref.FieldNumber]pref.Value |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 69 | ) |
| 70 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 71 | func (equalMessage) isMessageOp() {} |
| 72 | func (hasFields) isMessageOp() {} |
| 73 | func (getFields) isMessageOp() {} |
| 74 | func (setFields) isMessageOp() {} |
| 75 | func (clearFields) isMessageOp() {} |
| 76 | func (whichOneofs) isMessageOp() {} |
| 77 | func (messageFields) isMessageOp() {} |
| 78 | func (messageFieldsMutable) isMessageOp() {} |
| 79 | func (listFields) isMessageOp() {} |
| 80 | func (listFieldsMutable) isMessageOp() {} |
| 81 | func (mapFields) isMessageOp() {} |
| 82 | func (mapFieldsMutable) isMessageOp() {} |
| 83 | func (rangeFields) isMessageOp() {} |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 84 | |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 85 | // Test operations performed on a list. |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 86 | type ( |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 87 | // check that the list contents match |
| 88 | equalList struct{ pref.List } |
| 89 | // check that list length matches |
| 90 | lenList int |
| 91 | // check that specific list entries match |
| 92 | getList map[int]pref.Value |
| 93 | // set specific list entries |
| 94 | setList map[int]pref.Value |
| 95 | // append entries to the list |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 96 | appendList []pref.Value |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 97 | // apply messageOps on a newly appended message |
| 98 | appendMessageList messageOps |
| 99 | // truncate the list to the specified length |
| 100 | truncList int |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 101 | ) |
| 102 | |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 103 | func (equalList) isListOp() {} |
| 104 | func (lenList) isListOp() {} |
| 105 | func (getList) isListOp() {} |
| 106 | func (setList) isListOp() {} |
| 107 | func (appendList) isListOp() {} |
| 108 | func (appendMessageList) isListOp() {} |
| 109 | func (truncList) isListOp() {} |
| 110 | |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 111 | // Test operations performed on a map. |
| 112 | type ( |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 113 | // check that the map contents match |
| 114 | equalMap struct{ pref.Map } |
| 115 | // check that map length matches |
| 116 | lenMap int |
| 117 | // check presence for specific entries in the map |
| 118 | hasMap map[interface{}]bool |
| 119 | // check that specific map entries match |
| 120 | getMap map[interface{}]pref.Value |
| 121 | // set specific map entries |
| 122 | setMap map[interface{}]pref.Value |
| 123 | // clear specific entries in the map |
| 124 | clearMap []interface{} |
| 125 | // apply messageOps on each specified message entry |
| 126 | messageMap map[interface{}]messageOps |
| 127 | // range through all entries and check that they match |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 128 | rangeMap map[interface{}]pref.Value |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 129 | ) |
| 130 | |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 131 | func (equalMap) isMapOp() {} |
| 132 | func (lenMap) isMapOp() {} |
| 133 | func (hasMap) isMapOp() {} |
| 134 | func (getMap) isMapOp() {} |
| 135 | func (setMap) isMapOp() {} |
| 136 | func (clearMap) isMapOp() {} |
| 137 | func (messageMap) isMapOp() {} |
| 138 | func (rangeMap) isMapOp() {} |
| 139 | |
Joe Tsai | ce6edd3 | 2018-10-19 16:27:46 -0700 | [diff] [blame] | 140 | type ScalarProto2 struct { |
| 141 | Bool *bool `protobuf:"1"` |
| 142 | Int32 *int32 `protobuf:"2"` |
| 143 | Int64 *int64 `protobuf:"3"` |
| 144 | Uint32 *uint32 `protobuf:"4"` |
| 145 | Uint64 *uint64 `protobuf:"5"` |
| 146 | Float32 *float32 `protobuf:"6"` |
| 147 | Float64 *float64 `protobuf:"7"` |
| 148 | String *string `protobuf:"8"` |
| 149 | StringA []byte `protobuf:"9"` |
| 150 | Bytes []byte `protobuf:"10"` |
| 151 | BytesA *string `protobuf:"11"` |
| 152 | |
| 153 | MyBool *MyBool `protobuf:"12"` |
| 154 | MyInt32 *MyInt32 `protobuf:"13"` |
| 155 | MyInt64 *MyInt64 `protobuf:"14"` |
| 156 | MyUint32 *MyUint32 `protobuf:"15"` |
| 157 | MyUint64 *MyUint64 `protobuf:"16"` |
| 158 | MyFloat32 *MyFloat32 `protobuf:"17"` |
| 159 | MyFloat64 *MyFloat64 `protobuf:"18"` |
| 160 | MyString *MyString `protobuf:"19"` |
| 161 | MyStringA MyBytes `protobuf:"20"` |
| 162 | MyBytes MyBytes `protobuf:"21"` |
| 163 | MyBytesA *MyString `protobuf:"22"` |
| 164 | } |
| 165 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 166 | func mustMakeEnumDesc(path string, syntax pref.Syntax, enumDesc string) pref.EnumDescriptor { |
| 167 | s := fmt.Sprintf(`name:%q syntax:%q enum_type:[{%s}]`, path, syntax, enumDesc) |
| 168 | pb := new(descriptorpb.FileDescriptorProto) |
| 169 | if err := prototext.Unmarshal([]byte(s), pb); err != nil { |
| 170 | panic(err) |
| 171 | } |
| 172 | fd, err := pdesc.NewFile(pb, nil) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 173 | if err != nil { |
| 174 | panic(err) |
| 175 | } |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 176 | return fd.Enums().Get(0) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 177 | } |
| 178 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 179 | func mustMakeMessageDesc(path string, syntax pref.Syntax, fileDesc, msgDesc string, r pdesc.Resolver) pref.MessageDescriptor { |
| 180 | s := fmt.Sprintf(`name:%q syntax:%q %s message_type:[{%s}]`, path, syntax, fileDesc, msgDesc) |
| 181 | pb := new(descriptorpb.FileDescriptorProto) |
| 182 | if err := prototext.Unmarshal([]byte(s), pb); err != nil { |
| 183 | panic(err) |
| 184 | } |
| 185 | fd, err := pdesc.NewFile(pb, r) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 186 | if err != nil { |
| 187 | panic(err) |
| 188 | } |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 189 | return fd.Messages().Get(0) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 190 | } |
| 191 | |
| 192 | var V = pref.ValueOf |
| 193 | var VE = func(n pref.EnumNumber) pref.Value { return V(n) } |
| 194 | |
| 195 | type ( |
| 196 | MyBool bool |
| 197 | MyInt32 int32 |
| 198 | MyInt64 int64 |
| 199 | MyUint32 uint32 |
| 200 | MyUint64 uint64 |
| 201 | MyFloat32 float32 |
| 202 | MyFloat64 float64 |
| 203 | MyString string |
| 204 | MyBytes []byte |
| 205 | |
| 206 | ListStrings []MyString |
| 207 | ListBytes []MyBytes |
| 208 | |
| 209 | MapStrings map[MyString]MyString |
| 210 | MapBytes map[MyString]MyBytes |
| 211 | ) |
| 212 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 213 | var scalarProto2Type = pimpl.MessageInfo{GoReflectType: reflect.TypeOf(new(ScalarProto2)), Desc: mustMakeMessageDesc("scalar2.proto", pref.Proto2, "", ` |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 214 | name: "ScalarProto2" |
| 215 | field: [ |
| 216 | {name:"f1" number:1 label:LABEL_OPTIONAL type:TYPE_BOOL default_value:"true"}, |
| 217 | {name:"f2" number:2 label:LABEL_OPTIONAL type:TYPE_INT32 default_value:"2"}, |
| 218 | {name:"f3" number:3 label:LABEL_OPTIONAL type:TYPE_INT64 default_value:"3"}, |
| 219 | {name:"f4" number:4 label:LABEL_OPTIONAL type:TYPE_UINT32 default_value:"4"}, |
| 220 | {name:"f5" number:5 label:LABEL_OPTIONAL type:TYPE_UINT64 default_value:"5"}, |
| 221 | {name:"f6" number:6 label:LABEL_OPTIONAL type:TYPE_FLOAT default_value:"6"}, |
| 222 | {name:"f7" number:7 label:LABEL_OPTIONAL type:TYPE_DOUBLE default_value:"7"}, |
| 223 | {name:"f8" number:8 label:LABEL_OPTIONAL type:TYPE_STRING default_value:"8"}, |
| 224 | {name:"f9" number:9 label:LABEL_OPTIONAL type:TYPE_STRING default_value:"9"}, |
| 225 | {name:"f10" number:10 label:LABEL_OPTIONAL type:TYPE_BYTES default_value:"10"}, |
| 226 | {name:"f11" number:11 label:LABEL_OPTIONAL type:TYPE_BYTES default_value:"11"}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 227 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 228 | {name:"f12" number:12 label:LABEL_OPTIONAL type:TYPE_BOOL default_value:"true"}, |
| 229 | {name:"f13" number:13 label:LABEL_OPTIONAL type:TYPE_INT32 default_value:"13"}, |
| 230 | {name:"f14" number:14 label:LABEL_OPTIONAL type:TYPE_INT64 default_value:"14"}, |
| 231 | {name:"f15" number:15 label:LABEL_OPTIONAL type:TYPE_UINT32 default_value:"15"}, |
| 232 | {name:"f16" number:16 label:LABEL_OPTIONAL type:TYPE_UINT64 default_value:"16"}, |
| 233 | {name:"f17" number:17 label:LABEL_OPTIONAL type:TYPE_FLOAT default_value:"17"}, |
| 234 | {name:"f18" number:18 label:LABEL_OPTIONAL type:TYPE_DOUBLE default_value:"18"}, |
| 235 | {name:"f19" number:19 label:LABEL_OPTIONAL type:TYPE_STRING default_value:"19"}, |
| 236 | {name:"f20" number:20 label:LABEL_OPTIONAL type:TYPE_STRING default_value:"20"}, |
| 237 | {name:"f21" number:21 label:LABEL_OPTIONAL type:TYPE_BYTES default_value:"21"}, |
| 238 | {name:"f22" number:22 label:LABEL_OPTIONAL type:TYPE_BYTES default_value:"22"} |
| 239 | ] |
| 240 | `, nil), |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 241 | } |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 242 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 243 | func (m *ScalarProto2) ProtoReflect() pref.Message { return scalarProto2Type.MessageOf(m) } |
Joe Tsai | f0c01e4 | 2018-11-06 13:05:20 -0800 | [diff] [blame] | 244 | |
| 245 | func TestScalarProto2(t *testing.T) { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 246 | testMessage(t, nil, new(ScalarProto2).ProtoReflect(), messageOps{ |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 247 | hasFields{ |
| 248 | 1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, |
| 249 | 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false, 20: false, 21: false, 22: false, |
| 250 | }, |
| 251 | getFields{ |
| 252 | 1: V(bool(true)), 2: V(int32(2)), 3: V(int64(3)), 4: V(uint32(4)), 5: V(uint64(5)), 6: V(float32(6)), 7: V(float64(7)), 8: V(string("8")), 9: V(string("9")), 10: V([]byte("10")), 11: V([]byte("11")), |
| 253 | 12: V(bool(true)), 13: V(int32(13)), 14: V(int64(14)), 15: V(uint32(15)), 16: V(uint64(16)), 17: V(float32(17)), 18: V(float64(18)), 19: V(string("19")), 20: V(string("20")), 21: V([]byte("21")), 22: V([]byte("22")), |
| 254 | }, |
| 255 | setFields{ |
| 256 | 1: V(bool(false)), 2: V(int32(0)), 3: V(int64(0)), 4: V(uint32(0)), 5: V(uint64(0)), 6: V(float32(0)), 7: V(float64(0)), 8: V(string("")), 9: V(string("")), 10: V([]byte(nil)), 11: V([]byte(nil)), |
| 257 | 12: V(bool(false)), 13: V(int32(0)), 14: V(int64(0)), 15: V(uint32(0)), 16: V(uint64(0)), 17: V(float32(0)), 18: V(float64(0)), 19: V(string("")), 20: V(string("")), 21: V([]byte(nil)), 22: V([]byte(nil)), |
| 258 | }, |
| 259 | hasFields{ |
| 260 | 1: true, 2: true, 3: true, 4: true, 5: true, 6: true, 7: true, 8: true, 9: true, 10: true, 11: true, |
| 261 | 12: true, 13: true, 14: true, 15: true, 16: true, 17: true, 18: true, 19: true, 20: true, 21: true, 22: true, |
| 262 | }, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 263 | equalMessage{(&ScalarProto2{ |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 264 | new(bool), new(int32), new(int64), new(uint32), new(uint64), new(float32), new(float64), new(string), []byte{}, []byte{}, new(string), |
| 265 | new(MyBool), new(MyInt32), new(MyInt64), new(MyUint32), new(MyUint64), new(MyFloat32), new(MyFloat64), new(MyString), MyBytes{}, MyBytes{}, new(MyString), |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 266 | }).ProtoReflect()}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 267 | clearFields{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 268 | equalMessage{new(ScalarProto2).ProtoReflect()}, |
Joe Tsai | 3e6a39b | 2019-07-10 10:21:12 -0700 | [diff] [blame] | 269 | |
| 270 | // Setting a bytes field nil empty bytes should preserve presence. |
| 271 | setFields{10: V([]byte(nil)), 11: V([]byte(nil)), 21: V([]byte(nil)), 22: V([]byte(nil))}, |
| 272 | getFields{10: V([]byte{}), 11: V([]byte(nil)), 21: V([]byte{}), 22: V([]byte(nil))}, |
| 273 | hasFields{10: true, 11: true, 21: true, 22: true}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 274 | }) |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 275 | |
| 276 | // Test read-only operations on nil message. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 277 | testMessage(t, nil, (*ScalarProto2)(nil).ProtoReflect(), messageOps{ |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 278 | hasFields{ |
| 279 | 1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, |
| 280 | 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false, 20: false, 21: false, 22: false, |
| 281 | }, |
| 282 | getFields{ |
| 283 | 1: V(bool(true)), 2: V(int32(2)), 3: V(int64(3)), 4: V(uint32(4)), 5: V(uint64(5)), 6: V(float32(6)), 7: V(float64(7)), 8: V(string("8")), 9: V(string("9")), 10: V([]byte("10")), 11: V([]byte("11")), |
| 284 | 12: V(bool(true)), 13: V(int32(13)), 14: V(int64(14)), 15: V(uint32(15)), 16: V(uint64(16)), 17: V(float32(17)), 18: V(float64(18)), 19: V(string("19")), 20: V(string("20")), 21: V([]byte("21")), 22: V([]byte("22")), |
| 285 | }, |
| 286 | }) |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 287 | } |
| 288 | |
Joe Tsai | ce6edd3 | 2018-10-19 16:27:46 -0700 | [diff] [blame] | 289 | type ScalarProto3 struct { |
| 290 | Bool bool `protobuf:"1"` |
| 291 | Int32 int32 `protobuf:"2"` |
| 292 | Int64 int64 `protobuf:"3"` |
| 293 | Uint32 uint32 `protobuf:"4"` |
| 294 | Uint64 uint64 `protobuf:"5"` |
| 295 | Float32 float32 `protobuf:"6"` |
| 296 | Float64 float64 `protobuf:"7"` |
| 297 | String string `protobuf:"8"` |
| 298 | StringA []byte `protobuf:"9"` |
| 299 | Bytes []byte `protobuf:"10"` |
| 300 | BytesA string `protobuf:"11"` |
| 301 | |
| 302 | MyBool MyBool `protobuf:"12"` |
| 303 | MyInt32 MyInt32 `protobuf:"13"` |
| 304 | MyInt64 MyInt64 `protobuf:"14"` |
| 305 | MyUint32 MyUint32 `protobuf:"15"` |
| 306 | MyUint64 MyUint64 `protobuf:"16"` |
| 307 | MyFloat32 MyFloat32 `protobuf:"17"` |
| 308 | MyFloat64 MyFloat64 `protobuf:"18"` |
| 309 | MyString MyString `protobuf:"19"` |
| 310 | MyStringA MyBytes `protobuf:"20"` |
| 311 | MyBytes MyBytes `protobuf:"21"` |
| 312 | MyBytesA MyString `protobuf:"22"` |
| 313 | } |
| 314 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 315 | var scalarProto3Type = pimpl.MessageInfo{GoReflectType: reflect.TypeOf(new(ScalarProto3)), Desc: mustMakeMessageDesc("scalar3.proto", pref.Proto3, "", ` |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 316 | name: "ScalarProto3" |
| 317 | field: [ |
| 318 | {name:"f1" number:1 label:LABEL_OPTIONAL type:TYPE_BOOL}, |
| 319 | {name:"f2" number:2 label:LABEL_OPTIONAL type:TYPE_INT32}, |
| 320 | {name:"f3" number:3 label:LABEL_OPTIONAL type:TYPE_INT64}, |
| 321 | {name:"f4" number:4 label:LABEL_OPTIONAL type:TYPE_UINT32}, |
| 322 | {name:"f5" number:5 label:LABEL_OPTIONAL type:TYPE_UINT64}, |
| 323 | {name:"f6" number:6 label:LABEL_OPTIONAL type:TYPE_FLOAT}, |
| 324 | {name:"f7" number:7 label:LABEL_OPTIONAL type:TYPE_DOUBLE}, |
| 325 | {name:"f8" number:8 label:LABEL_OPTIONAL type:TYPE_STRING}, |
| 326 | {name:"f9" number:9 label:LABEL_OPTIONAL type:TYPE_STRING}, |
| 327 | {name:"f10" number:10 label:LABEL_OPTIONAL type:TYPE_BYTES}, |
| 328 | {name:"f11" number:11 label:LABEL_OPTIONAL type:TYPE_BYTES}, |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 329 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 330 | {name:"f12" number:12 label:LABEL_OPTIONAL type:TYPE_BOOL}, |
| 331 | {name:"f13" number:13 label:LABEL_OPTIONAL type:TYPE_INT32}, |
| 332 | {name:"f14" number:14 label:LABEL_OPTIONAL type:TYPE_INT64}, |
| 333 | {name:"f15" number:15 label:LABEL_OPTIONAL type:TYPE_UINT32}, |
| 334 | {name:"f16" number:16 label:LABEL_OPTIONAL type:TYPE_UINT64}, |
| 335 | {name:"f17" number:17 label:LABEL_OPTIONAL type:TYPE_FLOAT}, |
| 336 | {name:"f18" number:18 label:LABEL_OPTIONAL type:TYPE_DOUBLE}, |
| 337 | {name:"f19" number:19 label:LABEL_OPTIONAL type:TYPE_STRING}, |
| 338 | {name:"f20" number:20 label:LABEL_OPTIONAL type:TYPE_STRING}, |
| 339 | {name:"f21" number:21 label:LABEL_OPTIONAL type:TYPE_BYTES}, |
| 340 | {name:"f22" number:22 label:LABEL_OPTIONAL type:TYPE_BYTES} |
| 341 | ] |
| 342 | `, nil), |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 343 | } |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 344 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 345 | func (m *ScalarProto3) ProtoReflect() pref.Message { return scalarProto3Type.MessageOf(m) } |
Joe Tsai | f0c01e4 | 2018-11-06 13:05:20 -0800 | [diff] [blame] | 346 | |
| 347 | func TestScalarProto3(t *testing.T) { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 348 | testMessage(t, nil, new(ScalarProto3).ProtoReflect(), messageOps{ |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 349 | hasFields{ |
| 350 | 1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, |
| 351 | 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false, 20: false, 21: false, 22: false, |
| 352 | }, |
| 353 | getFields{ |
| 354 | 1: V(bool(false)), 2: V(int32(0)), 3: V(int64(0)), 4: V(uint32(0)), 5: V(uint64(0)), 6: V(float32(0)), 7: V(float64(0)), 8: V(string("")), 9: V(string("")), 10: V([]byte(nil)), 11: V([]byte(nil)), |
| 355 | 12: V(bool(false)), 13: V(int32(0)), 14: V(int64(0)), 15: V(uint32(0)), 16: V(uint64(0)), 17: V(float32(0)), 18: V(float64(0)), 19: V(string("")), 20: V(string("")), 21: V([]byte(nil)), 22: V([]byte(nil)), |
| 356 | }, |
| 357 | setFields{ |
| 358 | 1: V(bool(false)), 2: V(int32(0)), 3: V(int64(0)), 4: V(uint32(0)), 5: V(uint64(0)), 6: V(float32(0)), 7: V(float64(0)), 8: V(string("")), 9: V(string("")), 10: V([]byte(nil)), 11: V([]byte(nil)), |
| 359 | 12: V(bool(false)), 13: V(int32(0)), 14: V(int64(0)), 15: V(uint32(0)), 16: V(uint64(0)), 17: V(float32(0)), 18: V(float64(0)), 19: V(string("")), 20: V(string("")), 21: V([]byte(nil)), 22: V([]byte(nil)), |
| 360 | }, |
| 361 | hasFields{ |
| 362 | 1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, |
| 363 | 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false, 20: false, 21: false, 22: false, |
| 364 | }, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 365 | equalMessage{new(ScalarProto3).ProtoReflect()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 366 | setFields{ |
| 367 | 1: V(bool(true)), 2: V(int32(2)), 3: V(int64(3)), 4: V(uint32(4)), 5: V(uint64(5)), 6: V(float32(6)), 7: V(float64(7)), 8: V(string("8")), 9: V(string("9")), 10: V([]byte("10")), 11: V([]byte("11")), |
| 368 | 12: V(bool(true)), 13: V(int32(13)), 14: V(int64(14)), 15: V(uint32(15)), 16: V(uint64(16)), 17: V(float32(17)), 18: V(float64(18)), 19: V(string("19")), 20: V(string("20")), 21: V([]byte("21")), 22: V([]byte("22")), |
| 369 | }, |
| 370 | hasFields{ |
| 371 | 1: true, 2: true, 3: true, 4: true, 5: true, 6: true, 7: true, 8: true, 9: true, 10: true, 11: true, |
| 372 | 12: true, 13: true, 14: true, 15: true, 16: true, 17: true, 18: true, 19: true, 20: true, 21: true, 22: true, |
| 373 | }, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 374 | equalMessage{(&ScalarProto3{ |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 375 | true, 2, 3, 4, 5, 6, 7, "8", []byte("9"), []byte("10"), "11", |
| 376 | true, 13, 14, 15, 16, 17, 18, "19", []byte("20"), []byte("21"), "22", |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 377 | }).ProtoReflect()}, |
Joe Tsai | 44e389c | 2018-11-19 15:27:09 -0800 | [diff] [blame] | 378 | setFields{ |
| 379 | 2: V(int32(-2)), 3: V(int64(-3)), 6: V(float32(math.Inf(-1))), 7: V(float64(math.NaN())), |
| 380 | }, |
| 381 | hasFields{ |
| 382 | 2: true, 3: true, 6: true, 7: true, |
| 383 | }, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 384 | clearFields{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 385 | equalMessage{new(ScalarProto3).ProtoReflect()}, |
Joe Tsai | 060cdac | 2019-04-22 11:44:49 -0700 | [diff] [blame] | 386 | |
| 387 | // Verify that -0 triggers proper Has behavior. |
| 388 | hasFields{6: false, 7: false}, |
| 389 | setFields{6: V(float32(math.Copysign(0, -1))), 7: V(float64(math.Copysign(0, -1)))}, |
| 390 | hasFields{6: true, 7: true}, |
Joe Tsai | 3e6a39b | 2019-07-10 10:21:12 -0700 | [diff] [blame] | 391 | |
| 392 | // Setting a bytes field to non-nil empty bytes should not preserve presence. |
| 393 | setFields{10: V([]byte{}), 11: V([]byte{}), 21: V([]byte{}), 22: V([]byte{})}, |
| 394 | getFields{10: V([]byte(nil)), 11: V([]byte(nil)), 21: V([]byte(nil)), 22: V([]byte(nil))}, |
| 395 | hasFields{10: false, 11: false, 21: false, 22: false}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 396 | }) |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 397 | |
| 398 | // Test read-only operations on nil message. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 399 | testMessage(t, nil, (*ScalarProto3)(nil).ProtoReflect(), messageOps{ |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 400 | hasFields{ |
| 401 | 1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, |
| 402 | 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false, 20: false, 21: false, 22: false, |
| 403 | }, |
| 404 | getFields{ |
| 405 | 1: V(bool(false)), 2: V(int32(0)), 3: V(int64(0)), 4: V(uint32(0)), 5: V(uint64(0)), 6: V(float32(0)), 7: V(float64(0)), 8: V(string("")), 9: V(string("")), 10: V([]byte(nil)), 11: V([]byte(nil)), |
| 406 | 12: V(bool(false)), 13: V(int32(0)), 14: V(int64(0)), 15: V(uint32(0)), 16: V(uint64(0)), 17: V(float32(0)), 18: V(float64(0)), 19: V(string("")), 20: V(string("")), 21: V([]byte(nil)), 22: V([]byte(nil)), |
| 407 | }, |
| 408 | }) |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 409 | } |
| 410 | |
Joe Tsai | f0c01e4 | 2018-11-06 13:05:20 -0800 | [diff] [blame] | 411 | type ListScalars struct { |
Joe Tsai | ce6edd3 | 2018-10-19 16:27:46 -0700 | [diff] [blame] | 412 | Bools []bool `protobuf:"1"` |
| 413 | Int32s []int32 `protobuf:"2"` |
| 414 | Int64s []int64 `protobuf:"3"` |
| 415 | Uint32s []uint32 `protobuf:"4"` |
| 416 | Uint64s []uint64 `protobuf:"5"` |
| 417 | Float32s []float32 `protobuf:"6"` |
| 418 | Float64s []float64 `protobuf:"7"` |
| 419 | Strings []string `protobuf:"8"` |
| 420 | StringsA [][]byte `protobuf:"9"` |
| 421 | Bytes [][]byte `protobuf:"10"` |
| 422 | BytesA []string `protobuf:"11"` |
| 423 | |
| 424 | MyStrings1 []MyString `protobuf:"12"` |
| 425 | MyStrings2 []MyBytes `protobuf:"13"` |
| 426 | MyBytes1 []MyBytes `protobuf:"14"` |
| 427 | MyBytes2 []MyString `protobuf:"15"` |
| 428 | |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 429 | MyStrings3 ListStrings `protobuf:"16"` |
| 430 | MyStrings4 ListBytes `protobuf:"17"` |
| 431 | MyBytes3 ListBytes `protobuf:"18"` |
| 432 | MyBytes4 ListStrings `protobuf:"19"` |
Joe Tsai | ce6edd3 | 2018-10-19 16:27:46 -0700 | [diff] [blame] | 433 | } |
| 434 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 435 | var listScalarsType = pimpl.MessageInfo{GoReflectType: reflect.TypeOf(new(ListScalars)), Desc: mustMakeMessageDesc("list-scalars.proto", pref.Proto2, "", ` |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 436 | name: "ListScalars" |
| 437 | field: [ |
| 438 | {name:"f1" number:1 label:LABEL_REPEATED type:TYPE_BOOL}, |
| 439 | {name:"f2" number:2 label:LABEL_REPEATED type:TYPE_INT32}, |
| 440 | {name:"f3" number:3 label:LABEL_REPEATED type:TYPE_INT64}, |
| 441 | {name:"f4" number:4 label:LABEL_REPEATED type:TYPE_UINT32}, |
| 442 | {name:"f5" number:5 label:LABEL_REPEATED type:TYPE_UINT64}, |
| 443 | {name:"f6" number:6 label:LABEL_REPEATED type:TYPE_FLOAT}, |
| 444 | {name:"f7" number:7 label:LABEL_REPEATED type:TYPE_DOUBLE}, |
| 445 | {name:"f8" number:8 label:LABEL_REPEATED type:TYPE_STRING}, |
| 446 | {name:"f9" number:9 label:LABEL_REPEATED type:TYPE_STRING}, |
| 447 | {name:"f10" number:10 label:LABEL_REPEATED type:TYPE_BYTES}, |
| 448 | {name:"f11" number:11 label:LABEL_REPEATED type:TYPE_BYTES}, |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 449 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 450 | {name:"f12" number:12 label:LABEL_REPEATED type:TYPE_STRING}, |
| 451 | {name:"f13" number:13 label:LABEL_REPEATED type:TYPE_STRING}, |
| 452 | {name:"f14" number:14 label:LABEL_REPEATED type:TYPE_BYTES}, |
| 453 | {name:"f15" number:15 label:LABEL_REPEATED type:TYPE_BYTES}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 454 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 455 | {name:"f16" number:16 label:LABEL_REPEATED type:TYPE_STRING}, |
| 456 | {name:"f17" number:17 label:LABEL_REPEATED type:TYPE_STRING}, |
| 457 | {name:"f18" number:18 label:LABEL_REPEATED type:TYPE_BYTES}, |
| 458 | {name:"f19" number:19 label:LABEL_REPEATED type:TYPE_BYTES} |
| 459 | ] |
| 460 | `, nil), |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 461 | } |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 462 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 463 | func (m *ListScalars) ProtoReflect() pref.Message { return listScalarsType.MessageOf(m) } |
Joe Tsai | f0c01e4 | 2018-11-06 13:05:20 -0800 | [diff] [blame] | 464 | |
| 465 | func TestListScalars(t *testing.T) { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 466 | empty := new(ListScalars).ProtoReflect() |
| 467 | want := (&ListScalars{ |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 468 | Bools: []bool{true, false, true}, |
| 469 | Int32s: []int32{2, math.MinInt32, math.MaxInt32}, |
| 470 | Int64s: []int64{3, math.MinInt64, math.MaxInt64}, |
| 471 | Uint32s: []uint32{4, math.MaxUint32 / 2, math.MaxUint32}, |
| 472 | Uint64s: []uint64{5, math.MaxUint64 / 2, math.MaxUint64}, |
| 473 | Float32s: []float32{6, math.SmallestNonzeroFloat32, float32(math.NaN()), math.MaxFloat32}, |
| 474 | Float64s: []float64{7, math.SmallestNonzeroFloat64, float64(math.NaN()), math.MaxFloat64}, |
| 475 | Strings: []string{"8", "", "eight"}, |
| 476 | StringsA: [][]byte{[]byte("9"), nil, []byte("nine")}, |
| 477 | Bytes: [][]byte{[]byte("10"), nil, []byte("ten")}, |
| 478 | BytesA: []string{"11", "", "eleven"}, |
| 479 | |
| 480 | MyStrings1: []MyString{"12", "", "twelve"}, |
| 481 | MyStrings2: []MyBytes{[]byte("13"), nil, []byte("thirteen")}, |
| 482 | MyBytes1: []MyBytes{[]byte("14"), nil, []byte("fourteen")}, |
| 483 | MyBytes2: []MyString{"15", "", "fifteen"}, |
| 484 | |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 485 | MyStrings3: ListStrings{"16", "", "sixteen"}, |
| 486 | MyStrings4: ListBytes{[]byte("17"), nil, []byte("seventeen")}, |
| 487 | MyBytes3: ListBytes{[]byte("18"), nil, []byte("eighteen")}, |
| 488 | MyBytes4: ListStrings{"19", "", "nineteen"}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 489 | }).ProtoReflect() |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 490 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 491 | testMessage(t, nil, new(ListScalars).ProtoReflect(), messageOps{ |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 492 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 493 | getFields{1: getField(empty, 1), 3: getField(empty, 3), 5: getField(empty, 5), 7: getField(empty, 7), 9: getField(empty, 9), 11: getField(empty, 11), 13: getField(empty, 13), 15: getField(empty, 15), 17: getField(empty, 17), 19: getField(empty, 19)}, |
| 494 | setFields{1: getField(want, 1), 3: getField(want, 3), 5: getField(want, 5), 7: getField(want, 7), 9: getField(want, 9), 11: getField(want, 11), 13: getField(want, 13), 15: getField(want, 15), 17: getField(want, 17), 19: getField(want, 19)}, |
| 495 | listFieldsMutable{ |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 496 | 2: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 497 | lenList(0), |
| 498 | appendList{V(int32(2)), V(int32(math.MinInt32)), V(int32(math.MaxInt32))}, |
| 499 | getList{0: V(int32(2)), 1: V(int32(math.MinInt32)), 2: V(int32(math.MaxInt32))}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 500 | equalList{getField(want, 2).List()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 501 | }, |
| 502 | 4: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 503 | appendList{V(uint32(0)), V(uint32(0)), V(uint32(0))}, |
| 504 | setList{0: V(uint32(4)), 1: V(uint32(math.MaxUint32 / 2)), 2: V(uint32(math.MaxUint32))}, |
| 505 | lenList(3), |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 506 | }, |
| 507 | 6: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 508 | appendList{V(float32(6)), V(float32(math.SmallestNonzeroFloat32)), V(float32(math.NaN())), V(float32(math.MaxFloat32))}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 509 | equalList{getField(want, 6).List()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 510 | }, |
| 511 | 8: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 512 | appendList{V(""), V(""), V(""), V(""), V(""), V("")}, |
| 513 | lenList(6), |
| 514 | setList{0: V("8"), 2: V("eight")}, |
| 515 | truncList(3), |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 516 | equalList{getField(want, 8).List()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 517 | }, |
| 518 | 10: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 519 | appendList{V([]byte(nil)), V([]byte(nil))}, |
| 520 | setList{0: V([]byte("10"))}, |
| 521 | appendList{V([]byte("wrong"))}, |
| 522 | setList{2: V([]byte("ten"))}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 523 | equalList{getField(want, 10).List()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 524 | }, |
| 525 | 12: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 526 | appendList{V("12"), V("wrong"), V("twelve")}, |
| 527 | setList{1: V("")}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 528 | equalList{getField(want, 12).List()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 529 | }, |
| 530 | 14: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 531 | appendList{V([]byte("14")), V([]byte(nil)), V([]byte("fourteen"))}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 532 | equalList{getField(want, 14).List()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 533 | }, |
| 534 | 16: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 535 | appendList{V("16"), V(""), V("sixteen"), V("extra")}, |
| 536 | truncList(3), |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 537 | equalList{getField(want, 16).List()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 538 | }, |
| 539 | 18: { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 540 | appendList{V([]byte("18")), V([]byte(nil)), V([]byte("eighteen"))}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 541 | equalList{getField(want, 18).List()}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 542 | }, |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 543 | }, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 544 | hasFields{1: true, 2: true, 3: true, 4: true, 5: true, 6: true, 7: true, 8: true, 9: true, 10: true, 11: true, 12: true, 13: true, 14: true, 15: true, 16: true, 17: true, 18: true, 19: true}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 545 | equalMessage{want}, |
| 546 | clearFields{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19}, |
| 547 | equalMessage{empty}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 548 | }) |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 549 | |
| 550 | // Test read-only operations on nil message. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 551 | testMessage(t, nil, (*ListScalars)(nil).ProtoReflect(), messageOps{ |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 552 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false}, |
| 553 | listFields{2: {lenList(0)}, 4: {lenList(0)}, 6: {lenList(0)}, 8: {lenList(0)}, 10: {lenList(0)}, 12: {lenList(0)}, 14: {lenList(0)}, 16: {lenList(0)}, 18: {lenList(0)}}, |
| 554 | }) |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 555 | } |
| 556 | |
Joe Tsai | ce6edd3 | 2018-10-19 16:27:46 -0700 | [diff] [blame] | 557 | type MapScalars struct { |
| 558 | KeyBools map[bool]string `protobuf:"1"` |
| 559 | KeyInt32s map[int32]string `protobuf:"2"` |
| 560 | KeyInt64s map[int64]string `protobuf:"3"` |
| 561 | KeyUint32s map[uint32]string `protobuf:"4"` |
| 562 | KeyUint64s map[uint64]string `protobuf:"5"` |
| 563 | KeyStrings map[string]string `protobuf:"6"` |
| 564 | |
| 565 | ValBools map[string]bool `protobuf:"7"` |
| 566 | ValInt32s map[string]int32 `protobuf:"8"` |
| 567 | ValInt64s map[string]int64 `protobuf:"9"` |
| 568 | ValUint32s map[string]uint32 `protobuf:"10"` |
| 569 | ValUint64s map[string]uint64 `protobuf:"11"` |
| 570 | ValFloat32s map[string]float32 `protobuf:"12"` |
| 571 | ValFloat64s map[string]float64 `protobuf:"13"` |
| 572 | ValStrings map[string]string `protobuf:"14"` |
| 573 | ValStringsA map[string][]byte `protobuf:"15"` |
| 574 | ValBytes map[string][]byte `protobuf:"16"` |
| 575 | ValBytesA map[string]string `protobuf:"17"` |
| 576 | |
| 577 | MyStrings1 map[MyString]MyString `protobuf:"18"` |
| 578 | MyStrings2 map[MyString]MyBytes `protobuf:"19"` |
| 579 | MyBytes1 map[MyString]MyBytes `protobuf:"20"` |
| 580 | MyBytes2 map[MyString]MyString `protobuf:"21"` |
| 581 | |
| 582 | MyStrings3 MapStrings `protobuf:"22"` |
| 583 | MyStrings4 MapBytes `protobuf:"23"` |
| 584 | MyBytes3 MapBytes `protobuf:"24"` |
| 585 | MyBytes4 MapStrings `protobuf:"25"` |
| 586 | } |
| 587 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 588 | var mapScalarsType = pimpl.MessageInfo{GoReflectType: reflect.TypeOf(new(MapScalars)), Desc: mustMakeMessageDesc("map-scalars.proto", pref.Proto2, "", ` |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 589 | name: "MapScalars" |
| 590 | field: [ |
| 591 | {name:"f1" number:1 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F1Entry"}, |
| 592 | {name:"f2" number:2 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F2Entry"}, |
| 593 | {name:"f3" number:3 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F3Entry"}, |
| 594 | {name:"f4" number:4 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F4Entry"}, |
| 595 | {name:"f5" number:5 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F5Entry"}, |
| 596 | {name:"f6" number:6 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F6Entry"}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 597 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 598 | {name:"f7" number:7 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F7Entry"}, |
| 599 | {name:"f8" number:8 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F8Entry"}, |
| 600 | {name:"f9" number:9 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F9Entry"}, |
| 601 | {name:"f10" number:10 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F10Entry"}, |
| 602 | {name:"f11" number:11 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F11Entry"}, |
| 603 | {name:"f12" number:12 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F12Entry"}, |
| 604 | {name:"f13" number:13 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F13Entry"}, |
| 605 | {name:"f14" number:14 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F14Entry"}, |
| 606 | {name:"f15" number:15 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F15Entry"}, |
| 607 | {name:"f16" number:16 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F16Entry"}, |
| 608 | {name:"f17" number:17 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F17Entry"}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 609 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 610 | {name:"f18" number:18 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F18Entry"}, |
| 611 | {name:"f19" number:19 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F19Entry"}, |
| 612 | {name:"f20" number:20 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F20Entry"}, |
| 613 | {name:"f21" number:21 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F21Entry"}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 614 | |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 615 | {name:"f22" number:22 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F22Entry"}, |
| 616 | {name:"f23" number:23 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F23Entry"}, |
| 617 | {name:"f24" number:24 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F24Entry"}, |
| 618 | {name:"f25" number:25 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".MapScalars.F25Entry"} |
| 619 | ] |
| 620 | nested_type: [ |
| 621 | {name:"F1Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_BOOL}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 622 | {name:"F2Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_INT32}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 623 | {name:"F3Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_INT64}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 624 | {name:"F4Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_UINT32}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 625 | {name:"F5Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_UINT64}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 626 | {name:"F6Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 627 | |
| 628 | {name:"F7Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_BOOL}] options:{map_entry:true}}, |
| 629 | {name:"F8Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_INT32}] options:{map_entry:true}}, |
| 630 | {name:"F9Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_INT64}] options:{map_entry:true}}, |
| 631 | {name:"F10Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_UINT32}] options:{map_entry:true}}, |
| 632 | {name:"F11Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_UINT64}] options:{map_entry:true}}, |
| 633 | {name:"F12Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_FLOAT}] options:{map_entry:true}}, |
| 634 | {name:"F13Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_DOUBLE}] options:{map_entry:true}}, |
| 635 | {name:"F14Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 636 | {name:"F15Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 637 | {name:"F16Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_BYTES}] options:{map_entry:true}}, |
| 638 | {name:"F17Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_BYTES}] options:{map_entry:true}}, |
| 639 | |
| 640 | {name:"F18Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 641 | {name:"F19Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 642 | {name:"F20Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_BYTES}] options:{map_entry:true}}, |
| 643 | {name:"F21Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_BYTES}] options:{map_entry:true}}, |
| 644 | |
| 645 | {name:"F22Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 646 | {name:"F23Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_STRING}] options:{map_entry:true}}, |
| 647 | {name:"F24Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_BYTES}] options:{map_entry:true}}, |
| 648 | {name:"F25Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_BYTES}] options:{map_entry:true}} |
| 649 | ] |
| 650 | `, nil), |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 651 | } |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 652 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 653 | func (m *MapScalars) ProtoReflect() pref.Message { return mapScalarsType.MessageOf(m) } |
Joe Tsai | f0c01e4 | 2018-11-06 13:05:20 -0800 | [diff] [blame] | 654 | |
| 655 | func TestMapScalars(t *testing.T) { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 656 | empty := new(MapScalars).ProtoReflect() |
| 657 | want := (&MapScalars{ |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 658 | KeyBools: map[bool]string{true: "true", false: "false"}, |
| 659 | KeyInt32s: map[int32]string{0: "zero", -1: "one", 2: "two"}, |
| 660 | KeyInt64s: map[int64]string{0: "zero", -10: "ten", 20: "twenty"}, |
| 661 | KeyUint32s: map[uint32]string{0: "zero", 1: "one", 2: "two"}, |
| 662 | KeyUint64s: map[uint64]string{0: "zero", 10: "ten", 20: "twenty"}, |
| 663 | KeyStrings: map[string]string{"": "", "foo": "bar"}, |
| 664 | |
| 665 | ValBools: map[string]bool{"true": true, "false": false}, |
| 666 | ValInt32s: map[string]int32{"one": 1, "two": 2, "three": 3}, |
| 667 | ValInt64s: map[string]int64{"ten": 10, "twenty": -20, "thirty": 30}, |
| 668 | ValUint32s: map[string]uint32{"0x00": 0x00, "0xff": 0xff, "0xdead": 0xdead}, |
| 669 | ValUint64s: map[string]uint64{"0x00": 0x00, "0xff": 0xff, "0xdead": 0xdead}, |
| 670 | ValFloat32s: map[string]float32{"nan": float32(math.NaN()), "pi": float32(math.Pi)}, |
| 671 | ValFloat64s: map[string]float64{"nan": float64(math.NaN()), "pi": float64(math.Pi)}, |
| 672 | ValStrings: map[string]string{"s1": "s1", "s2": "s2"}, |
| 673 | ValStringsA: map[string][]byte{"s1": []byte("s1"), "s2": []byte("s2")}, |
| 674 | ValBytes: map[string][]byte{"s1": []byte("s1"), "s2": []byte("s2")}, |
| 675 | ValBytesA: map[string]string{"s1": "s1", "s2": "s2"}, |
| 676 | |
| 677 | MyStrings1: map[MyString]MyString{"s1": "s1", "s2": "s2"}, |
| 678 | MyStrings2: map[MyString]MyBytes{"s1": []byte("s1"), "s2": []byte("s2")}, |
| 679 | MyBytes1: map[MyString]MyBytes{"s1": []byte("s1"), "s2": []byte("s2")}, |
| 680 | MyBytes2: map[MyString]MyString{"s1": "s1", "s2": "s2"}, |
| 681 | |
| 682 | MyStrings3: MapStrings{"s1": "s1", "s2": "s2"}, |
| 683 | MyStrings4: MapBytes{"s1": []byte("s1"), "s2": []byte("s2")}, |
| 684 | MyBytes3: MapBytes{"s1": []byte("s1"), "s2": []byte("s2")}, |
| 685 | MyBytes4: MapStrings{"s1": "s1", "s2": "s2"}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 686 | }).ProtoReflect() |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 687 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 688 | testMessage(t, nil, new(MapScalars).ProtoReflect(), messageOps{ |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 689 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false, 20: false, 21: false, 22: false, 23: false, 24: false, 25: false}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 690 | getFields{1: getField(empty, 1), 3: getField(empty, 3), 5: getField(empty, 5), 7: getField(empty, 7), 9: getField(empty, 9), 11: getField(empty, 11), 13: getField(empty, 13), 15: getField(empty, 15), 17: getField(empty, 17), 19: getField(empty, 19), 21: getField(empty, 21), 23: getField(empty, 23), 25: getField(empty, 25)}, |
| 691 | setFields{1: getField(want, 1), 3: getField(want, 3), 5: getField(want, 5), 7: getField(want, 7), 9: getField(want, 9), 11: getField(want, 11), 13: getField(want, 13), 15: getField(want, 15), 17: getField(want, 17), 19: getField(want, 19), 21: getField(want, 21), 23: getField(want, 23), 25: getField(want, 25)}, |
| 692 | mapFieldsMutable{ |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 693 | 2: { |
| 694 | lenMap(0), |
| 695 | hasMap{int32(0): false, int32(-1): false, int32(2): false}, |
| 696 | setMap{int32(0): V("zero")}, |
| 697 | lenMap(1), |
| 698 | hasMap{int32(0): true, int32(-1): false, int32(2): false}, |
| 699 | setMap{int32(-1): V("one")}, |
| 700 | lenMap(2), |
| 701 | hasMap{int32(0): true, int32(-1): true, int32(2): false}, |
| 702 | setMap{int32(2): V("two")}, |
| 703 | lenMap(3), |
| 704 | hasMap{int32(0): true, int32(-1): true, int32(2): true}, |
| 705 | }, |
| 706 | 4: { |
| 707 | setMap{uint32(0): V("zero"), uint32(1): V("one"), uint32(2): V("two")}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 708 | equalMap{getField(want, 4).Map()}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 709 | }, |
| 710 | 6: { |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 711 | clearMap{"noexist"}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 712 | setMap{"foo": V("bar")}, |
| 713 | setMap{"": V("empty")}, |
| 714 | getMap{"": V("empty"), "foo": V("bar"), "noexist": V(nil)}, |
| 715 | setMap{"": V(""), "extra": V("extra")}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 716 | clearMap{"extra", "noexist"}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 717 | }, |
| 718 | 8: { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 719 | equalMap{getField(empty, 8).Map()}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 720 | setMap{"one": V(int32(1)), "two": V(int32(2)), "three": V(int32(3))}, |
| 721 | }, |
| 722 | 10: { |
| 723 | setMap{"0x00": V(uint32(0x00)), "0xff": V(uint32(0xff)), "0xdead": V(uint32(0xdead))}, |
| 724 | lenMap(3), |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 725 | equalMap{getField(want, 10).Map()}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 726 | getMap{"0x00": V(uint32(0x00)), "0xff": V(uint32(0xff)), "0xdead": V(uint32(0xdead)), "0xdeadbeef": V(nil)}, |
| 727 | }, |
| 728 | 12: { |
| 729 | setMap{"nan": V(float32(math.NaN())), "pi": V(float32(math.Pi)), "e": V(float32(math.E))}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 730 | clearMap{"e", "phi"}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 731 | rangeMap{"nan": V(float32(math.NaN())), "pi": V(float32(math.Pi))}, |
| 732 | }, |
| 733 | 14: { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 734 | equalMap{getField(empty, 14).Map()}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 735 | setMap{"s1": V("s1"), "s2": V("s2")}, |
| 736 | }, |
| 737 | 16: { |
| 738 | setMap{"s1": V([]byte("s1")), "s2": V([]byte("s2"))}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 739 | equalMap{getField(want, 16).Map()}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 740 | }, |
| 741 | 18: { |
| 742 | hasMap{"s1": false, "s2": false, "s3": false}, |
| 743 | setMap{"s1": V("s1"), "s2": V("s2")}, |
| 744 | hasMap{"s1": true, "s2": true, "s3": false}, |
| 745 | }, |
| 746 | 20: { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 747 | equalMap{getField(empty, 20).Map()}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 748 | setMap{"s1": V([]byte("s1")), "s2": V([]byte("s2"))}, |
| 749 | }, |
| 750 | 22: { |
| 751 | rangeMap{}, |
| 752 | setMap{"s1": V("s1"), "s2": V("s2")}, |
| 753 | rangeMap{"s1": V("s1"), "s2": V("s2")}, |
| 754 | lenMap(2), |
| 755 | }, |
| 756 | 24: { |
| 757 | setMap{"s1": V([]byte("s1")), "s2": V([]byte("s2"))}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 758 | equalMap{getField(want, 24).Map()}, |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 759 | }, |
| 760 | }, |
| 761 | hasFields{1: true, 2: true, 3: true, 4: true, 5: true, 6: true, 7: true, 8: true, 9: true, 10: true, 11: true, 12: true, 13: true, 14: true, 15: true, 16: true, 17: true, 18: true, 19: true, 20: true, 21: true, 22: true, 23: true, 24: true, 25: true}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 762 | equalMessage{want}, |
| 763 | clearFields{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25}, |
| 764 | equalMessage{empty}, |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 765 | }) |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 766 | |
| 767 | // Test read-only operations on nil message. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 768 | testMessage(t, nil, (*MapScalars)(nil).ProtoReflect(), messageOps{ |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 769 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false, 13: false, 14: false, 15: false, 16: false, 17: false, 18: false, 19: false, 20: false, 21: false, 22: false, 23: false, 24: false, 25: false}, |
| 770 | mapFields{2: {lenMap(0)}, 4: {lenMap(0)}, 6: {lenMap(0)}, 8: {lenMap(0)}, 10: {lenMap(0)}, 12: {lenMap(0)}, 14: {lenMap(0)}, 16: {lenMap(0)}, 18: {lenMap(0)}, 20: {lenMap(0)}, 22: {lenMap(0)}, 24: {lenMap(0)}}, |
| 771 | }) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 772 | } |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 773 | |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 774 | type OneofScalars struct { |
| 775 | Union isOneofScalars_Union `protobuf_oneof:"union"` |
| 776 | } |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 777 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 778 | var oneofScalarsType = pimpl.MessageInfo{GoReflectType: reflect.TypeOf(new(OneofScalars)), Desc: mustMakeMessageDesc("oneof-scalars.proto", pref.Proto2, "", ` |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 779 | name: "OneofScalars" |
| 780 | field: [ |
| 781 | {name:"f1" number:1 label:LABEL_OPTIONAL type:TYPE_BOOL default_value:"true" oneof_index:0}, |
| 782 | {name:"f2" number:2 label:LABEL_OPTIONAL type:TYPE_INT32 default_value:"2" oneof_index:0}, |
| 783 | {name:"f3" number:3 label:LABEL_OPTIONAL type:TYPE_INT64 default_value:"3" oneof_index:0}, |
| 784 | {name:"f4" number:4 label:LABEL_OPTIONAL type:TYPE_UINT32 default_value:"4" oneof_index:0}, |
| 785 | {name:"f5" number:5 label:LABEL_OPTIONAL type:TYPE_UINT64 default_value:"5" oneof_index:0}, |
| 786 | {name:"f6" number:6 label:LABEL_OPTIONAL type:TYPE_FLOAT default_value:"6" oneof_index:0}, |
| 787 | {name:"f7" number:7 label:LABEL_OPTIONAL type:TYPE_DOUBLE default_value:"7" oneof_index:0}, |
| 788 | {name:"f8" number:8 label:LABEL_OPTIONAL type:TYPE_STRING default_value:"8" oneof_index:0}, |
| 789 | {name:"f9" number:9 label:LABEL_OPTIONAL type:TYPE_STRING default_value:"9" oneof_index:0}, |
| 790 | {name:"f10" number:10 label:LABEL_OPTIONAL type:TYPE_STRING default_value:"10" oneof_index:0}, |
| 791 | {name:"f11" number:11 label:LABEL_OPTIONAL type:TYPE_BYTES default_value:"11" oneof_index:0}, |
| 792 | {name:"f12" number:12 label:LABEL_OPTIONAL type:TYPE_BYTES default_value:"12" oneof_index:0}, |
| 793 | {name:"f13" number:13 label:LABEL_OPTIONAL type:TYPE_BYTES default_value:"13" oneof_index:0} |
| 794 | ] |
| 795 | oneof_decl: [{name:"union"}] |
| 796 | `, nil), |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 797 | } |
Joe Tsai | f0c01e4 | 2018-11-06 13:05:20 -0800 | [diff] [blame] | 798 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 799 | func (m *OneofScalars) ProtoReflect() pref.Message { return oneofScalarsType.MessageOf(m) } |
Joe Tsai | f0c01e4 | 2018-11-06 13:05:20 -0800 | [diff] [blame] | 800 | |
Joe Tsai | f18ab53 | 2018-11-27 17:25:04 -0800 | [diff] [blame] | 801 | func (*OneofScalars) XXX_OneofWrappers() []interface{} { |
| 802 | return []interface{}{ |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 803 | (*OneofScalars_Bool)(nil), |
| 804 | (*OneofScalars_Int32)(nil), |
| 805 | (*OneofScalars_Int64)(nil), |
| 806 | (*OneofScalars_Uint32)(nil), |
| 807 | (*OneofScalars_Uint64)(nil), |
| 808 | (*OneofScalars_Float32)(nil), |
| 809 | (*OneofScalars_Float64)(nil), |
| 810 | (*OneofScalars_String)(nil), |
| 811 | (*OneofScalars_StringA)(nil), |
| 812 | (*OneofScalars_StringB)(nil), |
| 813 | (*OneofScalars_Bytes)(nil), |
| 814 | (*OneofScalars_BytesA)(nil), |
| 815 | (*OneofScalars_BytesB)(nil), |
| 816 | } |
| 817 | } |
| 818 | |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 819 | type ( |
| 820 | isOneofScalars_Union interface { |
| 821 | isOneofScalars_Union() |
| 822 | } |
| 823 | OneofScalars_Bool struct { |
| 824 | Bool bool `protobuf:"1"` |
| 825 | } |
| 826 | OneofScalars_Int32 struct { |
| 827 | Int32 MyInt32 `protobuf:"2"` |
| 828 | } |
| 829 | OneofScalars_Int64 struct { |
| 830 | Int64 int64 `protobuf:"3"` |
| 831 | } |
| 832 | OneofScalars_Uint32 struct { |
| 833 | Uint32 MyUint32 `protobuf:"4"` |
| 834 | } |
| 835 | OneofScalars_Uint64 struct { |
| 836 | Uint64 uint64 `protobuf:"5"` |
| 837 | } |
| 838 | OneofScalars_Float32 struct { |
| 839 | Float32 MyFloat32 `protobuf:"6"` |
| 840 | } |
| 841 | OneofScalars_Float64 struct { |
| 842 | Float64 float64 `protobuf:"7"` |
| 843 | } |
| 844 | OneofScalars_String struct { |
| 845 | String string `protobuf:"8"` |
| 846 | } |
| 847 | OneofScalars_StringA struct { |
| 848 | StringA []byte `protobuf:"9"` |
| 849 | } |
| 850 | OneofScalars_StringB struct { |
| 851 | StringB MyString `protobuf:"10"` |
| 852 | } |
| 853 | OneofScalars_Bytes struct { |
| 854 | Bytes []byte `protobuf:"11"` |
| 855 | } |
| 856 | OneofScalars_BytesA struct { |
| 857 | BytesA string `protobuf:"12"` |
| 858 | } |
| 859 | OneofScalars_BytesB struct { |
| 860 | BytesB MyBytes `protobuf:"13"` |
| 861 | } |
| 862 | ) |
| 863 | |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 864 | func (*OneofScalars_Bool) isOneofScalars_Union() {} |
| 865 | func (*OneofScalars_Int32) isOneofScalars_Union() {} |
| 866 | func (*OneofScalars_Int64) isOneofScalars_Union() {} |
| 867 | func (*OneofScalars_Uint32) isOneofScalars_Union() {} |
| 868 | func (*OneofScalars_Uint64) isOneofScalars_Union() {} |
| 869 | func (*OneofScalars_Float32) isOneofScalars_Union() {} |
| 870 | func (*OneofScalars_Float64) isOneofScalars_Union() {} |
| 871 | func (*OneofScalars_String) isOneofScalars_Union() {} |
| 872 | func (*OneofScalars_StringA) isOneofScalars_Union() {} |
| 873 | func (*OneofScalars_StringB) isOneofScalars_Union() {} |
| 874 | func (*OneofScalars_Bytes) isOneofScalars_Union() {} |
| 875 | func (*OneofScalars_BytesA) isOneofScalars_Union() {} |
| 876 | func (*OneofScalars_BytesB) isOneofScalars_Union() {} |
| 877 | |
| 878 | func TestOneofs(t *testing.T) { |
Joe Tsai | f0c01e4 | 2018-11-06 13:05:20 -0800 | [diff] [blame] | 879 | empty := &OneofScalars{} |
| 880 | want1 := &OneofScalars{Union: &OneofScalars_Bool{true}} |
| 881 | want2 := &OneofScalars{Union: &OneofScalars_Int32{20}} |
| 882 | want3 := &OneofScalars{Union: &OneofScalars_Int64{30}} |
| 883 | want4 := &OneofScalars{Union: &OneofScalars_Uint32{40}} |
| 884 | want5 := &OneofScalars{Union: &OneofScalars_Uint64{50}} |
| 885 | want6 := &OneofScalars{Union: &OneofScalars_Float32{60}} |
| 886 | want7 := &OneofScalars{Union: &OneofScalars_Float64{70}} |
| 887 | want8 := &OneofScalars{Union: &OneofScalars_String{string("80")}} |
| 888 | want9 := &OneofScalars{Union: &OneofScalars_StringA{[]byte("90")}} |
| 889 | want10 := &OneofScalars{Union: &OneofScalars_StringB{MyString("100")}} |
| 890 | want11 := &OneofScalars{Union: &OneofScalars_Bytes{[]byte("110")}} |
| 891 | want12 := &OneofScalars{Union: &OneofScalars_BytesA{string("120")}} |
| 892 | want13 := &OneofScalars{Union: &OneofScalars_BytesB{MyBytes("130")}} |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 893 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 894 | testMessage(t, nil, new(OneofScalars).ProtoReflect(), messageOps{ |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 895 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false, 13: false}, |
| 896 | getFields{1: V(bool(true)), 2: V(int32(2)), 3: V(int64(3)), 4: V(uint32(4)), 5: V(uint64(5)), 6: V(float32(6)), 7: V(float64(7)), 8: V(string("8")), 9: V(string("9")), 10: V(string("10")), 11: V([]byte("11")), 12: V([]byte("12")), 13: V([]byte("13"))}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 897 | whichOneofs{"union": 0}, |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 898 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 899 | setFields{1: V(bool(true))}, hasFields{1: true}, equalMessage{want1.ProtoReflect()}, |
| 900 | setFields{2: V(int32(20))}, hasFields{2: true}, equalMessage{want2.ProtoReflect()}, |
| 901 | setFields{3: V(int64(30))}, hasFields{3: true}, equalMessage{want3.ProtoReflect()}, |
| 902 | setFields{4: V(uint32(40))}, hasFields{4: true}, equalMessage{want4.ProtoReflect()}, |
| 903 | setFields{5: V(uint64(50))}, hasFields{5: true}, equalMessage{want5.ProtoReflect()}, |
| 904 | setFields{6: V(float32(60))}, hasFields{6: true}, equalMessage{want6.ProtoReflect()}, |
| 905 | setFields{7: V(float64(70))}, hasFields{7: true}, equalMessage{want7.ProtoReflect()}, |
Joe Tsai | 4ec39c7 | 2019-04-03 13:40:53 -0700 | [diff] [blame] | 906 | |
| 907 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: true, 8: false, 9: false, 10: false, 11: false, 12: false, 13: false}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 908 | whichOneofs{"union": 7}, |
Joe Tsai | 4ec39c7 | 2019-04-03 13:40:53 -0700 | [diff] [blame] | 909 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 910 | setFields{8: V(string("80"))}, hasFields{8: true}, equalMessage{want8.ProtoReflect()}, |
| 911 | setFields{9: V(string("90"))}, hasFields{9: true}, equalMessage{want9.ProtoReflect()}, |
| 912 | setFields{10: V(string("100"))}, hasFields{10: true}, equalMessage{want10.ProtoReflect()}, |
| 913 | setFields{11: V([]byte("110"))}, hasFields{11: true}, equalMessage{want11.ProtoReflect()}, |
| 914 | setFields{12: V([]byte("120"))}, hasFields{12: true}, equalMessage{want12.ProtoReflect()}, |
| 915 | setFields{13: V([]byte("130"))}, hasFields{13: true}, equalMessage{want13.ProtoReflect()}, |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 916 | |
| 917 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false, 13: true}, |
| 918 | getFields{1: V(bool(true)), 2: V(int32(2)), 3: V(int64(3)), 4: V(uint32(4)), 5: V(uint64(5)), 6: V(float32(6)), 7: V(float64(7)), 8: V(string("8")), 9: V(string("9")), 10: V(string("10")), 11: V([]byte("11")), 12: V([]byte("12")), 13: V([]byte("130"))}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 919 | clearFields{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 920 | whichOneofs{"union": 13}, |
| 921 | equalMessage{want13.ProtoReflect()}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 922 | clearFields{13}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 923 | whichOneofs{"union": 0}, |
| 924 | equalMessage{empty.ProtoReflect()}, |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 925 | }) |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 926 | |
| 927 | // Test read-only operations on nil message. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 928 | testMessage(t, nil, (*OneofScalars)(nil).ProtoReflect(), messageOps{ |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 929 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false, 13: false}, |
| 930 | getFields{1: V(bool(true)), 2: V(int32(2)), 3: V(int64(3)), 4: V(uint32(4)), 5: V(uint64(5)), 6: V(float32(6)), 7: V(float64(7)), 8: V(string("8")), 9: V(string("9")), 10: V(string("10")), 11: V([]byte("11")), 12: V([]byte("12")), 13: V([]byte("13"))}, |
| 931 | }) |
Joe Tsai | 2c870bb | 2018-10-17 11:46:52 -0700 | [diff] [blame] | 932 | } |
| 933 | |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 934 | type EnumProto2 int32 |
| 935 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 936 | var enumProto2Desc = mustMakeEnumDesc("enum2.proto", pref.Proto2, ` |
| 937 | name: "EnumProto2" |
| 938 | value: [{name:"DEAD" number:0xdead}, {name:"BEEF" number:0xbeef}] |
| 939 | `) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 940 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 941 | func (e EnumProto2) Descriptor() pref.EnumDescriptor { return enumProto2Desc } |
| 942 | func (e EnumProto2) Type() pref.EnumType { return e } |
Joe Tsai | 0fc49f8 | 2019-05-01 12:29:25 -0700 | [diff] [blame] | 943 | func (e EnumProto2) Enum() *EnumProto2 { return &e } |
| 944 | func (e EnumProto2) Number() pref.EnumNumber { return pref.EnumNumber(e) } |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 945 | func (t EnumProto2) New(n pref.EnumNumber) pref.Enum { return EnumProto2(n) } |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 946 | |
| 947 | type EnumProto3 int32 |
| 948 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 949 | var enumProto3Desc = mustMakeEnumDesc("enum3.proto", pref.Proto3, ` |
| 950 | name: "EnumProto3", |
| 951 | value: [{name:"ALPHA" number:0}, {name:"BRAVO" number:1}] |
| 952 | `) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 953 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 954 | func (e EnumProto3) Descriptor() pref.EnumDescriptor { return enumProto3Desc } |
| 955 | func (e EnumProto3) Type() pref.EnumType { return e } |
Joe Tsai | 0fc49f8 | 2019-05-01 12:29:25 -0700 | [diff] [blame] | 956 | func (e EnumProto3) Enum() *EnumProto3 { return &e } |
| 957 | func (e EnumProto3) Number() pref.EnumNumber { return pref.EnumNumber(e) } |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 958 | func (t EnumProto3) New(n pref.EnumNumber) pref.Enum { return EnumProto3(n) } |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 959 | |
| 960 | type EnumMessages struct { |
| 961 | EnumP2 *EnumProto2 `protobuf:"1"` |
| 962 | EnumP3 *EnumProto3 `protobuf:"2"` |
| 963 | MessageLegacy *proto2_20180125.Message `protobuf:"3"` |
| 964 | MessageCycle *EnumMessages `protobuf:"4"` |
| 965 | EnumList []EnumProto2 `protobuf:"5"` |
| 966 | MessageList []*ScalarProto2 `protobuf:"6"` |
| 967 | EnumMap map[string]EnumProto3 `protobuf:"7"` |
| 968 | MessageMap map[string]*ScalarProto3 `protobuf:"8"` |
| 969 | Union isEnumMessages_Union `protobuf_oneof:"union"` |
| 970 | } |
| 971 | |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 972 | var enumMessagesType = pimpl.MessageInfo{GoReflectType: reflect.TypeOf(new(EnumMessages)), Desc: mustMakeMessageDesc("enum-messages.proto", pref.Proto2, ` |
Joe Tsai | 55f1825 | 2020-01-11 00:25:01 -0800 | [diff] [blame] | 973 | dependency: ["enum2.proto", "enum3.proto", "scalar2.proto", "scalar3.proto", "proto2_20180125_92554152/test.proto"] |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 974 | `, ` |
| 975 | name: "EnumMessages" |
| 976 | field: [ |
| 977 | {name:"f1" number:1 label:LABEL_OPTIONAL type:TYPE_ENUM type_name:".EnumProto2" default_value:"BEEF"}, |
| 978 | {name:"f2" number:2 label:LABEL_OPTIONAL type:TYPE_ENUM type_name:".EnumProto3" default_value:"BRAVO"}, |
Herbie Ong | 20aefe9 | 2019-06-24 19:21:46 -0700 | [diff] [blame] | 979 | {name:"f3" number:3 label:LABEL_OPTIONAL type:TYPE_MESSAGE type_name:".google.golang.org.proto2_20180125.Message"}, |
| 980 | {name:"f4" number:4 label:LABEL_OPTIONAL type:TYPE_MESSAGE type_name:".EnumMessages"}, |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 981 | {name:"f5" number:5 label:LABEL_REPEATED type:TYPE_ENUM type_name:".EnumProto2"}, |
| 982 | {name:"f6" number:6 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".ScalarProto2"}, |
| 983 | {name:"f7" number:7 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".EnumMessages.F7Entry"}, |
| 984 | {name:"f8" number:8 label:LABEL_REPEATED type:TYPE_MESSAGE type_name:".EnumMessages.F8Entry"}, |
| 985 | {name:"f9" number:9 label:LABEL_OPTIONAL type:TYPE_ENUM type_name:".EnumProto2" oneof_index:0 default_value:"BEEF"}, |
| 986 | {name:"f10" number:10 label:LABEL_OPTIONAL type:TYPE_ENUM type_name:".EnumProto3" oneof_index:0 default_value:"BRAVO"}, |
| 987 | {name:"f11" number:11 label:LABEL_OPTIONAL type:TYPE_MESSAGE type_name:".ScalarProto2" oneof_index:0}, |
| 988 | {name:"f12" number:12 label:LABEL_OPTIONAL type:TYPE_MESSAGE type_name:".ScalarProto3" oneof_index:0} |
| 989 | ] |
| 990 | oneof_decl: [{name:"union"}] |
| 991 | nested_type: [ |
| 992 | {name:"F7Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_ENUM type_name:".EnumProto3"}] options:{map_entry:true}}, |
| 993 | {name:"F8Entry" field:[{name:"key" number:1 label:LABEL_OPTIONAL type:TYPE_STRING}, {name:"value" number:2 label:LABEL_OPTIONAL type:TYPE_MESSAGE type_name:".ScalarProto3"}] options:{map_entry:true}} |
| 994 | ] |
Damien Neil | c826885 | 2019-10-08 13:28:53 -0700 | [diff] [blame] | 995 | `, newFileRegistry( |
Damien Neil | 16163b4 | 2019-08-06 15:43:25 -0700 | [diff] [blame] | 996 | EnumProto2(0).Descriptor().ParentFile(), |
| 997 | EnumProto3(0).Descriptor().ParentFile(), |
| 998 | ((*ScalarProto2)(nil)).ProtoReflect().Descriptor().ParentFile(), |
| 999 | ((*ScalarProto3)(nil)).ProtoReflect().Descriptor().ParentFile(), |
| 1000 | pimpl.Export{}.MessageDescriptorOf((*proto2_20180125.Message)(nil)).ParentFile(), |
| 1001 | )), |
| 1002 | } |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1003 | |
Damien Neil | c826885 | 2019-10-08 13:28:53 -0700 | [diff] [blame] | 1004 | func newFileRegistry(files ...pref.FileDescriptor) *protoregistry.Files { |
| 1005 | r := new(protoregistry.Files) |
| 1006 | for _, file := range files { |
| 1007 | r.RegisterFile(file) |
| 1008 | } |
| 1009 | return r |
| 1010 | } |
| 1011 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1012 | func (m *EnumMessages) ProtoReflect() pref.Message { return enumMessagesType.MessageOf(m) } |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1013 | |
Joe Tsai | f18ab53 | 2018-11-27 17:25:04 -0800 | [diff] [blame] | 1014 | func (*EnumMessages) XXX_OneofWrappers() []interface{} { |
| 1015 | return []interface{}{ |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1016 | (*EnumMessages_OneofE2)(nil), |
| 1017 | (*EnumMessages_OneofE3)(nil), |
| 1018 | (*EnumMessages_OneofM2)(nil), |
| 1019 | (*EnumMessages_OneofM3)(nil), |
| 1020 | } |
| 1021 | } |
| 1022 | |
| 1023 | type ( |
| 1024 | isEnumMessages_Union interface { |
| 1025 | isEnumMessages_Union() |
| 1026 | } |
| 1027 | EnumMessages_OneofE2 struct { |
| 1028 | OneofE2 EnumProto2 `protobuf:"9"` |
| 1029 | } |
| 1030 | EnumMessages_OneofE3 struct { |
| 1031 | OneofE3 EnumProto3 `protobuf:"10"` |
| 1032 | } |
| 1033 | EnumMessages_OneofM2 struct { |
| 1034 | OneofM2 *ScalarProto2 `protobuf:"11"` |
| 1035 | } |
| 1036 | EnumMessages_OneofM3 struct { |
| 1037 | OneofM3 *ScalarProto3 `protobuf:"12"` |
| 1038 | } |
| 1039 | ) |
| 1040 | |
| 1041 | func (*EnumMessages_OneofE2) isEnumMessages_Union() {} |
| 1042 | func (*EnumMessages_OneofE3) isEnumMessages_Union() {} |
| 1043 | func (*EnumMessages_OneofM2) isEnumMessages_Union() {} |
| 1044 | func (*EnumMessages_OneofM3) isEnumMessages_Union() {} |
| 1045 | |
| 1046 | func TestEnumMessages(t *testing.T) { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1047 | emptyL := pimpl.Export{}.MessageOf(new(proto2_20180125.Message)) |
| 1048 | emptyM := new(EnumMessages).ProtoReflect() |
| 1049 | emptyM2 := new(ScalarProto2).ProtoReflect() |
| 1050 | emptyM3 := new(ScalarProto3).ProtoReflect() |
| 1051 | |
Damien Neil | a8a2cea | 2019-07-10 16:17:16 -0700 | [diff] [blame] | 1052 | wantL := pimpl.Export{}.MessageOf(&proto2_20180125.Message{OptionalFloat: proto.Float32(math.E)}) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1053 | wantM := (&EnumMessages{EnumP2: EnumProto2(1234).Enum()}).ProtoReflect() |
Damien Neil | a8a2cea | 2019-07-10 16:17:16 -0700 | [diff] [blame] | 1054 | wantM2a := &ScalarProto2{Float32: proto.Float32(math.Pi)} |
| 1055 | wantM2b := &ScalarProto2{Float32: proto.Float32(math.Phi)} |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1056 | wantM3a := &ScalarProto3{Float32: math.Pi} |
| 1057 | wantM3b := &ScalarProto3{Float32: math.Ln2} |
| 1058 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1059 | wantList5 := getField((&EnumMessages{EnumList: []EnumProto2{333, 222}}).ProtoReflect(), 5) |
| 1060 | wantList6 := getField((&EnumMessages{MessageList: []*ScalarProto2{wantM2a, wantM2b}}).ProtoReflect(), 6) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1061 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1062 | wantMap7 := getField((&EnumMessages{EnumMap: map[string]EnumProto3{"one": 1, "two": 2}}).ProtoReflect(), 7) |
| 1063 | wantMap8 := getField((&EnumMessages{MessageMap: map[string]*ScalarProto3{"pi": wantM3a, "ln2": wantM3b}}).ProtoReflect(), 8) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1064 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1065 | testMessage(t, nil, new(EnumMessages).ProtoReflect(), messageOps{ |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1066 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1067 | getFields{1: VE(0xbeef), 2: VE(1), 3: V(emptyL), 4: V(emptyM), 9: VE(0xbeef), 10: VE(1)}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1068 | |
| 1069 | // Test singular enums. |
| 1070 | setFields{1: VE(0xdead), 2: VE(0)}, |
| 1071 | getFields{1: VE(0xdead), 2: VE(0)}, |
| 1072 | hasFields{1: true, 2: true}, |
| 1073 | |
| 1074 | // Test singular messages. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1075 | messageFieldsMutable{3: messageOps{setFields{109: V(float32(math.E))}}}, |
| 1076 | messageFieldsMutable{4: messageOps{setFields{1: VE(1234)}}}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1077 | getFields{3: V(wantL), 4: V(wantM)}, |
| 1078 | clearFields{3, 4}, |
| 1079 | hasFields{3: false, 4: false}, |
| 1080 | setFields{3: V(wantL), 4: V(wantM)}, |
| 1081 | hasFields{3: true, 4: true}, |
| 1082 | |
| 1083 | // Test list of enums and messages. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1084 | listFieldsMutable{ |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1085 | 5: listOps{ |
| 1086 | appendList{VE(111), VE(222)}, |
| 1087 | setList{0: VE(333)}, |
| 1088 | getList{0: VE(333), 1: VE(222)}, |
| 1089 | lenList(2), |
| 1090 | }, |
| 1091 | 6: listOps{ |
| 1092 | appendMessageList{setFields{4: V(uint32(1e6))}}, |
| 1093 | appendMessageList{setFields{6: V(float32(math.Phi))}}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1094 | setList{0: V(wantM2a.ProtoReflect())}, |
| 1095 | getList{0: V(wantM2a.ProtoReflect()), 1: V(wantM2b.ProtoReflect())}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1096 | }, |
| 1097 | }, |
| 1098 | getFields{5: wantList5, 6: wantList6}, |
| 1099 | hasFields{5: true, 6: true}, |
| 1100 | listFields{5: listOps{truncList(0)}}, |
| 1101 | hasFields{5: false, 6: true}, |
| 1102 | |
| 1103 | // Test maps of enums and messages. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1104 | mapFieldsMutable{ |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1105 | 7: mapOps{ |
| 1106 | setMap{"one": VE(1), "two": VE(2)}, |
| 1107 | hasMap{"one": true, "two": true, "three": false}, |
| 1108 | lenMap(2), |
| 1109 | }, |
| 1110 | 8: mapOps{ |
| 1111 | messageMap{"pi": messageOps{setFields{6: V(float32(math.Pi))}}}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1112 | setMap{"ln2": V(wantM3b.ProtoReflect())}, |
| 1113 | getMap{"pi": V(wantM3a.ProtoReflect()), "ln2": V(wantM3b.ProtoReflect()), "none": V(nil)}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1114 | lenMap(2), |
| 1115 | }, |
| 1116 | }, |
| 1117 | getFields{7: wantMap7, 8: wantMap8}, |
| 1118 | hasFields{7: true, 8: true}, |
| 1119 | mapFields{8: mapOps{clearMap{"pi", "ln2", "none"}}}, |
| 1120 | hasFields{7: true, 8: false}, |
| 1121 | |
| 1122 | // Test oneofs of enums and messages. |
| 1123 | setFields{9: VE(0xdead)}, |
| 1124 | hasFields{1: true, 2: true, 9: true, 10: false, 11: false, 12: false}, |
| 1125 | setFields{10: VE(0)}, |
| 1126 | hasFields{1: true, 2: true, 9: false, 10: true, 11: false, 12: false}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1127 | messageFieldsMutable{11: messageOps{setFields{6: V(float32(math.Pi))}}}, |
| 1128 | getFields{11: V(wantM2a.ProtoReflect())}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1129 | hasFields{1: true, 2: true, 9: false, 10: false, 11: true, 12: false}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1130 | messageFieldsMutable{12: messageOps{setFields{6: V(float32(math.Pi))}}}, |
| 1131 | getFields{12: V(wantM3a.ProtoReflect())}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1132 | hasFields{1: true, 2: true, 9: false, 10: false, 11: false, 12: true}, |
| 1133 | |
| 1134 | // Check entire message. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1135 | rangeFields{1: VE(0xdead), 2: VE(0), 3: V(wantL), 4: V(wantM), 6: wantList6, 7: wantMap7, 12: V(wantM3a.ProtoReflect())}, |
| 1136 | equalMessage{(&EnumMessages{ |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1137 | EnumP2: EnumProto2(0xdead).Enum(), |
| 1138 | EnumP3: EnumProto3(0).Enum(), |
Damien Neil | a8a2cea | 2019-07-10 16:17:16 -0700 | [diff] [blame] | 1139 | MessageLegacy: &proto2_20180125.Message{OptionalFloat: proto.Float32(math.E)}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1140 | MessageCycle: wantM.Interface().(*EnumMessages), |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1141 | MessageList: []*ScalarProto2{wantM2a, wantM2b}, |
| 1142 | EnumMap: map[string]EnumProto3{"one": 1, "two": 2}, |
| 1143 | Union: &EnumMessages_OneofM3{wantM3a}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1144 | }).ProtoReflect()}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1145 | clearFields{1, 2, 3, 4, 6, 7, 12}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1146 | equalMessage{new(EnumMessages).ProtoReflect()}, |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1147 | }) |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 1148 | |
| 1149 | // Test read-only operations on nil message. |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1150 | testMessage(t, nil, (*EnumMessages)(nil).ProtoReflect(), messageOps{ |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 1151 | hasFields{1: false, 2: false, 3: false, 4: false, 5: false, 6: false, 7: false, 8: false, 9: false, 10: false, 11: false, 12: false}, |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1152 | getFields{1: VE(0xbeef), 2: VE(1), 3: V(emptyL), 4: V(emptyM), 9: VE(0xbeef), 10: VE(1), 11: V(emptyM2), 12: V(emptyM3)}, |
Joe Tsai | 6cf80c4 | 2018-12-01 04:57:09 -0800 | [diff] [blame] | 1153 | listFields{5: {lenList(0)}, 6: {lenList(0)}}, |
| 1154 | mapFields{7: {lenMap(0)}, 8: {lenMap(0)}}, |
| 1155 | }) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1156 | } |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 1157 | |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1158 | var cmpOpts = cmp.Options{ |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1159 | cmp.Comparer(func(x, y *proto2_20180125.Message) bool { |
Joe Tsai | d888139 | 2019-06-06 13:01:53 -0700 | [diff] [blame] | 1160 | mx := pimpl.Export{}.MessageOf(x).Interface() |
| 1161 | my := pimpl.Export{}.MessageOf(y).Interface() |
| 1162 | return proto.Equal(mx, my) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1163 | }), |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1164 | cmp.Transformer("UnwrapValue", func(pv pref.Value) interface{} { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1165 | switch v := pv.Interface().(type) { |
| 1166 | case pref.Message: |
| 1167 | out := make(map[pref.FieldNumber]pref.Value) |
| 1168 | v.Range(func(fd pref.FieldDescriptor, v pref.Value) bool { |
| 1169 | out[fd.Number()] = v |
| 1170 | return true |
| 1171 | }) |
| 1172 | return out |
| 1173 | case pref.List: |
| 1174 | var out []pref.Value |
| 1175 | for i := 0; i < v.Len(); i++ { |
| 1176 | out = append(out, v.Get(i)) |
| 1177 | } |
| 1178 | return out |
| 1179 | case pref.Map: |
| 1180 | out := make(map[interface{}]pref.Value) |
| 1181 | v.Range(func(k pref.MapKey, v pref.Value) bool { |
| 1182 | out[k.Interface()] = v |
| 1183 | return true |
| 1184 | }) |
| 1185 | return out |
| 1186 | default: |
| 1187 | return v |
| 1188 | } |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1189 | }), |
| 1190 | cmpopts.EquateNaNs(), |
| 1191 | } |
| 1192 | |
| 1193 | func testMessage(t *testing.T, p path, m pref.Message, tt messageOps) { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1194 | fieldDescs := m.Descriptor().Fields() |
| 1195 | oneofDescs := m.Descriptor().Oneofs() |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1196 | for i, op := range tt { |
| 1197 | p.Push(i) |
| 1198 | switch op := op.(type) { |
| 1199 | case equalMessage: |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1200 | if diff := cmp.Diff(V(op.Message), V(m), cmpOpts); diff != "" { |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1201 | t.Errorf("operation %v, message mismatch (-want, +got):\n%s", p, diff) |
| 1202 | } |
| 1203 | case hasFields: |
| 1204 | got := map[pref.FieldNumber]bool{} |
| 1205 | want := map[pref.FieldNumber]bool(op) |
| 1206 | for n := range want { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1207 | fd := fieldDescs.ByNumber(n) |
| 1208 | got[n] = m.Has(fd) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1209 | } |
| 1210 | if diff := cmp.Diff(want, got); diff != "" { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1211 | t.Errorf("operation %v, Message.Has mismatch (-want, +got):\n%s", p, diff) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1212 | } |
| 1213 | case getFields: |
| 1214 | got := map[pref.FieldNumber]pref.Value{} |
| 1215 | want := map[pref.FieldNumber]pref.Value(op) |
| 1216 | for n := range want { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1217 | fd := fieldDescs.ByNumber(n) |
| 1218 | got[n] = m.Get(fd) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1219 | } |
| 1220 | if diff := cmp.Diff(want, got, cmpOpts); diff != "" { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1221 | t.Errorf("operation %v, Message.Get mismatch (-want, +got):\n%s", p, diff) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1222 | } |
| 1223 | case setFields: |
| 1224 | for n, v := range op { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1225 | fd := fieldDescs.ByNumber(n) |
| 1226 | m.Set(fd, v) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1227 | } |
| 1228 | case clearFields: |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1229 | for _, n := range op { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1230 | fd := fieldDescs.ByNumber(n) |
| 1231 | m.Clear(fd) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1232 | } |
Joe Tsai | 4ec39c7 | 2019-04-03 13:40:53 -0700 | [diff] [blame] | 1233 | case whichOneofs: |
| 1234 | got := map[pref.Name]pref.FieldNumber{} |
| 1235 | want := map[pref.Name]pref.FieldNumber(op) |
| 1236 | for s := range want { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1237 | od := oneofDescs.ByName(s) |
| 1238 | fd := m.WhichOneof(od) |
| 1239 | if fd == nil { |
| 1240 | got[s] = 0 |
| 1241 | } else { |
| 1242 | got[s] = fd.Number() |
| 1243 | } |
Joe Tsai | 4ec39c7 | 2019-04-03 13:40:53 -0700 | [diff] [blame] | 1244 | } |
| 1245 | if diff := cmp.Diff(want, got); diff != "" { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1246 | t.Errorf("operation %v, Message.WhichOneof mismatch (-want, +got):\n%s", p, diff) |
Joe Tsai | 4ec39c7 | 2019-04-03 13:40:53 -0700 | [diff] [blame] | 1247 | } |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1248 | case messageFields: |
| 1249 | for n, tt := range op { |
| 1250 | p.Push(int(n)) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1251 | fd := fieldDescs.ByNumber(n) |
| 1252 | testMessage(t, p, m.Get(fd).Message(), tt) |
| 1253 | p.Pop() |
| 1254 | } |
| 1255 | case messageFieldsMutable: |
| 1256 | for n, tt := range op { |
| 1257 | p.Push(int(n)) |
| 1258 | fd := fieldDescs.ByNumber(n) |
| 1259 | testMessage(t, p, m.Mutable(fd).Message(), tt) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1260 | p.Pop() |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 1261 | } |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1262 | case listFields: |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1263 | for n, tt := range op { |
| 1264 | p.Push(int(n)) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1265 | fd := fieldDescs.ByNumber(n) |
| 1266 | testLists(t, p, m.Get(fd).List(), tt) |
| 1267 | p.Pop() |
| 1268 | } |
| 1269 | case listFieldsMutable: |
| 1270 | for n, tt := range op { |
| 1271 | p.Push(int(n)) |
| 1272 | fd := fieldDescs.ByNumber(n) |
| 1273 | testLists(t, p, m.Mutable(fd).List(), tt) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1274 | p.Pop() |
| 1275 | } |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 1276 | case mapFields: |
| 1277 | for n, tt := range op { |
| 1278 | p.Push(int(n)) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1279 | fd := fieldDescs.ByNumber(n) |
| 1280 | testMaps(t, p, m.Get(fd).Map(), tt) |
| 1281 | p.Pop() |
| 1282 | } |
| 1283 | case mapFieldsMutable: |
| 1284 | for n, tt := range op { |
| 1285 | p.Push(int(n)) |
| 1286 | fd := fieldDescs.ByNumber(n) |
| 1287 | testMaps(t, p, m.Mutable(fd).Map(), tt) |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 1288 | p.Pop() |
| 1289 | } |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1290 | case rangeFields: |
| 1291 | got := map[pref.FieldNumber]pref.Value{} |
| 1292 | want := map[pref.FieldNumber]pref.Value(op) |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1293 | m.Range(func(fd pref.FieldDescriptor, v pref.Value) bool { |
| 1294 | got[fd.Number()] = v |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1295 | return true |
| 1296 | }) |
| 1297 | if diff := cmp.Diff(want, got, cmpOpts); diff != "" { |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1298 | t.Errorf("operation %v, Message.Range mismatch (-want, +got):\n%s", p, diff) |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1299 | } |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1300 | default: |
| 1301 | t.Fatalf("operation %v, invalid operation: %T", p, op) |
| 1302 | } |
| 1303 | p.Pop() |
Joe Tsai | fa02f4e | 2018-09-12 16:20:37 -0700 | [diff] [blame] | 1304 | } |
| 1305 | } |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1306 | |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1307 | func testLists(t *testing.T, p path, v pref.List, tt listOps) { |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1308 | for i, op := range tt { |
| 1309 | p.Push(i) |
| 1310 | switch op := op.(type) { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1311 | case equalList: |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1312 | if diff := cmp.Diff(V(op.List), V(v), cmpOpts); diff != "" { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1313 | t.Errorf("operation %v, list mismatch (-want, +got):\n%s", p, diff) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1314 | } |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1315 | case lenList: |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1316 | if got, want := v.Len(), int(op); got != want { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1317 | t.Errorf("operation %v, List.Len = %d, want %d", p, got, want) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1318 | } |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1319 | case getList: |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1320 | got := map[int]pref.Value{} |
| 1321 | want := map[int]pref.Value(op) |
| 1322 | for n := range want { |
| 1323 | got[n] = v.Get(n) |
| 1324 | } |
| 1325 | if diff := cmp.Diff(want, got, cmpOpts); diff != "" { |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1326 | t.Errorf("operation %v, List.Get mismatch (-want, +got):\n%s", p, diff) |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1327 | } |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1328 | case setList: |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1329 | for n, e := range op { |
| 1330 | v.Set(n, e) |
| 1331 | } |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1332 | case appendList: |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1333 | for _, e := range op { |
| 1334 | v.Append(e) |
| 1335 | } |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1336 | case appendMessageList: |
Damien Neil | d91c422 | 2019-09-04 10:46:00 -0700 | [diff] [blame] | 1337 | e := v.NewElement() |
| 1338 | v.Append(e) |
| 1339 | testMessage(t, p, e.Message(), messageOps(op)) |
Joe Tsai | 4b7aff6 | 2018-11-14 14:05:19 -0800 | [diff] [blame] | 1340 | case truncList: |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1341 | v.Truncate(int(op)) |
| 1342 | default: |
| 1343 | t.Fatalf("operation %v, invalid operation: %T", p, op) |
| 1344 | } |
| 1345 | p.Pop() |
| 1346 | } |
| 1347 | } |
| 1348 | |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 1349 | func testMaps(t *testing.T, p path, m pref.Map, tt mapOps) { |
| 1350 | for i, op := range tt { |
| 1351 | p.Push(i) |
| 1352 | switch op := op.(type) { |
| 1353 | case equalMap: |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1354 | if diff := cmp.Diff(V(op.Map), V(m), cmpOpts); diff != "" { |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 1355 | t.Errorf("operation %v, map mismatch (-want, +got):\n%s", p, diff) |
| 1356 | } |
| 1357 | case lenMap: |
| 1358 | if got, want := m.Len(), int(op); got != want { |
| 1359 | t.Errorf("operation %v, Map.Len = %d, want %d", p, got, want) |
| 1360 | } |
| 1361 | case hasMap: |
| 1362 | got := map[interface{}]bool{} |
| 1363 | want := map[interface{}]bool(op) |
| 1364 | for k := range want { |
| 1365 | got[k] = m.Has(V(k).MapKey()) |
| 1366 | } |
| 1367 | if diff := cmp.Diff(want, got, cmpOpts); diff != "" { |
| 1368 | t.Errorf("operation %v, Map.Has mismatch (-want, +got):\n%s", p, diff) |
| 1369 | } |
| 1370 | case getMap: |
| 1371 | got := map[interface{}]pref.Value{} |
| 1372 | want := map[interface{}]pref.Value(op) |
| 1373 | for k := range want { |
| 1374 | got[k] = m.Get(V(k).MapKey()) |
| 1375 | } |
| 1376 | if diff := cmp.Diff(want, got, cmpOpts); diff != "" { |
| 1377 | t.Errorf("operation %v, Map.Get mismatch (-want, +got):\n%s", p, diff) |
| 1378 | } |
| 1379 | case setMap: |
| 1380 | for k, v := range op { |
| 1381 | m.Set(V(k).MapKey(), v) |
| 1382 | } |
| 1383 | case clearMap: |
Joe Tsai | 87b955b | 2018-11-14 21:59:49 -0800 | [diff] [blame] | 1384 | for _, k := range op { |
| 1385 | m.Clear(V(k).MapKey()) |
| 1386 | } |
| 1387 | case messageMap: |
| 1388 | for k, tt := range op { |
Damien Neil | 97e7f57 | 2018-12-07 14:28:33 -0800 | [diff] [blame] | 1389 | mk := V(k).MapKey() |
| 1390 | if !m.Has(mk) { |
Damien Neil | d91c422 | 2019-09-04 10:46:00 -0700 | [diff] [blame] | 1391 | m.Set(mk, m.NewValue()) |
Damien Neil | 97e7f57 | 2018-12-07 14:28:33 -0800 | [diff] [blame] | 1392 | } |
| 1393 | testMessage(t, p, m.Get(mk).Message(), tt) |
Joe Tsai | bbfaeb7 | 2018-10-17 00:27:21 +0000 | [diff] [blame] | 1394 | } |
| 1395 | case rangeMap: |
| 1396 | got := map[interface{}]pref.Value{} |
| 1397 | want := map[interface{}]pref.Value(op) |
| 1398 | m.Range(func(k pref.MapKey, v pref.Value) bool { |
| 1399 | got[k.Interface()] = v |
| 1400 | return true |
| 1401 | }) |
| 1402 | if diff := cmp.Diff(want, got, cmpOpts); diff != "" { |
| 1403 | t.Errorf("operation %v, Map.Range mismatch (-want, +got):\n%s", p, diff) |
| 1404 | } |
| 1405 | default: |
| 1406 | t.Fatalf("operation %v, invalid operation: %T", p, op) |
| 1407 | } |
| 1408 | p.Pop() |
| 1409 | } |
| 1410 | } |
| 1411 | |
Joe Tsai | 378c132 | 2019-04-25 23:48:08 -0700 | [diff] [blame] | 1412 | func getField(m pref.Message, n pref.FieldNumber) pref.Value { |
| 1413 | fd := m.Descriptor().Fields().ByNumber(n) |
| 1414 | return m.Get(fd) |
| 1415 | } |
| 1416 | |
Joe Tsai | 91e1466 | 2018-09-13 13:24:35 -0700 | [diff] [blame] | 1417 | type path []int |
| 1418 | |
| 1419 | func (p *path) Push(i int) { *p = append(*p, i) } |
| 1420 | func (p *path) Pop() { *p = (*p)[:len(*p)-1] } |
| 1421 | func (p path) String() string { |
| 1422 | var ss []string |
| 1423 | for _, i := range p { |
| 1424 | ss = append(ss, fmt.Sprint(i)) |
| 1425 | } |
| 1426 | return strings.Join(ss, ".") |
| 1427 | } |
Joe Tsai | 82760ce | 2019-06-20 03:09:57 -0700 | [diff] [blame] | 1428 | |
Joe Tsai | 5d63473 | 2020-07-26 22:37:47 -0700 | [diff] [blame] | 1429 | type UnknownFieldsA struct { |
| 1430 | XXX_unrecognized []byte |
| 1431 | } |
| 1432 | |
| 1433 | var unknownFieldsAType = pimpl.MessageInfo{ |
| 1434 | GoReflectType: reflect.TypeOf(new(UnknownFieldsA)), |
| 1435 | Desc: mustMakeMessageDesc("unknown.proto", pref.Proto2, "", `name: "UnknownFieldsA"`, nil), |
| 1436 | } |
| 1437 | |
| 1438 | func (m *UnknownFieldsA) ProtoReflect() pref.Message { return unknownFieldsAType.MessageOf(m) } |
| 1439 | |
| 1440 | type UnknownFieldsB struct { |
| 1441 | XXX_unrecognized *[]byte |
| 1442 | } |
| 1443 | |
| 1444 | var unknownFieldsBType = pimpl.MessageInfo{ |
| 1445 | GoReflectType: reflect.TypeOf(new(UnknownFieldsB)), |
| 1446 | Desc: mustMakeMessageDesc("unknown.proto", pref.Proto2, "", `name: "UnknownFieldsB"`, nil), |
| 1447 | } |
| 1448 | |
| 1449 | func (m *UnknownFieldsB) ProtoReflect() pref.Message { return unknownFieldsBType.MessageOf(m) } |
| 1450 | |
| 1451 | func TestUnknownFields(t *testing.T) { |
| 1452 | for _, m := range []proto.Message{new(UnknownFieldsA), new(UnknownFieldsB)} { |
| 1453 | t.Run(reflect.TypeOf(m).Elem().Name(), func(t *testing.T) { |
| 1454 | want := protopack.Message{ |
| 1455 | protopack.Tag{1, protopack.BytesType}, protopack.String("Hello, world!"), |
| 1456 | }.Marshal() |
| 1457 | m.ProtoReflect().SetUnknown(want) |
| 1458 | got := []byte(m.ProtoReflect().GetUnknown()) |
| 1459 | if diff := cmp.Diff(want, got); diff != "" { |
| 1460 | t.Errorf("UnknownFields mismatch (-want +got):\n%s", diff) |
| 1461 | } |
| 1462 | }) |
| 1463 | } |
| 1464 | } |
| 1465 | |
Joe Tsai | ae313d4 | 2019-10-16 10:25:54 -0700 | [diff] [blame] | 1466 | func TestReset(t *testing.T) { |
| 1467 | mi := new(testpb.TestAllTypes) |
| 1468 | |
| 1469 | // ProtoReflect is implemented using a messageState cache. |
| 1470 | m := mi.ProtoReflect() |
| 1471 | |
| 1472 | // Reset must not clear the messageState cache. |
| 1473 | mi.Reset() |
| 1474 | |
| 1475 | // If Reset accidentally cleared the messageState cache, this panics. |
| 1476 | m.Descriptor() |
| 1477 | } |
| 1478 | |
Damien Neil | b120a23 | 2019-12-16 13:23:25 -0800 | [diff] [blame] | 1479 | func TestIsValid(t *testing.T) { |
| 1480 | var m *testpb.TestAllTypes |
| 1481 | if got, want := m.ProtoReflect().IsValid(), false; got != want { |
| 1482 | t.Errorf("((*M)(nil)).ProtoReflect().IsValid() = %v, want %v", got, want) |
| 1483 | } |
| 1484 | m = &testpb.TestAllTypes{} |
| 1485 | if got, want := m.ProtoReflect().IsValid(), true; got != want { |
| 1486 | t.Errorf("(&M{}).ProtoReflect().IsValid() = %v, want %v", got, want) |
| 1487 | } |
| 1488 | } |
| 1489 | |
Joe Tsai | 82760ce | 2019-06-20 03:09:57 -0700 | [diff] [blame] | 1490 | // The MessageState implementation makes the assumption that when a |
| 1491 | // concrete message is unsafe casted as a *MessageState, the Go GC does |
| 1492 | // not reclaim the memory for the remainder of the concrete message. |
| 1493 | func TestUnsafeAssumptions(t *testing.T) { |
| 1494 | if !pimpl.UnsafeEnabled { |
| 1495 | t.Skip() |
| 1496 | } |
| 1497 | |
| 1498 | var wg sync.WaitGroup |
| 1499 | for i := 0; i < 10; i++ { |
| 1500 | wg.Add(1) |
| 1501 | go func() { |
| 1502 | var ms [10]pref.Message |
| 1503 | |
| 1504 | // Store the message only in its reflective form. |
| 1505 | // Trigger the GC after each iteration. |
| 1506 | for j := 0; j < 10; j++ { |
| 1507 | ms[j] = (&testpb.TestAllTypes{ |
Damien Neil | a8a2cea | 2019-07-10 16:17:16 -0700 | [diff] [blame] | 1508 | OptionalInt32: proto.Int32(int32(j)), |
| 1509 | OptionalFloat: proto.Float32(float32(j)), |
Joe Tsai | 82760ce | 2019-06-20 03:09:57 -0700 | [diff] [blame] | 1510 | RepeatedInt32: []int32{int32(j)}, |
| 1511 | RepeatedFloat: []float32{float32(j)}, |
Damien Neil | a8a2cea | 2019-07-10 16:17:16 -0700 | [diff] [blame] | 1512 | DefaultInt32: proto.Int32(int32(j)), |
| 1513 | DefaultFloat: proto.Float32(float32(j)), |
Joe Tsai | 82760ce | 2019-06-20 03:09:57 -0700 | [diff] [blame] | 1514 | }).ProtoReflect() |
| 1515 | runtime.GC() |
| 1516 | } |
| 1517 | |
| 1518 | // Convert the reflective form back into a concrete form. |
| 1519 | // Verify that the values written previously are still the same. |
| 1520 | for j := 0; j < 10; j++ { |
| 1521 | switch m := ms[j].Interface().(*testpb.TestAllTypes); { |
| 1522 | case m.GetOptionalInt32() != int32(j): |
| 1523 | case m.GetOptionalFloat() != float32(j): |
| 1524 | case m.GetRepeatedInt32()[0] != int32(j): |
| 1525 | case m.GetRepeatedFloat()[0] != float32(j): |
| 1526 | case m.GetDefaultInt32() != int32(j): |
| 1527 | case m.GetDefaultFloat() != float32(j): |
| 1528 | default: |
| 1529 | continue |
| 1530 | } |
| 1531 | t.Error("memory corrupted detected") |
| 1532 | } |
| 1533 | defer wg.Done() |
| 1534 | }() |
| 1535 | } |
| 1536 | wg.Wait() |
| 1537 | } |
| 1538 | |
| 1539 | func BenchmarkName(b *testing.B) { |
| 1540 | var sink pref.FullName |
| 1541 | b.Run("Value", func(b *testing.B) { |
| 1542 | b.ReportAllocs() |
| 1543 | m := new(descriptorpb.FileDescriptorProto) |
| 1544 | for i := 0; i < b.N; i++ { |
| 1545 | sink = m.ProtoReflect().Descriptor().FullName() |
| 1546 | } |
| 1547 | }) |
| 1548 | b.Run("Nil", func(b *testing.B) { |
| 1549 | b.ReportAllocs() |
| 1550 | m := (*descriptorpb.FileDescriptorProto)(nil) |
| 1551 | for i := 0; i < b.N; i++ { |
| 1552 | sink = m.ProtoReflect().Descriptor().FullName() |
| 1553 | } |
| 1554 | }) |
| 1555 | runtime.KeepAlive(sink) |
| 1556 | } |
Joe Tsai | 9d637ca | 2019-09-18 18:08:52 -0700 | [diff] [blame] | 1557 | |
| 1558 | func BenchmarkReflect(b *testing.B) { |
| 1559 | m := new(testpb.TestAllTypes).ProtoReflect() |
| 1560 | fds := m.Descriptor().Fields() |
| 1561 | vs := make([]pref.Value, fds.Len()) |
| 1562 | for i := range vs { |
| 1563 | vs[i] = m.NewField(fds.Get(i)) |
| 1564 | } |
| 1565 | |
| 1566 | b.Run("Has", func(b *testing.B) { |
| 1567 | b.ReportAllocs() |
| 1568 | for i := 0; i < b.N; i++ { |
| 1569 | for j := 0; j < fds.Len(); j++ { |
| 1570 | m.Has(fds.Get(j)) |
| 1571 | } |
| 1572 | } |
| 1573 | }) |
| 1574 | b.Run("Get", func(b *testing.B) { |
| 1575 | b.ReportAllocs() |
| 1576 | for i := 0; i < b.N; i++ { |
| 1577 | for j := 0; j < fds.Len(); j++ { |
| 1578 | m.Get(fds.Get(j)) |
| 1579 | } |
| 1580 | } |
| 1581 | }) |
| 1582 | b.Run("Set", func(b *testing.B) { |
| 1583 | b.ReportAllocs() |
| 1584 | for i := 0; i < b.N; i++ { |
| 1585 | for j := 0; j < fds.Len(); j++ { |
| 1586 | m.Set(fds.Get(j), vs[j]) |
| 1587 | } |
| 1588 | } |
| 1589 | }) |
| 1590 | b.Run("Clear", func(b *testing.B) { |
| 1591 | b.ReportAllocs() |
| 1592 | for i := 0; i < b.N; i++ { |
| 1593 | for j := 0; j < fds.Len(); j++ { |
| 1594 | m.Clear(fds.Get(j)) |
| 1595 | } |
| 1596 | } |
| 1597 | }) |
| 1598 | b.Run("Range", func(b *testing.B) { |
| 1599 | b.ReportAllocs() |
| 1600 | for i := 0; i < b.N; i++ { |
| 1601 | m.Range(func(pref.FieldDescriptor, pref.Value) bool { |
| 1602 | return true |
| 1603 | }) |
| 1604 | } |
| 1605 | }) |
| 1606 | } |