From debf122a285c6c9d701dcf628eda2e84c263276d Mon Sep 17 00:00:00 2001
From: zelig <viktor.tron@gmail.com>
Date: Wed, 17 Jan 2018 04:53:26 +0100
Subject: [PATCH] swarm/network: fix RegisterAndConnect test

---
 swarm/network/hive_test.go | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/swarm/network/hive_test.go b/swarm/network/hive_test.go
index 37aebd363e..daa9389291 100644
--- a/swarm/network/hive_test.go
+++ b/swarm/network/hive_test.go
@@ -43,18 +43,12 @@ func TestRegisterAndConnect(t *testing.T) {
 	pp.Start(s.Server)
 	defer pp.Stop()
 	// retrieve and broadcast
-	err := s.TestExchanges(p2ptest.Exchange{
-		Label: "getPeersMsg message",
-		Expects: []p2ptest.Expect{
-			p2ptest.Expect{
-				Code: 2,
-				Msg:  &subPeersMsg{0},
-				Peer: id,
-			},
-		},
+	err := s.TestDisconnected(&p2ptest.Disconnect{
+		Peer:  s.IDs[0],
+		Error: nil,
 	})
 
-	if err != nil {
-		t.Fatal(err)
+	if err == nil || err.Error() != "timed out waiting for peers to disconnect" {
+		t.Fatalf("expected peer to connect")
 	}
 }