github.com/insionng/yougam@v0.0.0-20170714101924-2bc18d833463/libraries/golang/protobuf/proto/message_set_test.go (about)

     1  // Go support for Protocol Buffers - Google's data interchange format
     2  //
     3  // Copyright 2014 The Go Authors.  All rights reserved.
     4  // https://yougam/libraries/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  	m := make(map[int32]Extension)
    54  	if err := UnmarshalMessageSet(b, m); err != nil {
    55  		t.Fatalf("UnmarshalMessageSet: %v", err)
    56  	}
    57  	ext, ok := m[12345]
    58  	if !ok {
    59  		t.Fatalf("Didn't retrieve extension 12345; map is %v", m)
    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  }