Update some tests now that adding connections is more reliable

diff --git a/client_test.go b/client_test.go
index 63bcd6637bd0cf9d3802e849353307f45c44e3f4..4d9c56ca9a4b3d7a459e35eb168ba5305c20a436 100644 (file)
--- a/client_test.go
+++ b/client_test.go
@@ -498,17 +498,8 @@ func TestSeedAfterDownloading(t *testing.T) {
        }()
        done := make(chan struct{})
        defer close(done)
-       go func() {
-               for {
-                       go leecherGreeting.AddClientPeer(seeder)
-                       go leecherGreeting.AddClientPeer(leecherLeecher)
-                       select {
-                       case <-done:
-                               return
-                       case <-time.After(time.Second):
-                       }
-               }
-       }()
+       go leecherGreeting.AddClientPeer(seeder)
+       go leecherGreeting.AddClientPeer(leecherLeecher)
        wg.Add(1)
        go func() {
                defer wg.Done()
@@ -860,16 +851,7 @@ func testDownloadCancel(t *testing.T, ps testDownloadCancelParams) {
        leecherGreeting.cl.mu.Unlock()
        done := make(chan struct{})
        defer close(done)
-       go func() {
-               for {
-                       leecherGreeting.AddClientPeer(seeder)
-                       select {
-                       case <-done:
-                               return
-                       case <-time.After(time.Second):
-                       }
-               }
-       }()
+       go leecherGreeting.AddClientPeer(seeder)
        completes := make(map[int]bool, 3)
        expected := func() map[int]bool {
                if ps.Cancel {