1 package proto_test
2
3 import (
4 "fmt"
5 "reflect"
6 "testing"
7
8 "github.com/gogo/protobuf/proto"
9 ppb "github.com/gogo/protobuf/proto/proto3_proto"
10 )
11
12 func TestMap(t *testing.T) {
13 var b []byte
14 fmt.Sscanf("a2010c0a044b657931120456616c31a201130a044b657932120556616c3261120456616c32a201240a044b6579330d05000000120556616c33621a0556616c3361120456616c331505000000a20100a201260a044b657934130a07536f6d6555524c1209536f6d655469746c651a08536e69707065743114", "%x", &b)
15
16 var m ppb.Message
17 if err := proto.Unmarshal(b, &m); err != nil {
18 t.Fatalf("proto.Unmarshal error: %v", err)
19 }
20
21 got := m.StringMap
22 want := map[string]string{
23 "": "",
24 "Key1": "Val1",
25 "Key2": "Val2",
26 "Key3": "Val3",
27 "Key4": "",
28 }
29
30 if !reflect.DeepEqual(got, want) {
31 t.Errorf("maps differ:\ngot %#v\nwant %#v", got, want)
32 }
33 }
34
35 func marshalled() []byte {
36 m := &ppb.IntMaps{}
37 for i := 0; i < 1000; i++ {
38 m.Maps = append(m.Maps, &ppb.IntMap{
39 Rtt: map[int32]int32{1: 2},
40 })
41 }
42 b, err := proto.Marshal(m)
43 if err != nil {
44 panic(fmt.Sprintf("Can't marshal %+v: %v", m, err))
45 }
46 return b
47 }
48
49 func BenchmarkConcurrentMapUnmarshal(b *testing.B) {
50 in := marshalled()
51 b.RunParallel(func(pb *testing.PB) {
52 for pb.Next() {
53 var out ppb.IntMaps
54 if err := proto.Unmarshal(in, &out); err != nil {
55 b.Errorf("Can't unmarshal ppb.IntMaps: %v", err)
56 }
57 }
58 })
59 }
60
61 func BenchmarkSequentialMapUnmarshal(b *testing.B) {
62 in := marshalled()
63 b.ResetTimer()
64 for i := 0; i < b.N; i++ {
65 var out ppb.IntMaps
66 if err := proto.Unmarshal(in, &out); err != nil {
67 b.Errorf("Can't unmarshal ppb.IntMaps: %v", err)
68 }
69 }
70 }
71
View as plain text