vendor/github.com/golang/protobuf/proto/message_set_test.go

   1 // Go support for Protocol Buffers - Google's data interchange format
   2 //
   3 // Copyright 2014 The Go Authors.  All rights reserved.
   4 // https://github.com/golang/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 //     * Neither the name of Google Inc. nor the names of its
  17 // contributors may be used to endorse or promote products derived from
  18 // this software without specific prior written permission.
  19 //
  20 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  24 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  25 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  26 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  27 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  28 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  29 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  30 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31
  32 package proto
  33
  34 import (
  35         "bytes"
  36         "testing"
  37 )
  38
  39 func TestUnmarshalMessageSetWithDuplicate(t *testing.T) {
  40         // Check that a repeated message set entry will be concatenated.
  41         in := &messageSet{
  42                 Item: []*_MessageSet_Item{
  43                         {TypeId: Int32(12345), Message: []byte("hoo")},
  44                         {TypeId: Int32(12345), Message: []byte("hah")},
  45                 },
  46         }
  47         b, err := Marshal(in)
  48         if err != nil {
  49                 t.Fatalf("Marshal: %v", err)
  50         }
  51         t.Logf("Marshaled bytes: %q", b)
  52
  53         var extensions XXX_InternalExtensions
  54         if err := UnmarshalMessageSet(b, &extensions); err != nil {
  55                 t.Fatalf("UnmarshalMessageSet: %v", err)
  56         }
  57         ext, ok := extensions.p.extensionMap[12345]
  58         if !ok {
  59                 t.Fatalf("Didn't retrieve extension 12345; map is %v", extensions.p.extensionMap)
  60         }
  61         // Skip wire type/field number and length varints.
  62         got := skipVarint(skipVarint(ext.enc))
  63         if want := []byte("hoohah"); !bytes.Equal(got, want) {
  64                 t.Errorf("Combined extension is %q, want %q", got, want)
  65         }
  66 }