blob: cd5f37656da461562d89fdce8d7c90f4384a69e2 [file] [log] [blame]
Joe Tsai90fe9962018-10-18 11:06:29 -07001// 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
5package impl
6
7import (
Joe Tsaie2afdc22018-10-25 14:06:56 -07008 "bytes"
9 "math"
Joe Tsai90fe9962018-10-18 11:06:29 -070010 "reflect"
11 "testing"
12
Joe Tsai95b02902018-10-31 18:23:42 -070013 protoV1 "github.com/golang/protobuf/proto"
Joe Tsaiea118132018-11-11 17:56:21 -080014 pack "github.com/golang/protobuf/v2/internal/encoding/pack"
15 pragma "github.com/golang/protobuf/v2/internal/pragma"
Joe Tsai009e0672018-11-27 18:45:07 -080016 scalar "github.com/golang/protobuf/v2/internal/scalar"
Joe Tsai90fe9962018-10-18 11:06:29 -070017 pref "github.com/golang/protobuf/v2/reflect/protoreflect"
18 ptype "github.com/golang/protobuf/v2/reflect/prototype"
Joe Tsaiea118132018-11-11 17:56:21 -080019 cmp "github.com/google/go-cmp/cmp"
20
21 proto2_20160225 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto2.v0.0.0-20160225-2fc053c5"
22 proto2_20160519 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto2.v0.0.0-20160519-a4ab9ec5"
23 proto2_20180125 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto2.v1.0.0-20180125-92554152"
24 proto2_20180430 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto2.v1.1.0-20180430-b4deda09"
25 proto2_20180814 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto2.v1.2.0-20180814-aa810b61"
Joe Tsai25cc69d2018-11-28 23:43:49 -080026 proto2_20181126 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto2.v1.2.1-20181126-8d0c54c1"
Joe Tsaiea118132018-11-11 17:56:21 -080027 proto3_20160225 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto3.v0.0.0-20160225-2fc053c5"
28 proto3_20160519 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto3.v0.0.0-20160519-a4ab9ec5"
29 proto3_20180125 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto3.v1.0.0-20180125-92554152"
30 proto3_20180430 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto3.v1.1.0-20180430-b4deda09"
31 proto3_20180814 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto3.v1.2.0-20180814-aa810b61"
Joe Tsai25cc69d2018-11-28 23:43:49 -080032 proto3_20181126 "github.com/golang/protobuf/v2/internal/testprotos/legacy/proto3.v1.2.1-20181126-8d0c54c1"
Joe Tsai90fe9962018-10-18 11:06:29 -070033)
34
Joe Tsaiea118132018-11-11 17:56:21 -080035func mustLoadFileDesc(b []byte, _ []int) pref.FileDescriptor {
Joe Tsaif0c01e42018-11-06 13:05:20 -080036 fd, err := ptype.NewFileFromDescriptorProto(legacyLoadFileDesc(b), nil)
Joe Tsai90fe9962018-10-18 11:06:29 -070037 if err != nil {
38 panic(err)
39 }
40 return fd
41}
42
Joe Tsaie2afdc22018-10-25 14:06:56 -070043func TestLegacyDescriptor(t *testing.T) {
Joe Tsaiea118132018-11-11 17:56:21 -080044 var tests []struct{ got, want pref.Descriptor }
45
46 fileDescP2_20160225 := mustLoadFileDesc(new(proto2_20160225.Message).Descriptor())
47 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -080048 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20160225.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -080049 want: fileDescP2_20160225.Enums().ByName("SiblingEnum"),
Joe Tsai90fe9962018-10-18 11:06:29 -070050 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080051 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20160225.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -080052 want: fileDescP2_20160225.Messages().ByName("Message").Enums().ByName("ChildEnum"),
Joe Tsai90fe9962018-10-18 11:06:29 -070053 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080054 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -080055 want: fileDescP2_20160225.Messages().ByName("SiblingMessage"),
Joe Tsai90fe9962018-10-18 11:06:29 -070056 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080057 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -080058 want: fileDescP2_20160225.Messages().ByName("Message").Messages().ByName("ChildMessage"),
Joe Tsai90fe9962018-10-18 11:06:29 -070059 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080060 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -080061 want: fileDescP2_20160225.Messages().ByName("Message"),
Joe Tsai90fe9962018-10-18 11:06:29 -070062 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080063 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message_NamedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -080064 want: fileDescP2_20160225.Messages().ByName("Message").Messages().ByName("NamedGroup"),
Joe Tsai90fe9962018-10-18 11:06:29 -070065 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080066 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message_OptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -080067 want: fileDescP2_20160225.Messages().ByName("Message").Messages().ByName("OptionalGroup"),
Joe Tsai90fe9962018-10-18 11:06:29 -070068 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080069 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message_RequiredGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -080070 want: fileDescP2_20160225.Messages().ByName("Message").Messages().ByName("RequiredGroup"),
Joe Tsai90fe9962018-10-18 11:06:29 -070071 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080072 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message_RepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -080073 want: fileDescP2_20160225.Messages().ByName("Message").Messages().ByName("RepeatedGroup"),
Joe Tsai90fe9962018-10-18 11:06:29 -070074 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080075 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message_OneofGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -080076 want: fileDescP2_20160225.Messages().ByName("Message").Messages().ByName("OneofGroup"),
Joe Tsai90fe9962018-10-18 11:06:29 -070077 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080078 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message_ExtensionOptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -080079 want: fileDescP2_20160225.Messages().ByName("Message").Messages().ByName("ExtensionOptionalGroup"),
Joe Tsai90fe9962018-10-18 11:06:29 -070080 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080081 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160225.Message_ExtensionRepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -080082 want: fileDescP2_20160225.Messages().ByName("Message").Messages().ByName("ExtensionRepeatedGroup"),
83 }}...)
84
85 fileDescP3_20160225 := mustLoadFileDesc(new(proto3_20160225.Message).Descriptor())
86 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -080087 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20160225.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -080088 want: fileDescP3_20160225.Enums().ByName("SiblingEnum"),
Joe Tsai90fe9962018-10-18 11:06:29 -070089 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080090 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20160225.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -080091 want: fileDescP3_20160225.Messages().ByName("Message").Enums().ByName("ChildEnum"),
Joe Tsai90fe9962018-10-18 11:06:29 -070092 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080093 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20160225.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -080094 want: fileDescP3_20160225.Messages().ByName("SiblingMessage"),
Joe Tsai90fe9962018-10-18 11:06:29 -070095 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080096 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20160225.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -080097 want: fileDescP3_20160225.Messages().ByName("Message").Messages().ByName("ChildMessage"),
98 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -080099 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20160225.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800100 want: fileDescP3_20160225.Messages().ByName("Message"),
101 }}...)
102
103 fileDescP2_20160519 := mustLoadFileDesc(new(proto2_20160519.Message).Descriptor())
104 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -0800105 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20160519.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800106 want: fileDescP2_20160519.Enums().ByName("SiblingEnum"),
107 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800108 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20160519.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800109 want: fileDescP2_20160519.Messages().ByName("Message").Enums().ByName("ChildEnum"),
110 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800111 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800112 want: fileDescP2_20160519.Messages().ByName("SiblingMessage"),
113 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800114 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800115 want: fileDescP2_20160519.Messages().ByName("Message").Messages().ByName("ChildMessage"),
116 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800117 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800118 want: fileDescP2_20160519.Messages().ByName("Message"),
119 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800120 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message_NamedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800121 want: fileDescP2_20160519.Messages().ByName("Message").Messages().ByName("NamedGroup"),
122 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800123 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message_OptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800124 want: fileDescP2_20160519.Messages().ByName("Message").Messages().ByName("OptionalGroup"),
125 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800126 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message_RequiredGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800127 want: fileDescP2_20160519.Messages().ByName("Message").Messages().ByName("RequiredGroup"),
128 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800129 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message_RepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800130 want: fileDescP2_20160519.Messages().ByName("Message").Messages().ByName("RepeatedGroup"),
131 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800132 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message_OneofGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800133 want: fileDescP2_20160519.Messages().ByName("Message").Messages().ByName("OneofGroup"),
134 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800135 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message_ExtensionOptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800136 want: fileDescP2_20160519.Messages().ByName("Message").Messages().ByName("ExtensionOptionalGroup"),
137 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800138 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20160519.Message_ExtensionRepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800139 want: fileDescP2_20160519.Messages().ByName("Message").Messages().ByName("ExtensionRepeatedGroup"),
140 }}...)
141
142 fileDescP3_20160519 := mustLoadFileDesc(new(proto3_20160519.Message).Descriptor())
143 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -0800144 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20160519.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800145 want: fileDescP3_20160519.Enums().ByName("SiblingEnum"),
146 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800147 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20160519.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800148 want: fileDescP3_20160519.Messages().ByName("Message").Enums().ByName("ChildEnum"),
149 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800150 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20160519.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800151 want: fileDescP3_20160519.Messages().ByName("SiblingMessage"),
152 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800153 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20160519.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800154 want: fileDescP3_20160519.Messages().ByName("Message").Messages().ByName("ChildMessage"),
155 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800156 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20160519.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800157 want: fileDescP3_20160519.Messages().ByName("Message"),
158 }}...)
159
160 fileDescP2_20180125 := mustLoadFileDesc(new(proto2_20180125.Message).Descriptor())
161 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -0800162 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20180125.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800163 want: fileDescP2_20180125.Enums().ByName("SiblingEnum"),
164 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800165 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20180125.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800166 want: fileDescP2_20180125.Messages().ByName("Message").Enums().ByName("ChildEnum"),
167 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800168 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800169 want: fileDescP2_20180125.Messages().ByName("SiblingMessage"),
170 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800171 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800172 want: fileDescP2_20180125.Messages().ByName("Message").Messages().ByName("ChildMessage"),
173 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800174 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800175 want: fileDescP2_20180125.Messages().ByName("Message"),
176 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800177 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message_NamedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800178 want: fileDescP2_20180125.Messages().ByName("Message").Messages().ByName("NamedGroup"),
179 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800180 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message_OptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800181 want: fileDescP2_20180125.Messages().ByName("Message").Messages().ByName("OptionalGroup"),
182 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800183 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message_RequiredGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800184 want: fileDescP2_20180125.Messages().ByName("Message").Messages().ByName("RequiredGroup"),
185 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800186 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message_RepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800187 want: fileDescP2_20180125.Messages().ByName("Message").Messages().ByName("RepeatedGroup"),
188 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800189 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message_OneofGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800190 want: fileDescP2_20180125.Messages().ByName("Message").Messages().ByName("OneofGroup"),
191 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800192 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message_ExtensionOptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800193 want: fileDescP2_20180125.Messages().ByName("Message").Messages().ByName("ExtensionOptionalGroup"),
194 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800195 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180125.Message_ExtensionRepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800196 want: fileDescP2_20180125.Messages().ByName("Message").Messages().ByName("ExtensionRepeatedGroup"),
197 }}...)
198
199 fileDescP3_20180125 := mustLoadFileDesc(new(proto3_20180125.Message).Descriptor())
200 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -0800201 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20180125.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800202 want: fileDescP3_20180125.Enums().ByName("SiblingEnum"),
203 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800204 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20180125.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800205 want: fileDescP3_20180125.Messages().ByName("Message").Enums().ByName("ChildEnum"),
206 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800207 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180125.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800208 want: fileDescP3_20180125.Messages().ByName("SiblingMessage"),
209 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800210 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180125.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800211 want: fileDescP3_20180125.Messages().ByName("Message").Messages().ByName("ChildMessage"),
212 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800213 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180125.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800214 want: fileDescP3_20180125.Messages().ByName("Message"),
215 }}...)
216
217 fileDescP2_20180430 := mustLoadFileDesc(new(proto2_20180430.Message).Descriptor())
218 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -0800219 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20180430.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800220 want: fileDescP2_20180430.Enums().ByName("SiblingEnum"),
221 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800222 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20180430.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800223 want: fileDescP2_20180430.Messages().ByName("Message").Enums().ByName("ChildEnum"),
224 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800225 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800226 want: fileDescP2_20180430.Messages().ByName("SiblingMessage"),
227 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800228 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800229 want: fileDescP2_20180430.Messages().ByName("Message").Messages().ByName("ChildMessage"),
230 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800231 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800232 want: fileDescP2_20180430.Messages().ByName("Message"),
233 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800234 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message_NamedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800235 want: fileDescP2_20180430.Messages().ByName("Message").Messages().ByName("NamedGroup"),
236 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800237 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message_OptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800238 want: fileDescP2_20180430.Messages().ByName("Message").Messages().ByName("OptionalGroup"),
239 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800240 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message_RequiredGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800241 want: fileDescP2_20180430.Messages().ByName("Message").Messages().ByName("RequiredGroup"),
242 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800243 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message_RepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800244 want: fileDescP2_20180430.Messages().ByName("Message").Messages().ByName("RepeatedGroup"),
245 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800246 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message_OneofGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800247 want: fileDescP2_20180430.Messages().ByName("Message").Messages().ByName("OneofGroup"),
248 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800249 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message_ExtensionOptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800250 want: fileDescP2_20180430.Messages().ByName("Message").Messages().ByName("ExtensionOptionalGroup"),
251 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800252 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180430.Message_ExtensionRepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800253 want: fileDescP2_20180430.Messages().ByName("Message").Messages().ByName("ExtensionRepeatedGroup"),
254 }}...)
255
256 fileDescP3_20180430 := mustLoadFileDesc(new(proto3_20180430.Message).Descriptor())
257 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -0800258 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20180430.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800259 want: fileDescP3_20180430.Enums().ByName("SiblingEnum"),
260 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800261 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20180430.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800262 want: fileDescP3_20180430.Messages().ByName("Message").Enums().ByName("ChildEnum"),
263 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800264 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180430.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800265 want: fileDescP3_20180430.Messages().ByName("SiblingMessage"),
266 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800267 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180430.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800268 want: fileDescP3_20180430.Messages().ByName("Message").Messages().ByName("ChildMessage"),
269 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800270 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180430.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800271 want: fileDescP3_20180430.Messages().ByName("Message"),
272 }}...)
273
274 fileDescP2_20180814 := mustLoadFileDesc(new(proto2_20180814.Message).Descriptor())
275 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -0800276 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20180814.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800277 want: fileDescP2_20180814.Enums().ByName("SiblingEnum"),
278 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800279 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20180814.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800280 want: fileDescP2_20180814.Messages().ByName("Message").Enums().ByName("ChildEnum"),
281 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800282 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800283 want: fileDescP2_20180814.Messages().ByName("SiblingMessage"),
284 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800285 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800286 want: fileDescP2_20180814.Messages().ByName("Message").Messages().ByName("ChildMessage"),
287 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800288 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800289 want: fileDescP2_20180814.Messages().ByName("Message"),
290 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800291 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message_NamedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800292 want: fileDescP2_20180814.Messages().ByName("Message").Messages().ByName("NamedGroup"),
293 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800294 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message_OptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800295 want: fileDescP2_20180814.Messages().ByName("Message").Messages().ByName("OptionalGroup"),
296 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800297 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message_RequiredGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800298 want: fileDescP2_20180814.Messages().ByName("Message").Messages().ByName("RequiredGroup"),
299 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800300 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message_RepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800301 want: fileDescP2_20180814.Messages().ByName("Message").Messages().ByName("RepeatedGroup"),
302 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800303 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message_OneofGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800304 want: fileDescP2_20180814.Messages().ByName("Message").Messages().ByName("OneofGroup"),
305 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800306 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message_ExtensionOptionalGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800307 want: fileDescP2_20180814.Messages().ByName("Message").Messages().ByName("ExtensionOptionalGroup"),
308 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800309 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20180814.Message_ExtensionRepeatedGroup))),
Joe Tsaiea118132018-11-11 17:56:21 -0800310 want: fileDescP2_20180814.Messages().ByName("Message").Messages().ByName("ExtensionRepeatedGroup"),
311 }}...)
312
313 fileDescP3_20180814 := mustLoadFileDesc(new(proto3_20180814.Message).Descriptor())
314 tests = append(tests, []struct{ got, want pref.Descriptor }{{
Joe Tsaif0c01e42018-11-06 13:05:20 -0800315 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20180814.SiblingEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800316 want: fileDescP3_20180814.Enums().ByName("SiblingEnum"),
317 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800318 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20180814.Message_ChildEnum(0))),
Joe Tsaiea118132018-11-11 17:56:21 -0800319 want: fileDescP3_20180814.Messages().ByName("Message").Enums().ByName("ChildEnum"),
320 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800321 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180814.SiblingMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800322 want: fileDescP3_20180814.Messages().ByName("SiblingMessage"),
323 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800324 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180814.Message_ChildMessage))),
Joe Tsaiea118132018-11-11 17:56:21 -0800325 want: fileDescP3_20180814.Messages().ByName("Message").Messages().ByName("ChildMessage"),
326 }, {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800327 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20180814.Message))),
Joe Tsaiea118132018-11-11 17:56:21 -0800328 want: fileDescP3_20180814.Messages().ByName("Message"),
329 }}...)
Joe Tsai90fe9962018-10-18 11:06:29 -0700330
Joe Tsai25cc69d2018-11-28 23:43:49 -0800331 fileDescP2_20181126 := mustLoadFileDesc(new(proto2_20181126.Message).Descriptor())
332 tests = append(tests, []struct{ got, want pref.Descriptor }{{
333 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20181126.SiblingEnum(0))),
334 want: fileDescP2_20181126.Enums().ByName("SiblingEnum"),
335 }, {
336 got: legacyLoadEnumDesc(reflect.TypeOf(proto2_20181126.Message_ChildEnum(0))),
337 want: fileDescP2_20181126.Messages().ByName("Message").Enums().ByName("ChildEnum"),
338 }, {
339 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.SiblingMessage))),
340 want: fileDescP2_20181126.Messages().ByName("SiblingMessage"),
341 }, {
342 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message_ChildMessage))),
343 want: fileDescP2_20181126.Messages().ByName("Message").Messages().ByName("ChildMessage"),
344 }, {
345 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message))),
346 want: fileDescP2_20181126.Messages().ByName("Message"),
347 }, {
348 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message_NamedGroup))),
349 want: fileDescP2_20181126.Messages().ByName("Message").Messages().ByName("NamedGroup"),
350 }, {
351 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message_OptionalGroup))),
352 want: fileDescP2_20181126.Messages().ByName("Message").Messages().ByName("OptionalGroup"),
353 }, {
354 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message_RequiredGroup))),
355 want: fileDescP2_20181126.Messages().ByName("Message").Messages().ByName("RequiredGroup"),
356 }, {
357 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message_RepeatedGroup))),
358 want: fileDescP2_20181126.Messages().ByName("Message").Messages().ByName("RepeatedGroup"),
359 }, {
360 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message_OneofGroup))),
361 want: fileDescP2_20181126.Messages().ByName("Message").Messages().ByName("OneofGroup"),
362 }, {
363 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message_ExtensionOptionalGroup))),
364 want: fileDescP2_20181126.Messages().ByName("Message").Messages().ByName("ExtensionOptionalGroup"),
365 }, {
366 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto2_20181126.Message_ExtensionRepeatedGroup))),
367 want: fileDescP2_20181126.Messages().ByName("Message").Messages().ByName("ExtensionRepeatedGroup"),
368 }}...)
369
370 fileDescP3_20181126 := mustLoadFileDesc(new(proto3_20181126.Message).Descriptor())
371 tests = append(tests, []struct{ got, want pref.Descriptor }{{
372 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20181126.SiblingEnum(0))),
373 want: fileDescP3_20181126.Enums().ByName("SiblingEnum"),
374 }, {
375 got: legacyLoadEnumDesc(reflect.TypeOf(proto3_20181126.Message_ChildEnum(0))),
376 want: fileDescP3_20181126.Messages().ByName("Message").Enums().ByName("ChildEnum"),
377 }, {
378 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20181126.SiblingMessage))),
379 want: fileDescP3_20181126.Messages().ByName("SiblingMessage"),
380 }, {
381 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20181126.Message_ChildMessage))),
382 want: fileDescP3_20181126.Messages().ByName("Message").Messages().ByName("ChildMessage"),
383 }, {
384 got: legacyLoadMessageDesc(reflect.TypeOf(new(proto3_20181126.Message))),
385 want: fileDescP3_20181126.Messages().ByName("Message"),
386 }}...)
387
Joe Tsai90fe9962018-10-18 11:06:29 -0700388 type list interface {
389 Len() int
390 pragma.DoNotImplement
391 }
392 opts := cmp.Options{
393 cmp.Transformer("", func(x list) []interface{} {
394 out := make([]interface{}, x.Len())
395 v := reflect.ValueOf(x)
396 for i := 0; i < x.Len(); i++ {
397 m := v.MethodByName("Get")
398 out[i] = m.Call([]reflect.Value{reflect.ValueOf(i)})[0].Interface()
399 }
400 return out
401 }),
402 cmp.Transformer("", func(x pref.Descriptor) map[string]interface{} {
403 out := make(map[string]interface{})
404 v := reflect.ValueOf(x)
405 for i := 0; i < v.NumMethod(); i++ {
406 name := v.Type().Method(i).Name
407 if m := v.Method(i); m.Type().NumIn() == 0 && m.Type().NumOut() == 1 {
408 switch name {
409 case "Index":
410 // Ignore index since legacy descriptors have no parent.
Damien Neil204f1c02018-10-23 15:03:38 -0700411 case "Options":
412 // Ignore descriptor options since protos are not cmperable.
Joe Tsaiea118132018-11-11 17:56:21 -0800413 case "Enums", "Messages", "Extensions":
Joe Tsai90fe9962018-10-18 11:06:29 -0700414 // Ignore nested message and enum declarations since
415 // legacy descriptors are all created standalone.
Joe Tsaiea118132018-11-11 17:56:21 -0800416 case "OneofType", "ExtendedType", "EnumType", "MessageType":
Joe Tsai90fe9962018-10-18 11:06:29 -0700417 // Avoid descending into a dependency to avoid a cycle.
418 // Just record the full name if available.
419 //
420 // TODO: Cycle support in cmp would be useful here.
421 v := m.Call(nil)[0]
422 if !v.IsNil() {
423 out[name] = v.Interface().(pref.Descriptor).FullName()
424 }
425 default:
426 out[name] = m.Call(nil)[0].Interface()
427 }
428 }
429 }
430 return out
431 }),
432 cmp.Transformer("", func(v pref.Value) interface{} {
433 return v.Interface()
434 }),
435 }
436
437 for _, tt := range tests {
438 t.Run(string(tt.want.FullName()), func(t *testing.T) {
439 if diff := cmp.Diff(&tt.want, &tt.got, opts); diff != "" {
440 t.Errorf("descriptor mismatch (-want, +got):\n%s", diff)
441 }
442 })
443 }
444}
Joe Tsaie2afdc22018-10-25 14:06:56 -0700445
Joe Tsaif0c01e42018-11-06 13:05:20 -0800446type legacyTestMessage struct {
Joe Tsai95b02902018-10-31 18:23:42 -0700447 XXX_unrecognized []byte
448 protoV1.XXX_InternalExtensions
449}
450
Joe Tsaif0c01e42018-11-06 13:05:20 -0800451func (*legacyTestMessage) Reset() {}
452func (*legacyTestMessage) String() string { return "" }
453func (*legacyTestMessage) ProtoMessage() {}
454func (*legacyTestMessage) ExtensionRangeArray() []protoV1.ExtensionRange {
455 return []protoV1.ExtensionRange{{Start: 10, End: 20}, {Start: 40, End: 80}, {Start: 10000, End: 20000}}
Joe Tsai95b02902018-10-31 18:23:42 -0700456}
457
Joe Tsaie2afdc22018-10-25 14:06:56 -0700458func TestLegacyUnknown(t *testing.T) {
459 rawOf := func(toks ...pack.Token) pref.RawFields {
460 return pref.RawFields(pack.Message(toks).Marshal())
461 }
462 raw1a := rawOf(pack.Tag{1, pack.VarintType}, pack.Svarint(-4321)) // 08c143
463 raw1b := rawOf(pack.Tag{1, pack.Fixed32Type}, pack.Uint32(0xdeadbeef)) // 0defbeadde
464 raw1c := rawOf(pack.Tag{1, pack.Fixed64Type}, pack.Float64(math.Pi)) // 09182d4454fb210940
465 raw2a := rawOf(pack.Tag{2, pack.BytesType}, pack.String("hello, world!")) // 120d68656c6c6f2c20776f726c6421
466 raw2b := rawOf(pack.Tag{2, pack.VarintType}, pack.Uvarint(1234)) // 10d209
467 raw3a := rawOf(pack.Tag{3, pack.StartGroupType}, pack.Tag{3, pack.EndGroupType}) // 1b1c
468 raw3b := rawOf(pack.Tag{3, pack.BytesType}, pack.Bytes("\xde\xad\xbe\xef")) // 1a04deadbeef
469
Joe Tsai95b02902018-10-31 18:23:42 -0700470 raw1 := rawOf(pack.Tag{1, pack.BytesType}, pack.Bytes("1")) // 0a0131
471 raw3 := rawOf(pack.Tag{3, pack.BytesType}, pack.Bytes("3")) // 1a0133
472 raw10 := rawOf(pack.Tag{10, pack.BytesType}, pack.Bytes("10")) // 52023130 - extension
473 raw15 := rawOf(pack.Tag{15, pack.BytesType}, pack.Bytes("15")) // 7a023135 - extension
474 raw26 := rawOf(pack.Tag{26, pack.BytesType}, pack.Bytes("26")) // d201023236
475 raw32 := rawOf(pack.Tag{32, pack.BytesType}, pack.Bytes("32")) // 8202023332
476 raw45 := rawOf(pack.Tag{45, pack.BytesType}, pack.Bytes("45")) // ea02023435 - extension
477 raw46 := rawOf(pack.Tag{45, pack.BytesType}, pack.Bytes("46")) // ea02023436 - extension
478 raw47 := rawOf(pack.Tag{45, pack.BytesType}, pack.Bytes("47")) // ea02023437 - extension
479 raw99 := rawOf(pack.Tag{99, pack.BytesType}, pack.Bytes("99")) // 9a06023939
480
Joe Tsaie2afdc22018-10-25 14:06:56 -0700481 joinRaw := func(bs ...pref.RawFields) (out []byte) {
482 for _, b := range bs {
483 out = append(out, b...)
484 }
485 return out
486 }
487
Joe Tsaif0c01e42018-11-06 13:05:20 -0800488 m := new(legacyTestMessage)
489 fs := MessageOf(m).UnknownFields()
Joe Tsai95b02902018-10-31 18:23:42 -0700490
Joe Tsaie2afdc22018-10-25 14:06:56 -0700491 if got, want := fs.Len(), 0; got != want {
492 t.Errorf("Len() = %d, want %d", got, want)
493 }
Joe Tsai95b02902018-10-31 18:23:42 -0700494 if got, want := m.XXX_unrecognized, joinRaw(); !bytes.Equal(got, want) {
Joe Tsaie2afdc22018-10-25 14:06:56 -0700495 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
496 }
497
498 fs.Set(1, raw1a)
499 fs.Set(1, append(fs.Get(1), raw1b...))
500 fs.Set(1, append(fs.Get(1), raw1c...))
501 if got, want := fs.Len(), 1; got != want {
502 t.Errorf("Len() = %d, want %d", got, want)
503 }
Joe Tsai95b02902018-10-31 18:23:42 -0700504 if got, want := m.XXX_unrecognized, joinRaw(raw1a, raw1b, raw1c); !bytes.Equal(got, want) {
Joe Tsaie2afdc22018-10-25 14:06:56 -0700505 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
506 }
507
508 fs.Set(2, raw2a)
509 if got, want := fs.Len(), 2; got != want {
510 t.Errorf("Len() = %d, want %d", got, want)
511 }
Joe Tsai95b02902018-10-31 18:23:42 -0700512 if got, want := m.XXX_unrecognized, joinRaw(raw1a, raw1b, raw1c, raw2a); !bytes.Equal(got, want) {
Joe Tsaie2afdc22018-10-25 14:06:56 -0700513 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
514 }
515
516 if got, want := fs.Get(1), joinRaw(raw1a, raw1b, raw1c); !bytes.Equal(got, want) {
517 t.Errorf("Get(%d) = %x, want %x", 1, got, want)
518 }
519 if got, want := fs.Get(2), joinRaw(raw2a); !bytes.Equal(got, want) {
520 t.Errorf("Get(%d) = %x, want %x", 2, got, want)
521 }
522 if got, want := fs.Get(3), joinRaw(); !bytes.Equal(got, want) {
523 t.Errorf("Get(%d) = %x, want %x", 3, got, want)
524 }
525
526 fs.Set(1, nil) // remove field 1
527 if got, want := fs.Len(), 1; got != want {
528 t.Errorf("Len() = %d, want %d", got, want)
529 }
Joe Tsai95b02902018-10-31 18:23:42 -0700530 if got, want := m.XXX_unrecognized, joinRaw(raw2a); !bytes.Equal(got, want) {
Joe Tsaie2afdc22018-10-25 14:06:56 -0700531 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
532 }
533
534 // Simulate manual appending of raw field data.
Joe Tsai95b02902018-10-31 18:23:42 -0700535 m.XXX_unrecognized = append(m.XXX_unrecognized, joinRaw(raw3a, raw1a, raw1b, raw2b, raw3b, raw1c)...)
Joe Tsaie2afdc22018-10-25 14:06:56 -0700536 if got, want := fs.Len(), 3; got != want {
537 t.Errorf("Len() = %d, want %d", got, want)
538 }
539
540 // Verify range iteration order.
541 var i int
542 want := []struct {
543 num pref.FieldNumber
544 raw pref.RawFields
545 }{
Joe Tsaie2afdc22018-10-25 14:06:56 -0700546 {2, joinRaw(raw2a, raw2b)},
Joe Tsai2d5a1692018-10-29 02:10:42 -0700547 {3, joinRaw(raw3a, raw3b)},
Joe Tsaie2afdc22018-10-25 14:06:56 -0700548 {1, joinRaw(raw1a, raw1b, raw1c)},
549 }
550 fs.Range(func(num pref.FieldNumber, raw pref.RawFields) bool {
551 if i < len(want) {
552 if num != want[i].num || !bytes.Equal(raw, want[i].raw) {
553 t.Errorf("Range(%d) = (%d, %x), want (%d, %x)", i, num, raw, want[i].num, want[i].raw)
554 }
555 } else {
556 t.Errorf("unexpected Range iteration: %d", i)
557 }
558 i++
559 return true
560 })
561
562 fs.Set(2, fs.Get(2)) // moves field 2 to the end
563 if got, want := fs.Len(), 3; got != want {
564 t.Errorf("Len() = %d, want %d", got, want)
565 }
Joe Tsai95b02902018-10-31 18:23:42 -0700566 if got, want := m.XXX_unrecognized, joinRaw(raw3a, raw1a, raw1b, raw3b, raw1c, raw2a, raw2b); !bytes.Equal(got, want) {
Joe Tsaie2afdc22018-10-25 14:06:56 -0700567 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
568 }
569 fs.Set(1, nil) // remove field 1
570 if got, want := fs.Len(), 2; got != want {
571 t.Errorf("Len() = %d, want %d", got, want)
572 }
Joe Tsai95b02902018-10-31 18:23:42 -0700573 if got, want := m.XXX_unrecognized, joinRaw(raw3a, raw3b, raw2a, raw2b); !bytes.Equal(got, want) {
Joe Tsaie2afdc22018-10-25 14:06:56 -0700574 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
575 }
576
577 // Remove all fields.
578 fs.Range(func(n pref.FieldNumber, b pref.RawFields) bool {
579 fs.Set(n, nil)
580 return true
581 })
582 if got, want := fs.Len(), 0; got != want {
583 t.Errorf("Len() = %d, want %d", got, want)
584 }
Joe Tsai95b02902018-10-31 18:23:42 -0700585 if got, want := m.XXX_unrecognized, joinRaw(); !bytes.Equal(got, want) {
Joe Tsaie2afdc22018-10-25 14:06:56 -0700586 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
587 }
Joe Tsai95b02902018-10-31 18:23:42 -0700588
589 fs.Set(1, raw1)
590 if got, want := fs.Len(), 1; got != want {
591 t.Errorf("Len() = %d, want %d", got, want)
592 }
593 if got, want := m.XXX_unrecognized, joinRaw(raw1); !bytes.Equal(got, want) {
594 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
595 }
596
597 fs.Set(45, raw45)
598 fs.Set(10, raw10) // extension
599 fs.Set(32, raw32)
600 fs.Set(1, nil) // deletion
601 fs.Set(26, raw26)
602 fs.Set(47, raw47) // extension
603 fs.Set(46, raw46) // extension
604 if got, want := fs.Len(), 6; got != want {
605 t.Errorf("Len() = %d, want %d", got, want)
606 }
607 if got, want := m.XXX_unrecognized, joinRaw(raw32, raw26); !bytes.Equal(got, want) {
608 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
609 }
610
611 // Verify iteration order.
612 i = 0
613 want = []struct {
614 num pref.FieldNumber
615 raw pref.RawFields
616 }{
617 {32, raw32},
618 {26, raw26},
619 {10, raw10}, // extension
620 {45, raw45}, // extension
621 {46, raw46}, // extension
622 {47, raw47}, // extension
623 }
624 fs.Range(func(num pref.FieldNumber, raw pref.RawFields) bool {
625 if i < len(want) {
626 if num != want[i].num || !bytes.Equal(raw, want[i].raw) {
627 t.Errorf("Range(%d) = (%d, %x), want (%d, %x)", i, num, raw, want[i].num, want[i].raw)
628 }
629 } else {
630 t.Errorf("unexpected Range iteration: %d", i)
631 }
632 i++
633 return true
634 })
635
636 // Perform partial deletion while iterating.
637 i = 0
638 fs.Range(func(num pref.FieldNumber, raw pref.RawFields) bool {
639 if i%2 == 0 {
640 fs.Set(num, nil)
641 }
642 i++
643 return true
644 })
645
646 if got, want := fs.Len(), 3; got != want {
647 t.Errorf("Len() = %d, want %d", got, want)
648 }
649 if got, want := m.XXX_unrecognized, joinRaw(raw26); !bytes.Equal(got, want) {
650 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
651 }
652
653 fs.Set(15, raw15) // extension
654 fs.Set(3, raw3)
655 fs.Set(99, raw99)
656 if got, want := fs.Len(), 6; got != want {
657 t.Errorf("Len() = %d, want %d", got, want)
658 }
659 if got, want := m.XXX_unrecognized, joinRaw(raw26, raw3, raw99); !bytes.Equal(got, want) {
660 t.Errorf("data mismatch:\ngot: %x\nwant: %x", got, want)
661 }
662
663 // Perform partial iteration.
664 i = 0
665 want = []struct {
666 num pref.FieldNumber
667 raw pref.RawFields
668 }{
669 {26, raw26},
670 {3, raw3},
671 }
672 fs.Range(func(num pref.FieldNumber, raw pref.RawFields) bool {
673 if i < len(want) {
674 if num != want[i].num || !bytes.Equal(raw, want[i].raw) {
675 t.Errorf("Range(%d) = (%d, %x), want (%d, %x)", i, num, raw, want[i].num, want[i].raw)
676 }
677 } else {
678 t.Errorf("unexpected Range iteration: %d", i)
679 }
680 i++
681 return i < 2
682 })
Joe Tsaie2afdc22018-10-25 14:06:56 -0700683}
Joe Tsaif0c01e42018-11-06 13:05:20 -0800684
685func TestLegactExtensions(t *testing.T) {
686 extensions := []pref.ExtensionType{
687 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
688 ExtendedType: (*legacyTestMessage)(nil),
689 ExtensionType: (*bool)(nil),
690 Field: 10000,
691 Name: "fizz.buzz.optional_bool",
692 Tag: "varint,10000,opt,name=optional_bool,json=optionalBool,def=1",
693 Filename: "fizz/buzz/test.proto",
694 }),
695 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
696 ExtendedType: (*legacyTestMessage)(nil),
697 ExtensionType: (*int32)(nil),
698 Field: 10001,
699 Name: "fizz.buzz.optional_int32",
700 Tag: "varint,10001,opt,name=optional_int32,json=optionalInt32,def=-12345",
701 Filename: "fizz/buzz/test.proto",
702 }),
703 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
704 ExtendedType: (*legacyTestMessage)(nil),
705 ExtensionType: (*uint32)(nil),
706 Field: 10002,
707 Name: "fizz.buzz.optional_uint32",
708 Tag: "varint,10002,opt,name=optional_uint32,json=optionalUint32,def=3200",
709 Filename: "fizz/buzz/test.proto",
710 }),
711 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
712 ExtendedType: (*legacyTestMessage)(nil),
713 ExtensionType: (*float32)(nil),
714 Field: 10003,
715 Name: "fizz.buzz.optional_float",
716 Tag: "fixed32,10003,opt,name=optional_float,json=optionalFloat,def=3.14159",
717 Filename: "fizz/buzz/test.proto",
718 }),
719 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
720 ExtendedType: (*legacyTestMessage)(nil),
721 ExtensionType: (*string)(nil),
722 Field: 10004,
723 Name: "fizz.buzz.optional_string",
724 Tag: "bytes,10004,opt,name=optional_string,json=optionalString,def=hello, \"world!\"\n",
725 Filename: "fizz/buzz/test.proto",
726 }),
727 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
728 ExtendedType: (*legacyTestMessage)(nil),
729 ExtensionType: ([]byte)(nil),
730 Field: 10005,
731 Name: "fizz.buzz.optional_bytes",
732 Tag: "bytes,10005,opt,name=optional_bytes,json=optionalBytes,def=dead\\336\\255\\276\\357beef",
733 Filename: "fizz/buzz/test.proto",
734 }),
735 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
736 ExtendedType: (*legacyTestMessage)(nil),
737 ExtensionType: (*proto2_20180125.Message_ChildEnum)(nil),
738 Field: 10006,
Joe Tsai87b955b2018-11-14 21:59:49 -0800739 Name: "fizz.buzz.optional_enum_v1",
740 Tag: "varint,10006,opt,name=optional_enum_v1,json=optionalEnumV1,enum=google.golang.org.proto2_20180125.Message_ChildEnum,def=0",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800741 Filename: "fizz/buzz/test.proto",
742 }),
743 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
744 ExtendedType: (*legacyTestMessage)(nil),
745 ExtensionType: (*proto2_20180125.Message_ChildMessage)(nil),
746 Field: 10007,
Joe Tsai87b955b2018-11-14 21:59:49 -0800747 Name: "fizz.buzz.optional_message_v1",
748 Tag: "bytes,10007,opt,name=optional_message_v1,json=optionalMessageV1",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800749 Filename: "fizz/buzz/test.proto",
750 }),
Joe Tsai87b955b2018-11-14 21:59:49 -0800751 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
752 ExtendedType: (*legacyTestMessage)(nil),
753 ExtensionType: (*EnumProto2)(nil),
754 Field: 10008,
755 Name: "fizz.buzz.optional_enum_v2",
756 Tag: "varint,10008,opt,name=optional_enum_v2,json=optionalEnumV2,enum=EnumProto2,def=57005",
757 Filename: "fizz/buzz/test.proto",
758 }),
759 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
760 ExtendedType: (*legacyTestMessage)(nil),
761 ExtensionType: (*EnumMessages)(nil),
762 Field: 10009,
763 Name: "fizz.buzz.optional_message_v2",
764 Tag: "bytes,10009,opt,name=optional_message_v2,json=optionalMessageV2",
765 Filename: "fizz/buzz/test.proto",
766 }),
Joe Tsaif0c01e42018-11-06 13:05:20 -0800767 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
768 ExtendedType: (*legacyTestMessage)(nil),
769 ExtensionType: ([]bool)(nil),
Joe Tsai87b955b2018-11-14 21:59:49 -0800770 Field: 10010,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800771 Name: "fizz.buzz.repeated_bool",
Joe Tsai87b955b2018-11-14 21:59:49 -0800772 Tag: "varint,10010,rep,name=repeated_bool,json=repeatedBool",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800773 Filename: "fizz/buzz/test.proto",
774 }),
775 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
776 ExtendedType: (*legacyTestMessage)(nil),
777 ExtensionType: ([]int32)(nil),
Joe Tsai87b955b2018-11-14 21:59:49 -0800778 Field: 10011,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800779 Name: "fizz.buzz.repeated_int32",
Joe Tsai87b955b2018-11-14 21:59:49 -0800780 Tag: "varint,10011,rep,name=repeated_int32,json=repeatedInt32",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800781 Filename: "fizz/buzz/test.proto",
782 }),
783 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
784 ExtendedType: (*legacyTestMessage)(nil),
785 ExtensionType: ([]uint32)(nil),
Joe Tsai87b955b2018-11-14 21:59:49 -0800786 Field: 10012,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800787 Name: "fizz.buzz.repeated_uint32",
Joe Tsai87b955b2018-11-14 21:59:49 -0800788 Tag: "varint,10012,rep,name=repeated_uint32,json=repeatedUint32",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800789 Filename: "fizz/buzz/test.proto",
790 }),
791 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
792 ExtendedType: (*legacyTestMessage)(nil),
793 ExtensionType: ([]float32)(nil),
Joe Tsai87b955b2018-11-14 21:59:49 -0800794 Field: 10013,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800795 Name: "fizz.buzz.repeated_float",
Joe Tsai87b955b2018-11-14 21:59:49 -0800796 Tag: "fixed32,10013,rep,name=repeated_float,json=repeatedFloat",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800797 Filename: "fizz/buzz/test.proto",
798 }),
799 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
800 ExtendedType: (*legacyTestMessage)(nil),
801 ExtensionType: ([]string)(nil),
Joe Tsai87b955b2018-11-14 21:59:49 -0800802 Field: 10014,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800803 Name: "fizz.buzz.repeated_string",
Joe Tsai87b955b2018-11-14 21:59:49 -0800804 Tag: "bytes,10014,rep,name=repeated_string,json=repeatedString",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800805 Filename: "fizz/buzz/test.proto",
806 }),
807 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
808 ExtendedType: (*legacyTestMessage)(nil),
809 ExtensionType: ([][]byte)(nil),
Joe Tsai87b955b2018-11-14 21:59:49 -0800810 Field: 10015,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800811 Name: "fizz.buzz.repeated_bytes",
Joe Tsai87b955b2018-11-14 21:59:49 -0800812 Tag: "bytes,10015,rep,name=repeated_bytes,json=repeatedBytes",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800813 Filename: "fizz/buzz/test.proto",
814 }),
815 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
816 ExtendedType: (*legacyTestMessage)(nil),
817 ExtensionType: ([]proto2_20180125.Message_ChildEnum)(nil),
Joe Tsai87b955b2018-11-14 21:59:49 -0800818 Field: 10016,
819 Name: "fizz.buzz.repeated_enum_v1",
820 Tag: "varint,10016,rep,name=repeated_enum_v1,json=repeatedEnumV1,enum=google.golang.org.proto2_20180125.Message_ChildEnum",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800821 Filename: "fizz/buzz/test.proto",
822 }),
823 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
824 ExtendedType: (*legacyTestMessage)(nil),
825 ExtensionType: ([]*proto2_20180125.Message_ChildMessage)(nil),
Joe Tsai87b955b2018-11-14 21:59:49 -0800826 Field: 10017,
827 Name: "fizz.buzz.repeated_message_v1",
828 Tag: "bytes,10017,rep,name=repeated_message_v1,json=repeatedMessageV1",
Joe Tsaif0c01e42018-11-06 13:05:20 -0800829 Filename: "fizz/buzz/test.proto",
830 }),
Joe Tsai87b955b2018-11-14 21:59:49 -0800831 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
832 ExtendedType: (*legacyTestMessage)(nil),
833 ExtensionType: ([]EnumProto2)(nil),
834 Field: 10018,
835 Name: "fizz.buzz.repeated_enum_v2",
836 Tag: "varint,10018,rep,name=repeated_enum_v2,json=repeatedEnumV2,enum=EnumProto2",
837 Filename: "fizz/buzz/test.proto",
838 }),
839 legacyExtensionTypeOf(&protoV1.ExtensionDesc{
840 ExtendedType: (*legacyTestMessage)(nil),
841 ExtensionType: ([]*EnumMessages)(nil),
842 Field: 10019,
843 Name: "fizz.buzz.repeated_message_v2",
844 Tag: "bytes,10019,rep,name=repeated_message_v2,json=repeatedMessageV2",
845 Filename: "fizz/buzz/test.proto",
846 }),
Joe Tsaif0c01e42018-11-06 13:05:20 -0800847 }
848 opts := cmp.Options{cmp.Comparer(func(x, y *proto2_20180125.Message_ChildMessage) bool {
849 return x == y // pointer compare messages for object identity
850 })}
851
852 m := new(legacyTestMessage)
853 fs := MessageOf(m).KnownFields()
854 ts := fs.ExtensionTypes()
855
856 if n := fs.Len(); n != 0 {
857 t.Errorf("KnownFields.Len() = %v, want 0", n)
858 }
859 if n := ts.Len(); n != 0 {
860 t.Errorf("ExtensionFieldTypes.Len() = %v, want 0", n)
861 }
862
863 // Register all the extension types.
864 for _, xt := range extensions {
865 ts.Register(xt)
866 }
867
868 // Check that getting the zero value returns the default value for scalars,
Joe Tsai4b7aff62018-11-14 14:05:19 -0800869 // nil for singular messages, and an empty list for repeated fields.
Joe Tsaif0c01e42018-11-06 13:05:20 -0800870 defaultValues := []interface{}{
871 bool(true),
872 int32(-12345),
873 uint32(3200),
874 float32(3.14159),
875 string("hello, \"world!\"\n"),
876 []byte("dead\xde\xad\xbe\xefbeef"),
877 proto2_20180125.Message_ALPHA,
878 nil,
Joe Tsai87b955b2018-11-14 21:59:49 -0800879 EnumProto2(0xdead),
880 nil,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800881 new([]bool),
882 new([]int32),
883 new([]uint32),
884 new([]float32),
885 new([]string),
886 new([][]byte),
887 new([]proto2_20180125.Message_ChildEnum),
888 new([]*proto2_20180125.Message_ChildMessage),
Joe Tsai87b955b2018-11-14 21:59:49 -0800889 new([]EnumProto2),
890 new([]*EnumMessages),
Joe Tsaif0c01e42018-11-06 13:05:20 -0800891 }
892 for i, xt := range extensions {
893 var got interface{}
Joe Tsaif6d4a422018-11-19 14:26:06 -0800894 if v := fs.Get(xt.Number()); v.IsValid() {
895 got = xt.InterfaceOf(v)
Joe Tsaif0c01e42018-11-06 13:05:20 -0800896 }
897 want := defaultValues[i]
898 if diff := cmp.Diff(want, got, opts); diff != "" {
899 t.Errorf("KnownFields.Get(%d) mismatch (-want +got):\n%v", xt.Number(), diff)
900 }
901 }
902
903 // All fields should be unpopulated.
904 for _, xt := range extensions {
905 if fs.Has(xt.Number()) {
906 t.Errorf("KnownFields.Has(%d) = true, want false", xt.Number())
907 }
908 }
909
Joe Tsai4b7aff62018-11-14 14:05:19 -0800910 // Set some values and append to values to the lists.
Joe Tsai009e0672018-11-27 18:45:07 -0800911 m1a := &proto2_20180125.Message_ChildMessage{F1: scalar.String("m1a")}
912 m1b := &proto2_20180125.Message_ChildMessage{F1: scalar.String("m2b")}
Joe Tsai87b955b2018-11-14 21:59:49 -0800913 m2a := &EnumMessages{EnumP2: EnumProto2(0x1b).Enum()}
914 m2b := &EnumMessages{EnumP2: EnumProto2(0x2b).Enum()}
Joe Tsaif0c01e42018-11-06 13:05:20 -0800915 setValues := []interface{}{
916 bool(false),
917 int32(-54321),
918 uint32(6400),
919 float32(2.71828),
920 string("goodbye, \"world!\"\n"),
921 []byte("live\xde\xad\xbe\xefchicken"),
922 proto2_20180125.Message_CHARLIE,
Joe Tsai87b955b2018-11-14 21:59:49 -0800923 m1a,
924 EnumProto2(0xbeef),
925 m2a,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800926 &[]bool{true},
927 &[]int32{-1000},
928 &[]uint32{1280},
929 &[]float32{1.6180},
930 &[]string{"zero"},
931 &[][]byte{[]byte("zero")},
932 &[]proto2_20180125.Message_ChildEnum{proto2_20180125.Message_BRAVO},
Joe Tsai87b955b2018-11-14 21:59:49 -0800933 &[]*proto2_20180125.Message_ChildMessage{m1b},
934 &[]EnumProto2{0xdead},
935 &[]*EnumMessages{m2b},
Joe Tsaif0c01e42018-11-06 13:05:20 -0800936 }
937 for i, xt := range extensions {
938 fs.Set(xt.Number(), xt.ValueOf(setValues[i]))
939 }
940 for i, xt := range extensions[len(extensions)/2:] {
941 v := extensions[i].ValueOf(setValues[i])
Joe Tsai4b7aff62018-11-14 14:05:19 -0800942 fs.Get(xt.Number()).List().Append(v)
Joe Tsaif0c01e42018-11-06 13:05:20 -0800943 }
944
945 // Get the values and check for equality.
946 getValues := []interface{}{
947 bool(false),
948 int32(-54321),
949 uint32(6400),
950 float32(2.71828),
951 string("goodbye, \"world!\"\n"),
952 []byte("live\xde\xad\xbe\xefchicken"),
953 proto2_20180125.Message_ChildEnum(proto2_20180125.Message_CHARLIE),
Joe Tsai87b955b2018-11-14 21:59:49 -0800954 m1a,
955 EnumProto2(0xbeef),
956 m2a,
Joe Tsaif0c01e42018-11-06 13:05:20 -0800957 &[]bool{true, false},
958 &[]int32{-1000, -54321},
959 &[]uint32{1280, 6400},
960 &[]float32{1.6180, 2.71828},
961 &[]string{"zero", "goodbye, \"world!\"\n"},
962 &[][]byte{[]byte("zero"), []byte("live\xde\xad\xbe\xefchicken")},
963 &[]proto2_20180125.Message_ChildEnum{proto2_20180125.Message_BRAVO, proto2_20180125.Message_CHARLIE},
Joe Tsai87b955b2018-11-14 21:59:49 -0800964 &[]*proto2_20180125.Message_ChildMessage{m1b, m1a},
965 &[]EnumProto2{0xdead, 0xbeef},
966 &[]*EnumMessages{m2b, m2a},
Joe Tsaif0c01e42018-11-06 13:05:20 -0800967 }
968 for i, xt := range extensions {
969 got := xt.InterfaceOf(fs.Get(xt.Number()))
970 want := getValues[i]
971 if diff := cmp.Diff(want, got, opts); diff != "" {
972 t.Errorf("KnownFields.Get(%d) mismatch (-want +got):\n%v", xt.Number(), diff)
973 }
974 }
975
Joe Tsai87b955b2018-11-14 21:59:49 -0800976 if n := fs.Len(); n != 20 {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800977 t.Errorf("KnownFields.Len() = %v, want 0", n)
978 }
Joe Tsai87b955b2018-11-14 21:59:49 -0800979 if n := ts.Len(); n != 20 {
980 t.Errorf("ExtensionFieldTypes.Len() = %v, want 20", n)
Joe Tsaif0c01e42018-11-06 13:05:20 -0800981 }
982
983 // Clear the field for all extension types.
Joe Tsaif6d4a422018-11-19 14:26:06 -0800984 for _, xt := range extensions[:len(extensions)/2] {
Joe Tsaif0c01e42018-11-06 13:05:20 -0800985 fs.Clear(xt.Number())
986 }
Joe Tsaif6d4a422018-11-19 14:26:06 -0800987 for i, xt := range extensions[len(extensions)/2:] {
988 if i%2 == 0 {
989 fs.Clear(xt.Number())
990 } else {
991 fs.Get(xt.Number()).List().Truncate(0)
992 }
993 }
Joe Tsaif0c01e42018-11-06 13:05:20 -0800994 if n := fs.Len(); n != 0 {
995 t.Errorf("KnownFields.Len() = %v, want 0", n)
996 }
Joe Tsai87b955b2018-11-14 21:59:49 -0800997 if n := ts.Len(); n != 20 {
998 t.Errorf("ExtensionFieldTypes.Len() = %v, want 20", n)
Joe Tsaif0c01e42018-11-06 13:05:20 -0800999 }
1000
1001 // De-register all extension types.
1002 for _, xt := range extensions {
1003 ts.Remove(xt)
1004 }
1005 if n := fs.Len(); n != 0 {
1006 t.Errorf("KnownFields.Len() = %v, want 0", n)
1007 }
1008 if n := ts.Len(); n != 0 {
1009 t.Errorf("ExtensionFieldTypes.Len() = %v, want 0", n)
1010 }
1011
1012}