| // Copyright 2018 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package legacy_test |
| |
| import ( |
| "reflect" |
| "testing" |
| |
| pimpl "google.golang.org/protobuf/internal/impl" |
| plegacy "google.golang.org/protobuf/internal/legacy" |
| ptype "google.golang.org/protobuf/internal/prototype" |
| pref "google.golang.org/protobuf/reflect/protoreflect" |
| piface "google.golang.org/protobuf/runtime/protoiface" |
| |
| proto2_20180125 "google.golang.org/protobuf/internal/testprotos/legacy/proto2.v1.0.0-20180125-92554152" |
| ) |
| |
| type legacyTestMessage struct { |
| XXX_unrecognized []byte |
| XXX_InternalExtensions map[int32]pimpl.ExtensionFieldV1 |
| } |
| |
| func (*legacyTestMessage) Reset() {} |
| func (*legacyTestMessage) String() string { return "" } |
| func (*legacyTestMessage) ProtoMessage() {} |
| func (*legacyTestMessage) ExtensionRangeArray() []piface.ExtensionRangeV1 { |
| return []piface.ExtensionRangeV1{{Start: 10000, End: 20000}} |
| } |
| |
| func mustMakeExtensionType(x *ptype.StandaloneExtension, v interface{}) pref.ExtensionType { |
| xd, err := ptype.NewExtension(x) |
| if err != nil { |
| panic(err) |
| } |
| return plegacy.ExtensionTypeOf(xd, reflect.TypeOf(v)) |
| } |
| |
| var ( |
| parentDesc = pimpl.Export{}.MessageDescriptorOf((*legacyTestMessage)(nil)) |
| messageV1Desc = pimpl.Export{}.MessageDescriptorOf((*proto2_20180125.Message_ChildMessage)(nil)) |
| |
| wantType = mustMakeExtensionType(&ptype.StandaloneExtension{ |
| FullName: "fizz.buzz.optional_message_v1", |
| Number: 10007, |
| Cardinality: pref.Optional, |
| Kind: pref.MessageKind, |
| MessageType: messageV1Desc, |
| ExtendedType: parentDesc, |
| }, (*proto2_20180125.Message_ChildMessage)(nil)) |
| wantDesc = &piface.ExtensionDescV1{ |
| ExtendedType: (*legacyTestMessage)(nil), |
| ExtensionType: (*proto2_20180125.Message_ChildMessage)(nil), |
| Field: 10007, |
| Name: "fizz.buzz.optional_message_v1", |
| Tag: "bytes,10007,opt,name=optional_message_v1", |
| } |
| ) |
| |
| func BenchmarkConvert(b *testing.B) { |
| b.ReportAllocs() |
| for i := 0; i < b.N; i++ { |
| xd := plegacy.Export{}.ExtensionDescFromType(wantType) |
| gotType := plegacy.Export{}.ExtensionTypeFromDesc(xd) |
| if gotType != wantType { |
| b.Fatalf("ExtensionType mismatch: got %p, want %p", gotType, wantType) |
| } |
| |
| xt := plegacy.Export{}.ExtensionTypeFromDesc(wantDesc) |
| gotDesc := plegacy.Export{}.ExtensionDescFromType(xt) |
| if gotDesc != wantDesc { |
| b.Fatalf("ExtensionDesc mismatch: got %p, want %p", gotDesc, wantDesc) |
| } |
| } |
| } |