rx_test.go

package gofast

import "testing"
import "bytes"
import "reflect"

func TestReadtagp(t *testing.T) {
	// read tag and its payload
	payload := []byte{
		216, 43, 191, 216, 44, 2, 216, 45, 87, 159, 109, 116, 101, 115,
		116, 116, 114, 97, 110, 115, 112, 111, 114, 116, 1, 26, 0,
		160, 0, 0, 96, 255, 255}
	tag, bs := readtp(payload)
	// read tagMsg and its payload
	if tag != tagMsg {
		t.Errorf("expected %v, got %v", tagMsg, tag)
	} else if bytes.Compare(payload[2:], bs) != 0 {
		t.Errorf("expected %v, got %v", payload[2:], bs)
	}
}

func TestUnmessage(t *testing.T) {
	addr := <-testBindAddrs
	lis, serverch := newServer("server", addr, "") // init server
	transc := newClient("client", addr, "")
	if err := transc.Handshake(); err != nil { // init client
		t.Fatal(err)
	}
	transv := <-serverch

	// read tag and its payload
	payload := []byte{
		216, 43, 191, 216, 44, 2, 216, 45, 82,
		6, 99, 108, 105, 101, 110, 116, 1, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0,
		255}

	_, bs := readtp(payload)
	// unmessage
	var wai whoamiMsg
	ref := newWhoami(transc)
	bmsg := transc.unmessage(100, bs)
	wai.transport = transc
	wai.version = transc.version
	wai.Decode(bmsg.Data)
	if ref.name != wai.name {
		t.Errorf("expected %v, got %v", ref.name, wai.name)
	} else if !reflect.DeepEqual(ref.version, wai.version) {
		t.Errorf("expected %v, got %v", ref.version, wai.version)
	} else if ref.buffersize != wai.buffersize {
		t.Errorf("expected %v, got %v", ref.buffersize, wai.buffersize)
	} else if ref.tags != wai.tags {
		t.Errorf("expected %v, got %v", ref.tags, wai.tags)
	}

	lis.Close()
	transc.Close()
	transv.Close()
}

func BenchmarkReadtagp(b *testing.B) {
	// read tag and its payload
	payload := []byte{
		216, 43, 191, 216, 44, 2, 216, 45, 87,
		159, 109, 116, 101, 115, 116, 116, 114, 97, 110, 115, 112, 111,
		114, 116, 1, 26, 0, 160, 0, 0, 96, 255,
		255}
	for i := 0; i < b.N; i++ {
		readtp(payload)
	}
}

func BenchmarkUnmessage(b *testing.B) {
	addr := <-testBindAddrs
	lis, serverch := newServer("server", addr, "") // init server
	transc := newClient("client", addr, "")
	if err := transc.Handshake(); err != nil { // init client
		b.Fatal(err)
	}
	transv := <-serverch

	// read tag and its payload
	payload := []byte{
		216, 43, 191, 216, 44, 2, 216, 45, 78,
		159, 70, 99, 108, 105, 101, 110, 116, 1, 25, 2, 0, 64, 255,
		255}
	_, bs := readtp(payload)
	b.ResetTimer()
	// unmessage
	for i := 0; i < b.N; i++ {
		transc.unmessage(100, bs)
	}

	lis.Close()
	transc.Close()
	transv.Close()
}