| // Go support for Protocol Buffers - Google's data interchange format |
| // |
| // Copyright 2011 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" |
| |
| pb "./testdata" |
| . "github.com/golang/protobuf/proto" |
| ) |
| |
| // Four identical base messages. |
| // The init function adds extensions to some of them. |
| var messageWithoutExtension = &pb.MyMessage{Count: Int32(7)} |
| var messageWithExtension1a = &pb.MyMessage{Count: Int32(7)} |
| var messageWithExtension1b = &pb.MyMessage{Count: Int32(7)} |
| var messageWithExtension2 = &pb.MyMessage{Count: Int32(7)} |
| |
| // Two messages with non-message extensions. |
| var messageWithInt32Extension1 = &pb.MyMessage{Count: Int32(8)} |
| var messageWithInt32Extension2 = &pb.MyMessage{Count: Int32(8)} |
| |
| func init() { |
| ext1 := &pb.Ext{Data: String("Kirk")} |
| ext2 := &pb.Ext{Data: String("Picard")} |
| |
| // messageWithExtension1a has ext1, but never marshals it. |
| if err := SetExtension(messageWithExtension1a, pb.E_Ext_More, ext1); err != nil { |
| panic("SetExtension on 1a failed: " + err.Error()) |
| } |
| |
| // messageWithExtension1b is the unmarshaled form of messageWithExtension1a. |
| if err := SetExtension(messageWithExtension1b, pb.E_Ext_More, ext1); err != nil { |
| panic("SetExtension on 1b failed: " + err.Error()) |
| } |
| buf, err := Marshal(messageWithExtension1b) |
| if err != nil { |
| panic("Marshal of 1b failed: " + err.Error()) |
| } |
| messageWithExtension1b.Reset() |
| if err := Unmarshal(buf, messageWithExtension1b); err != nil { |
| panic("Unmarshal of 1b failed: " + err.Error()) |
| } |
| |
| // messageWithExtension2 has ext2. |
| if err := SetExtension(messageWithExtension2, pb.E_Ext_More, ext2); err != nil { |
| panic("SetExtension on 2 failed: " + err.Error()) |
| } |
| |
| if err := SetExtension(messageWithInt32Extension1, pb.E_Ext_Number, Int32(23)); err != nil { |
| panic("SetExtension on Int32-1 failed: " + err.Error()) |
| } |
| if err := SetExtension(messageWithInt32Extension1, pb.E_Ext_Number, Int32(24)); err != nil { |
| panic("SetExtension on Int32-2 failed: " + err.Error()) |
| } |
| } |
| |
| var EqualTests = []struct { |
| desc string |
| a, b Message |
| exp bool |
| }{ |
| {"different types", &pb.GoEnum{}, &pb.GoTestField{}, false}, |
| {"equal empty", &pb.GoEnum{}, &pb.GoEnum{}, true}, |
| {"nil vs nil", nil, nil, true}, |
| {"typed nil vs typed nil", (*pb.GoEnum)(nil), (*pb.GoEnum)(nil), true}, |
| {"typed nil vs empty", (*pb.GoEnum)(nil), &pb.GoEnum{}, false}, |
| {"different typed nil", (*pb.GoEnum)(nil), (*pb.GoTestField)(nil), false}, |
| |
| {"one set field, one unset field", &pb.GoTestField{Label: String("foo")}, &pb.GoTestField{}, false}, |
| {"one set field zero, one unset field", &pb.GoTest{Param: Int32(0)}, &pb.GoTest{}, false}, |
| {"different set fields", &pb.GoTestField{Label: String("foo")}, &pb.GoTestField{Label: String("bar")}, false}, |
| {"equal set", &pb.GoTestField{Label: String("foo")}, &pb.GoTestField{Label: String("foo")}, true}, |
| |
| {"repeated, one set", &pb.GoTest{F_Int32Repeated: []int32{2, 3}}, &pb.GoTest{}, false}, |
| {"repeated, different length", &pb.GoTest{F_Int32Repeated: []int32{2, 3}}, &pb.GoTest{F_Int32Repeated: []int32{2}}, false}, |
| {"repeated, different value", &pb.GoTest{F_Int32Repeated: []int32{2}}, &pb.GoTest{F_Int32Repeated: []int32{3}}, false}, |
| {"repeated, equal", &pb.GoTest{F_Int32Repeated: []int32{2, 4}}, &pb.GoTest{F_Int32Repeated: []int32{2, 4}}, true}, |
| {"repeated, nil equal nil", &pb.GoTest{F_Int32Repeated: nil}, &pb.GoTest{F_Int32Repeated: nil}, true}, |
| {"repeated, nil equal empty", &pb.GoTest{F_Int32Repeated: nil}, &pb.GoTest{F_Int32Repeated: []int32{}}, true}, |
| {"repeated, empty equal nil", &pb.GoTest{F_Int32Repeated: []int32{}}, &pb.GoTest{F_Int32Repeated: nil}, true}, |
| |
| { |
| "nested, different", |
| &pb.GoTest{RequiredField: &pb.GoTestField{Label: String("foo")}}, |
| &pb.GoTest{RequiredField: &pb.GoTestField{Label: String("bar")}}, |
| false, |
| }, |
| { |
| "nested, equal", |
| &pb.GoTest{RequiredField: &pb.GoTestField{Label: String("wow")}}, |
| &pb.GoTest{RequiredField: &pb.GoTestField{Label: String("wow")}}, |
| true, |
| }, |
| |
| {"bytes", &pb.OtherMessage{Value: []byte("foo")}, &pb.OtherMessage{Value: []byte("foo")}, true}, |
| {"bytes, empty", &pb.OtherMessage{Value: []byte{}}, &pb.OtherMessage{Value: []byte{}}, true}, |
| {"bytes, empty vs nil", &pb.OtherMessage{Value: []byte{}}, &pb.OtherMessage{Value: nil}, false}, |
| { |
| "repeated bytes", |
| &pb.MyMessage{RepBytes: [][]byte{[]byte("sham"), []byte("wow")}}, |
| &pb.MyMessage{RepBytes: [][]byte{[]byte("sham"), []byte("wow")}}, |
| true, |
| }, |
| |
| {"extension vs. no extension", messageWithoutExtension, messageWithExtension1a, false}, |
| {"extension vs. same extension", messageWithExtension1a, messageWithExtension1b, true}, |
| {"extension vs. different extension", messageWithExtension1a, messageWithExtension2, false}, |
| |
| {"int32 extension vs. itself", messageWithInt32Extension1, messageWithInt32Extension1, true}, |
| {"int32 extension vs. a different int32", messageWithInt32Extension1, messageWithInt32Extension2, false}, |
| |
| { |
| "message with group", |
| &pb.MyMessage{ |
| Count: Int32(1), |
| Somegroup: &pb.MyMessage_SomeGroup{ |
| GroupField: Int32(5), |
| }, |
| }, |
| &pb.MyMessage{ |
| Count: Int32(1), |
| Somegroup: &pb.MyMessage_SomeGroup{ |
| GroupField: Int32(5), |
| }, |
| }, |
| true, |
| }, |
| |
| { |
| "map same", |
| &pb.MessageWithMap{NameMapping: map[int32]string{1: "Ken"}}, |
| &pb.MessageWithMap{NameMapping: map[int32]string{1: "Ken"}}, |
| true, |
| }, |
| { |
| "map different entry", |
| &pb.MessageWithMap{NameMapping: map[int32]string{1: "Ken"}}, |
| &pb.MessageWithMap{NameMapping: map[int32]string{2: "Rob"}}, |
| false, |
| }, |
| { |
| "map different key only", |
| &pb.MessageWithMap{NameMapping: map[int32]string{1: "Ken"}}, |
| &pb.MessageWithMap{NameMapping: map[int32]string{2: "Ken"}}, |
| false, |
| }, |
| { |
| "map different value only", |
| &pb.MessageWithMap{NameMapping: map[int32]string{1: "Ken"}}, |
| &pb.MessageWithMap{NameMapping: map[int32]string{1: "Rob"}}, |
| false, |
| }, |
| } |
| |
| func TestEqual(t *testing.T) { |
| for _, tc := range EqualTests { |
| if res := Equal(tc.a, tc.b); res != tc.exp { |
| t.Errorf("%v: Equal(%v, %v) = %v, want %v", tc.desc, tc.a, tc.b, res, tc.exp) |
| } |
| } |
| } |