Use relative availabilities to determine piece request order

[btrtrc.git] / client_test.go

diff --git a/client_test.go b/client_test.go
index cbfe48bbee9323fcc5b54ec8b498e2a9dcb77f94..f4c3085162232e095a0c6acfc6b3804945f2c282 100644 (file)
--- a/client_test.go
+++ b/client_test.go
@@ -537,8 +537,10 @@ func TestPeerInvalidHave(t *testing.T) {
        assert.True(t, _new)
        defer tt.Drop()
        cn := &PeerConn{Peer: Peer{
-               t: tt,
+               t:         tt,
+               callbacks: &cfg.Callbacks,
        }}
+       tt.conns[cn] = struct{}{}
        cn.peerImpl = cn
        cl.lock()
        defer cl.unlock()