client_test.go | 24 +++--------------------- diff --git a/client_test.go b/client_test.go index 63bcd6637bd0cf9d3802e849353307f45c44e3f4..4d9c56ca9a4b3d7a459e35eb168ba5305c20a436 100644 --- a/client_test.go +++ b/client_test.go @@ -498,17 +498,8 @@ assert.EqualValues(t, testutil.GreetingFileContents, b) }() 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 @@ } 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 {