swarm/network: fix RegisterAndConnect test

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{
+	err := s.TestDisconnected(&p2ptest.Disconnect{
-		Label: "getPeersMsg message",
+		Peer:  s.IDs[0],
-		Expects: []p2ptest.Expect{
+		Error: nil,
-			p2ptest.Expect{
-				Code: 2,
-				Msg:  &subPeersMsg{0},
-				Peer: id,
-			},
-		},
 	})
 
-	if err != nil {
+	if err == nil || err.Error() != "timed out waiting for peers to disconnect" {
-		t.Fatal(err)
+		t.Fatalf("expected peer to connect")
 	}
 }