Fix pending peer counts in transfer tests

[btrtrc.git] / test / transfer_test.go

diff --git a/test/transfer_test.go b/test/transfer_test.go
index 4fe9d5623db684b55f27dcca9f9ff9aa89ded974..74dd656e685bcf3df31f839e7afb8a119fe6bb26 100644 (file)
--- a/test/transfer_test.go
+++ b/test/transfer_test.go
@@ -117,11 +117,11 @@ func testClientTransfer(t *testing.T, ps testClientTransferParams) {
        //logPieceStateChanges(leecherTorrent)
 
        // Now do some things with leecher and seeder.
-       leecherTorrent.AddClientPeer(seeder)
+       added := leecherTorrent.AddClientPeer(seeder)
        // The Torrent should not be interested in obtaining peers, so the one we
        // just added should be the only one.
        assert.False(t, leecherTorrent.Seeding())
-       assert.EqualValues(t, 1, leecherTorrent.Stats().PendingPeers)
+       assert.EqualValues(t, added, leecherTorrent.Stats().PendingPeers)
        r := leecherTorrent.NewReader()
        defer r.Close()
        if ps.Responsive {