peer_test.go

package gyre

import (
	"bytes"
	crand "crypto/rand"
	"io"
	"testing"

	zmq "github.com/pebbe/zmq4"
	"github.com/zeromq/gyre/zre/msg"
)

func TestPeer(t *testing.T) {

	mailbox, err := zmq.NewSocket(zmq.DEALER)
	if err != nil {
		t.Fatal(err)
	}
	mailbox.Bind("tcp://127.0.0.1:5551")

	me := make([]byte, 16)
	io.ReadFull(crand.Reader, me)

	you := make([]byte, 16)
	io.ReadFull(crand.Reader, you)

	peer := newPeer(string(you))
	if peer.connected {
		t.Fatal("Peer shouldn't be connected yet")
	}
	err = peer.connect(me, "tcp://127.0.0.1:5551")
	if err != nil {
		t.Fatal(err)
	}
	if !peer.connected {
		t.Fatal("Peer should be connected")
	}

	m := msg.NewHello()
	m.Endpoint = "tcp://127.0.0.1:5551"
	peer.send(m)

	exp, err := m.Marshal()
	if err != nil {
		t.Fatal(err)
	}

	got, err := mailbox.Recv(0)
	if err != nil {
		t.Fatal(err)
	}
	gotb := []byte(got)

	if !bytes.Equal(gotb, exp) {
		t.Error("Hello message was corrupted")
	}

	peer.destroy()
}