Extract waiting loop for peers into a function

swarm/network/streamer_test.go

@@ -17,6 +17,7 @@
 package network
 
 import (
+	"errors"
 	"io/ioutil"
 	"os"
 	"testing"
@@ -113,18 +114,9 @@ func TestStreamerRegisterIncoming(t *testing.T) {
 		}, nil
 	})
 
-	tick := time.NewTicker(10 * time.Millisecond)
+	err = waitForPeers(streamer, 1*time.Second)
-	timeout := time.NewTimer(1 * time.Second)
+	if err != nil {
-WAIT:
+		t.Fatal("timeout: peer is not created")
-	for {
-		select {
-		case <-tick.C:
-			if len(streamer.peers) > 0 {
-				break WAIT
-			}
-		case <-timeout.C:
-			t.Fatal("timeout")
-		}
 	}
 
 	err = streamer.Subscribe(tester.IDs[0], "foo", nil, 0, 0, Top, true)
@@ -132,3 +124,18 @@ WAIT:
 		t.Fatalf("Expected no error, got %v", err)
 	}
 }
+
+func waitForPeers(streamer *Streamer, timeout time.Duration) error {
+	ticker := time.NewTicker(10 * time.Millisecond)
+	timeoutTimer := time.NewTimer(timeout)
+	for {
+		select {
+		case <-ticker.C:
+			if len(streamer.peers) > 0 {
+				return nil
+			}
+		case <-timeoutTimer.C:
+			return errors.New("timeout")
+		}
+	}
+}