Selaa lähdekoodia

p2p: add test for NewPeer

Felix Lange 11 vuotta sitten
vanhempi
commit
cfd7e74c25
1 muutettua tiedostoa jossa 15 lisäystä ja 0 poistoa
  1. 15 0
      p2p/peer_test.go

+ 15 - 0
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)
+}