You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package protobuf
|
|
|
|
import (
|
|
"git.noahlan.cn/noahlan/nnet/serialize/protobuf/testdata"
|
|
"google.golang.org/protobuf/proto"
|
|
"testing"
|
|
)
|
|
|
|
func TestProtobufSerializer_Serialize(t *testing.T) {
|
|
m := &testdata.Ping{Content: "hello"}
|
|
s := NewSerializer()
|
|
|
|
b, err := s.Marshal(m)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
m1 := &testdata.Ping{}
|
|
if err := s.Unmarshal(b, m1); err != nil {
|
|
t.Fatalf("unmarshal failed: %v", err)
|
|
}
|
|
// refer: https://developers.google.com/protocol-buffers/docs/reference/go/faq#deepequal
|
|
if !proto.Equal(m, m1) {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func BenchmarkSerializer_Serialize(b *testing.B) {
|
|
m := &testdata.Ping{Content: "hello"}
|
|
s := NewSerializer()
|
|
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
if _, err := s.Marshal(m); err != nil {
|
|
b.Fatalf("unmarshal failed: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkSerializer_Deserialize(b *testing.B) {
|
|
m := &testdata.Ping{Content: "hello"}
|
|
s := NewSerializer()
|
|
|
|
d, err := s.Marshal(m)
|
|
if err != nil {
|
|
b.Error(err)
|
|
}
|
|
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
m1 := &testdata.Ping{}
|
|
if err := s.Unmarshal(d, m1); err != nil {
|
|
b.Fatalf("unmarshal failed: %v", err)
|
|
}
|
|
}
|
|
}
|