Merge pull request #21 from petroav/add-tests

test: add unit tests

Author: Stebalien

loggables_test.go

@@ -0,0 +1,74 @@
+package loggables
+
+import (
+	"errors"
+	ic "github.com/libp2p/go-libp2p-crypto"
+	peer "github.com/libp2p/go-libp2p-peer"
+
+	"crypto/rand"
+	ma "github.com/multiformats/go-multiaddr"
+	"net"
+	"testing"
+)
+
+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)
+		}
+	}
+
+}