1 // Protocol Buffers for Go with Gadgets 2 // 3 // Copyright (c) 2019, The GoGo Authors. All rights reserved. 4 // http://github.com/gogo/protobuf 5 // 6 // Redistribution and use in source and binary forms, with or without 7 // modification, are permitted provided that the following conditions are 8 // met: 9 // 10 // * Redistributions of source code must retain the above copyright 11 // notice, this list of conditions and the following disclaimer. 12 // * Redistributions in binary form must reproduce the above 13 // copyright notice, this list of conditions and the following disclaimer 14 // in the documentation and/or other materials provided with the 15 // distribution. 16 // 17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 29 package protobuffer 30 31 import ( 32 "bytes" 33 "testing" 34 35 proto "github.com/gogo/protobuf/proto" 36 ) 37 38 func TestProtoBufferMarshal12(t *testing.T) { 39 i := int32(33) 40 m := &PBuffMarshal{Field1: []byte(string("Tester123")), Field2: &i} 41 mer := &PBuffMarshaler{Field1: []byte(string("Tester123")), Field2: &i} 42 testCases := []struct { 43 name string 44 size int 45 m proto.Message 46 }{ 47 {"MarshalLarge", 1024, m}, 48 {"MarshalSmall", 0, m}, 49 {"MarshalerLarge", 1024, mer}, 50 {"MarshalerSmall", 0, mer}, 51 } 52 for _, tc := range testCases { 53 t.Run(tc.name, func(t *testing.T) { 54 marshalCheck(t, tc.m, tc.size) 55 }) 56 } 57 } 58 59 func marshalCheck(t *testing.T, m proto.Message, size int) { 60 buf := proto.NewBuffer(make([]byte, 0, size)) 61 err := buf.Marshal(m) 62 if err != nil { 63 t.Fatal(err) 64 } 65 err = buf.Marshal(m) 66 if err != nil { 67 t.Fatal(err) 68 } 69 bufferBytes := buf.Bytes() 70 protoBytes, err := proto.Marshal(m) 71 if err != nil { 72 t.Fatal(err) 73 } 74 protoBytes = append(protoBytes, protoBytes...) 75 if !bytes.Equal(bufferBytes, protoBytes) { 76 t.Fatalf("proto.Buffer Marshal != proto.Marshal (%v != %v)\n", bufferBytes, protoBytes) 77 } 78 } 79