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
32package proto
33
34import (
35	"bytes"
36	"testing"
37)
38
39func 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}
67
68func TestMarshalMessageSetJSON_UnknownType(t *testing.T) {
69	extMap := map[int32]Extension{12345: {}}
70	got, err := MarshalMessageSetJSON(extMap)
71	if err != nil {
72		t.Fatalf("MarshalMessageSetJSON: %v", err)
73	}
74	if want := []byte("{}"); !bytes.Equal(got, want) {
75		t.Errorf("MarshalMessageSetJSON(%v) = %q, want %q", extMap, got, want)
76	}
77}
78