1package proto_test 2 3import ( 4 "fmt" 5 "reflect" 6 "testing" 7 8 "github.com/golang/protobuf/proto" 9 ppb "github.com/golang/protobuf/proto/proto3_proto" 10) 11 12func 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 35func marshaled() []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 49func BenchmarkConcurrentMapUnmarshal(b *testing.B) { 50 in := marshaled() 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 61func BenchmarkSequentialMapUnmarshal(b *testing.B) { 62 in := marshaled() 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