mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 14:20:23 -04:00
171 lines
5.4 KiB
Go
171 lines
5.4 KiB
Go
![]() |
// Go support for Protocol Buffers - Google's data interchange format
|
||
|
//
|
||
|
// Copyright 2017 The Go Authors. All rights reserved.
|
||
|
// https://github.com/golang/protobuf
|
||
|
//
|
||
|
// Redistribution and use in source and binary forms, with or without
|
||
|
// modification, are permitted provided that the following conditions are
|
||
|
// met:
|
||
|
//
|
||
|
// * Redistributions of source code must retain the above copyright
|
||
|
// notice, this list of conditions and the following disclaimer.
|
||
|
// * Redistributions in binary form must reproduce the above
|
||
|
// copyright notice, this list of conditions and the following disclaimer
|
||
|
// in the documentation and/or other materials provided with the
|
||
|
// distribution.
|
||
|
// * Neither the name of Google Inc. nor the names of its
|
||
|
// contributors may be used to endorse or promote products derived from
|
||
|
// this software without specific prior written permission.
|
||
|
//
|
||
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
||
|
package proto_test
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
"github.com/golang/protobuf/proto"
|
||
|
|
||
|
proto3pb "github.com/golang/protobuf/proto/proto3_proto"
|
||
|
pb "github.com/golang/protobuf/proto/test_proto"
|
||
|
)
|
||
|
|
||
|
func TestDiscardUnknown(t *testing.T) {
|
||
|
tests := []struct {
|
||
|
desc string
|
||
|
in, want proto.Message
|
||
|
}{{
|
||
|
desc: "Nil",
|
||
|
in: nil, want: nil, // Should not panic
|
||
|
}, {
|
||
|
desc: "NilPtr",
|
||
|
in: (*proto3pb.Message)(nil), want: (*proto3pb.Message)(nil), // Should not panic
|
||
|
}, {
|
||
|
desc: "Nested",
|
||
|
in: &proto3pb.Message{
|
||
|
Name: "Aaron",
|
||
|
Nested: &proto3pb.Nested{Cute: true, XXX_unrecognized: []byte("blah")},
|
||
|
XXX_unrecognized: []byte("blah"),
|
||
|
},
|
||
|
want: &proto3pb.Message{
|
||
|
Name: "Aaron",
|
||
|
Nested: &proto3pb.Nested{Cute: true},
|
||
|
},
|
||
|
}, {
|
||
|
desc: "Slice",
|
||
|
in: &proto3pb.Message{
|
||
|
Name: "Aaron",
|
||
|
Children: []*proto3pb.Message{
|
||
|
{Name: "Sarah", XXX_unrecognized: []byte("blah")},
|
||
|
{Name: "Abraham", XXX_unrecognized: []byte("blah")},
|
||
|
},
|
||
|
XXX_unrecognized: []byte("blah"),
|
||
|
},
|
||
|
want: &proto3pb.Message{
|
||
|
Name: "Aaron",
|
||
|
Children: []*proto3pb.Message{
|
||
|
{Name: "Sarah"},
|
||
|
{Name: "Abraham"},
|
||
|
},
|
||
|
},
|
||
|
}, {
|
||
|
desc: "OneOf",
|
||
|
in: &pb.Communique{
|
||
|
Union: &pb.Communique_Msg{&pb.Strings{
|
||
|
StringField: proto.String("123"),
|
||
|
XXX_unrecognized: []byte("blah"),
|
||
|
}},
|
||
|
XXX_unrecognized: []byte("blah"),
|
||
|
},
|
||
|
want: &pb.Communique{
|
||
|
Union: &pb.Communique_Msg{&pb.Strings{StringField: proto.String("123")}},
|
||
|
},
|
||
|
}, {
|
||
|
desc: "Map",
|
||
|
in: &pb.MessageWithMap{MsgMapping: map[int64]*pb.FloatingPoint{
|
||
|
0x4002: &pb.FloatingPoint{
|
||
|
Exact: proto.Bool(true),
|
||
|
XXX_unrecognized: []byte("blah"),
|
||
|
},
|
||
|
}},
|
||
|
want: &pb.MessageWithMap{MsgMapping: map[int64]*pb.FloatingPoint{
|
||
|
0x4002: &pb.FloatingPoint{Exact: proto.Bool(true)},
|
||
|
}},
|
||
|
}, {
|
||
|
desc: "Extension",
|
||
|
in: func() proto.Message {
|
||
|
m := &pb.MyMessage{
|
||
|
Count: proto.Int32(42),
|
||
|
Somegroup: &pb.MyMessage_SomeGroup{
|
||
|
GroupField: proto.Int32(6),
|
||
|
XXX_unrecognized: []byte("blah"),
|
||
|
},
|
||
|
XXX_unrecognized: []byte("blah"),
|
||
|
}
|
||
|
proto.SetExtension(m, pb.E_Ext_More, &pb.Ext{
|
||
|
Data: proto.String("extension"),
|
||
|
XXX_unrecognized: []byte("blah"),
|
||
|
})
|
||
|
return m
|
||
|
}(),
|
||
|
want: func() proto.Message {
|
||
|
m := &pb.MyMessage{
|
||
|
Count: proto.Int32(42),
|
||
|
Somegroup: &pb.MyMessage_SomeGroup{GroupField: proto.Int32(6)},
|
||
|
}
|
||
|
proto.SetExtension(m, pb.E_Ext_More, &pb.Ext{Data: proto.String("extension")})
|
||
|
return m
|
||
|
}(),
|
||
|
}}
|
||
|
|
||
|
// Test the legacy code path.
|
||
|
for _, tt := range tests {
|
||
|
// Clone the input so that we don't alter the original.
|
||
|
in := tt.in
|
||
|
if in != nil {
|
||
|
in = proto.Clone(tt.in)
|
||
|
}
|
||
|
|
||
|
var m LegacyMessage
|
||
|
m.Message, _ = in.(*proto3pb.Message)
|
||
|
m.Communique, _ = in.(*pb.Communique)
|
||
|
m.MessageWithMap, _ = in.(*pb.MessageWithMap)
|
||
|
m.MyMessage, _ = in.(*pb.MyMessage)
|
||
|
proto.DiscardUnknown(&m)
|
||
|
if !proto.Equal(in, tt.want) {
|
||
|
t.Errorf("test %s/Legacy, expected unknown fields to be discarded\ngot %v\nwant %v", tt.desc, in, tt.want)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
for _, tt := range tests {
|
||
|
proto.DiscardUnknown(tt.in)
|
||
|
if !proto.Equal(tt.in, tt.want) {
|
||
|
t.Errorf("test %s, expected unknown fields to be discarded\ngot %v\nwant %v", tt.desc, tt.in, tt.want)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// LegacyMessage is a proto.Message that has several nested messages.
|
||
|
// This does not have the XXX_DiscardUnknown method and so forces DiscardUnknown
|
||
|
// to use the legacy fallback logic.
|
||
|
type LegacyMessage struct {
|
||
|
Message *proto3pb.Message
|
||
|
Communique *pb.Communique
|
||
|
MessageWithMap *pb.MessageWithMap
|
||
|
MyMessage *pb.MyMessage
|
||
|
}
|
||
|
|
||
|
func (m *LegacyMessage) Reset() { *m = LegacyMessage{} }
|
||
|
func (m *LegacyMessage) String() string { return proto.CompactTextString(m) }
|
||
|
func (*LegacyMessage) ProtoMessage() {}
|