From a47367d67a836188aac141169d65235a1162f3ac Mon Sep 17 00:00:00 2001
From: vyzo <vyzo@hackzen.org>
Date: Tue, 24 Apr 2018 17:05:05 +0300
Subject: [PATCH] basic service test

---
 package.json |  12 ++++
 svc_test.go  | 156 +++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 168 insertions(+)
 create mode 100644 svc_test.go

diff --git a/package.json b/package.json
index cf8f9f1..c925c9c 100644
--- a/package.json
+++ b/package.json
@@ -51,6 +51,18 @@
       "hash": "QmZ4Qi3GaRbjcx28Sme5eMH7RQjGkt8wHxt2a65oLaeFEV",
       "name": "gogo-protobuf",
       "version": "0.0.0"
+    },
+    {
+      "author": "whyrusleeping",
+      "hash": "Qmc64U41EEB4nPG7wxjEqFwKJajS2f8kk5q2TvUrQf78Xu",
+      "name": "go-libp2p-blankhost",
+      "version": "0.2.7"
+    },
+    {
+      "author": "whyrusleeping",
+      "hash": "Qmb6BsZf6Y3kxffXMNTubGPF1w1bkHtpvhfYbmnwP3NQyw",
+      "name": "go-libp2p-netutil",
+      "version": "0.3.11"
     }
   ],
   "gxVersion": "0.12.1",
diff --git a/svc_test.go b/svc_test.go
new file mode 100644
index 0000000..662901f
--- /dev/null
+++ b/svc_test.go
@@ -0,0 +1,156 @@
+package rendezvous
+
+import (
+	"context"
+	"testing"
+
+	bhost "github.com/libp2p/go-libp2p-blankhost"
+	host "github.com/libp2p/go-libp2p-host"
+	netutil "github.com/libp2p/go-libp2p-netutil"
+)
+
+func getRendezvousHosts(t *testing.T, ctx context.Context, n int) []host.Host {
+	hosts := getNetHosts(t, ctx, n)
+	for i := 1; i < len(hosts); i++ {
+		connect(t, hosts[0], hosts[i])
+	}
+	return hosts
+}
+
+func getNetHosts(t *testing.T, ctx context.Context, n int) []host.Host {
+	var out []host.Host
+
+	for i := 0; i < n; i++ {
+		netw := netutil.GenSwarmNetwork(t, ctx)
+		h := bhost.NewBlankHost(netw)
+		out = append(out, h)
+	}
+
+	return out
+}
+
+func connect(t *testing.T, a, b host.Host) {
+	pinfo := a.Peerstore().PeerInfo(a.ID())
+	err := b.Connect(context.Background(), pinfo)
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func getRendezvousClients(t *testing.T, hosts []host.Host) []Rendezvous {
+	clients := make([]Rendezvous, len(hosts)-1)
+	for i, host := range hosts[1:] {
+		clients[i] = NewRendezvousClient(host, hosts[0].ID())
+	}
+	return clients
+}
+
+func TestSVCRegistrationAndDiscovery(t *testing.T) {
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	hosts := getRendezvousHosts(t, ctx, 5)
+
+	svc, err := NewRendezvousService(ctx, hosts[0], ":memory:")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer svc.DB.Close()
+
+	clients := getRendezvousClients(t, hosts)
+
+	err = clients[0].Register(ctx, "foo1", 60)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	rrs, cookie, err := clients[0].Discover(ctx, "foo1", 10, nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(rrs) != 1 {
+		t.Fatal("Expected 1 registration")
+	}
+	checkHostRegistration(t, rrs[0], hosts[1])
+
+	for i, client := range clients[1:] {
+		err = client.Register(ctx, "foo1", 60)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		rrs, cookie, err = clients[0].Discover(ctx, "foo1", 10, cookie)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(rrs) != 1 {
+			t.Fatal("Expected 1 registration")
+		}
+		checkHostRegistration(t, rrs[0], hosts[2+i])
+	}
+
+	for _, client := range clients[1:] {
+		rrs, _, err = client.Discover(ctx, "foo1", 10, nil)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(rrs) != 4 {
+			t.Fatal("Expected 4 registrations")
+		}
+
+		for j, rr := range rrs {
+			checkHostRegistration(t, rr, hosts[1+j])
+		}
+	}
+
+	err = clients[0].Unregister(ctx, "foo1")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	for _, client := range clients[0:] {
+		rrs, _, err = client.Discover(ctx, "foo1", 10, nil)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(rrs) != 3 {
+			t.Fatal("Expected 3 registrations")
+		}
+
+		for j, rr := range rrs {
+			checkHostRegistration(t, rr, hosts[2+j])
+		}
+	}
+
+	err = clients[1].Unregister(ctx, "")
+	for _, client := range clients[0:] {
+		rrs, _, err = client.Discover(ctx, "foo1", 10, nil)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(rrs) != 2 {
+			t.Fatal("Expected 2 registrations")
+		}
+
+		for j, rr := range rrs {
+			checkHostRegistration(t, rr, hosts[3+j])
+		}
+	}
+}
+
+func checkHostRegistration(t *testing.T, rr Registration, host host.Host) {
+	if rr.Peer.ID != host.ID() {
+		t.Fatal("bad registration: peer ID doesn't match host ID")
+	}
+	addrs := host.Addrs()
+	raddrs := rr.Peer.Addrs
+	if len(addrs) != len(raddrs) {
+		t.Fatal("bad registration: peer address length mismatch")
+	}
+	for i, addr := range addrs {
+		raddr := raddrs[i]
+		if !addr.Equal(raddr) {
+			t.Fatal("bad registration: peer address mismatch")
+		}
+	}
+}