nnet/serialize/protobuf/protobuf_test.go

package protobuf

import (
	"git.noahlan.cn/northlan/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)
		}
	}
}