From cfd7e74c25fa7d1b443f8527fca8afad14ef4419 Mon Sep 17 00:00:00 2001
From: Felix Lange <fjl@twurst.com>
Date: Wed, 26 Nov 2014 22:49:40 +0100
Subject: p2p: add test for NewPeer

---
 p2p/peer_test.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'p2p')

diff --git a/p2p/peer_test.go b/p2p/peer_test.go
index 0994683a2..d9640292f 100644
--- a/p2p/peer_test.go
+++ b/p2p/peer_test.go
@@ -6,6 +6,7 @@ import (
 	"encoding/hex"
 	"io/ioutil"
 	"net"
+	"reflect"
 	"testing"
 	"time"
 )
@@ -222,3 +223,17 @@ func TestPeerActivity(t *testing.T) {
 		t.Fatal("peer error", err)
 	}
 }
+
+func TestNewPeer(t *testing.T) {
+	id := NewSimpleClientIdentity("clientid", "version", "customid", "pubkey")
+	caps := []Cap{{"foo", 2}, {"bar", 3}}
+	p := NewPeer(id, caps)
+	if !reflect.DeepEqual(p.Caps(), caps) {
+		t.Errorf("Caps mismatch: got %v, expected %v", p.Caps(), caps)
+	}
+	if p.Identity() != id {
+		t.Errorf("Identity mismatch: got %v, expected %v", p.Identity(), id)
+	}
+	// Should not hang.
+	p.Disconnect(DiscAlreadyConnected)
+}
-- 
cgit v1.2.3