loggables_test.go

package loggables

import (
	"crypto/rand"
	"errors"
	"net"
	"testing"

	ic "github.com/libp2p/go-libp2p-core/crypto"
	"github.com/libp2p/go-libp2p-core/peer"

	ma "github.com/multiformats/go-multiaddr"
)

func TestNetConn(t *testing.T) {
	connA, _ := net.Pipe()
	loggable := NetConn(connA)
	if _, ok := loggable.Loggable()["localAddr"]; !ok {
		t.Fatalf("loggable missing localAddr")
	}
	if _, ok := loggable.Loggable()["remoteAddr"]; !ok {
		t.Fatalf("loggable missing remoteAddr")
	}
}

func TestError(t *testing.T) {
	loggable := Error(errors.New("test"))
	if _, ok := loggable.Loggable()["error"]; !ok {
		t.Fatalf("loggable missing error")
	}
}

func TestUuid(t *testing.T) {
	loggable := Uuid("testKey")
	if _, ok := loggable.Loggable()["testKey"]; !ok {
		t.Fatalf("loggable does not contain expected key")
	}
}

func TestDial(t *testing.T) {
	leftPriv, _, _ := ic.GenerateECDSAKeyPair(rand.Reader)
	lid, err := peer.IDFromPrivateKey(leftPriv)
	if err != nil {
		t.Fatalf("failed to create leftId: %s", err.Error())
	}
	rightPriv, _, _ := ic.GenerateECDSAKeyPair(rand.Reader)
	rid, err := peer.IDFromPrivateKey(rightPriv)
	if err != nil {
		t.Fatalf("failed to create rightId: %s", err.Error())
	}
	laddr, err := ma.NewMultiaddr("/ip4/127.0.0.1/tcp/80")
	if err != nil {
		t.Fatalf("failed to create laddr: %s", err.Error())
	}
	raddr, err := ma.NewMultiaddr("/ip4/127.0.0.1/tcp/81")
	if err != nil {
		t.Fatalf("failed to create raddr: %s", err.Error())
	}

	metadata := Dial("test", lid, rid, laddr, raddr)
	loggable := metadata.Loggable()

	expected := [][]string{
		{"localPeer", lid.Pretty()},
		{"localAddr", laddr.String()},
		{"remotePeer", rid.Pretty()},
		{"remoteAddr", raddr.String()},
	}
	for _, tuple := range expected {
		if actual, ok := loggable[tuple[0]]; !ok || actual != tuple[1] {
			t.Fatalf("Expected %s but got %s", tuple[1], actual)
		}
	}

}