mirror of https://github.com/tidwall/tile38.git
140 lines
3.9 KiB
Go
140 lines
3.9 KiB
Go
// Copyright 2015-2018 The NATS Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package test
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/nats-io/go-nats"
|
|
|
|
"github.com/nats-io/go-nats/encoders/protobuf"
|
|
pb "github.com/nats-io/go-nats/encoders/protobuf/testdata"
|
|
)
|
|
|
|
func NewProtoEncodedConn(tl TestLogger) *nats.EncodedConn {
|
|
ec, err := nats.NewEncodedConn(NewConnection(tl, TEST_PORT), protobuf.PROTOBUF_ENCODER)
|
|
if err != nil {
|
|
tl.Fatalf("Failed to create an encoded connection: %v\n", err)
|
|
}
|
|
return ec
|
|
}
|
|
|
|
func TestEncProtoMarshalStruct(t *testing.T) {
|
|
s := RunServerOnPort(TEST_PORT)
|
|
defer s.Shutdown()
|
|
|
|
ec := NewProtoEncodedConn(t)
|
|
defer ec.Close()
|
|
ch := make(chan bool)
|
|
|
|
me := &pb.Person{Name: "derek", Age: 22, Address: "140 New Montgomery St"}
|
|
me.Children = make(map[string]*pb.Person)
|
|
|
|
me.Children["sam"] = &pb.Person{Name: "sam", Age: 19, Address: "140 New Montgomery St"}
|
|
me.Children["meg"] = &pb.Person{Name: "meg", Age: 17, Address: "140 New Montgomery St"}
|
|
|
|
ec.Subscribe("protobuf_test", func(p *pb.Person) {
|
|
if !reflect.DeepEqual(p, me) {
|
|
t.Fatal("Did not receive the correct protobuf response")
|
|
}
|
|
ch <- true
|
|
})
|
|
|
|
ec.Publish("protobuf_test", me)
|
|
if e := Wait(ch); e != nil {
|
|
t.Fatal("Did not receive the message")
|
|
}
|
|
}
|
|
|
|
func TestEncProtoNilRequest(t *testing.T) {
|
|
s := RunServerOnPort(TEST_PORT)
|
|
defer s.Shutdown()
|
|
|
|
ec := NewProtoEncodedConn(t)
|
|
defer ec.Close()
|
|
|
|
testPerson := &pb.Person{Name: "Anatolii", Age: 25, Address: "Ukraine, Nikolaev"}
|
|
|
|
//Subscribe with empty interface shouldn't failed on empty message
|
|
ec.Subscribe("nil_test", func(_, reply string, _ interface{}) {
|
|
ec.Publish(reply, testPerson)
|
|
})
|
|
|
|
resp := new(pb.Person)
|
|
|
|
//Request with nil argument shouldn't failed with nil argument
|
|
err := ec.Request("nil_test", nil, resp, 100*time.Millisecond)
|
|
ec.Flush()
|
|
|
|
if err != nil {
|
|
t.Error("Fail to send empty message via encoded proto connection")
|
|
}
|
|
|
|
if !reflect.DeepEqual(testPerson, resp) {
|
|
t.Error("Fail to receive encoded response")
|
|
}
|
|
}
|
|
|
|
func BenchmarkProtobufMarshalStruct(b *testing.B) {
|
|
me := &pb.Person{Name: "derek", Age: 22, Address: "140 New Montgomery St"}
|
|
me.Children = make(map[string]*pb.Person)
|
|
|
|
me.Children["sam"] = &pb.Person{Name: "sam", Age: 19, Address: "140 New Montgomery St"}
|
|
me.Children["meg"] = &pb.Person{Name: "meg", Age: 17, Address: "140 New Montgomery St"}
|
|
|
|
encoder := &protobuf.ProtobufEncoder{}
|
|
for n := 0; n < b.N; n++ {
|
|
if _, err := encoder.Encode("protobuf_test", me); err != nil {
|
|
b.Fatal("Couldn't serialize object", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkPublishProtobufStruct(b *testing.B) {
|
|
// stop benchmark for set-up
|
|
b.StopTimer()
|
|
|
|
s := RunServerOnPort(TEST_PORT)
|
|
defer s.Shutdown()
|
|
|
|
ec := NewProtoEncodedConn(b)
|
|
defer ec.Close()
|
|
ch := make(chan bool)
|
|
|
|
me := &pb.Person{Name: "derek", Age: 22, Address: "140 New Montgomery St"}
|
|
me.Children = make(map[string]*pb.Person)
|
|
|
|
me.Children["sam"] = &pb.Person{Name: "sam", Age: 19, Address: "140 New Montgomery St"}
|
|
me.Children["meg"] = &pb.Person{Name: "meg", Age: 17, Address: "140 New Montgomery St"}
|
|
|
|
ec.Subscribe("protobuf_test", func(p *pb.Person) {
|
|
if !reflect.DeepEqual(p, me) {
|
|
b.Fatalf("Did not receive the correct protobuf response")
|
|
}
|
|
ch <- true
|
|
})
|
|
|
|
// resume benchmark
|
|
b.StartTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
ec.Publish("protobuf_test", me)
|
|
if e := Wait(ch); e != nil {
|
|
b.Fatal("Did not receive the message")
|
|
}
|
|
}
|
|
}
|