Fix write error handling

[btrtrc.git] / prioritized_peers_test.go

diff --git a/prioritized_peers_test.go b/prioritized_peers_test.go
index c08c5a33e83d8cebe7a367dcc3cb26b2703398d4..5e61c25f1f56c4d8c0df6f92f3114a83a3d494cc 100644 (file)
--- a/prioritized_peers_test.go
+++ b/prioritized_peers_test.go
@@ -11,14 +11,14 @@ import (
 func TestPrioritizedPeers(t *testing.T) {
        pp := prioritizedPeers{
                om: btree.New(3),
-               getPrio: func(p Peer) peerPriority {
+               getPrio: func(p PeerInfo) peerPriority {
                        return bep40PriorityIgnoreError(p.addr(), IpPort{IP: net.ParseIP("0.0.0.0")})
                },
        }
        _, ok := pp.DeleteMin()
        assert.Panics(t, func() { pp.PopMax() })
        assert.False(t, ok)
-       ps := []Peer{
+       ps := []PeerInfo{
                {Addr: ipPortAddr{IP: net.ParseIP("1.2.3.4")}},
                {Addr: ipPortAddr{IP: net.ParseIP("1::2")}},
                {Addr: ipPortAddr{IP: net.ParseIP("")}},
@@ -30,14 +30,14 @@ func TestPrioritizedPeers(t *testing.T) {
                assert.True(t, pp.Add(p))
                assert.Equal(t, i+1, pp.Len())
        }
-       pop := func(expected *Peer) {
+       pop := func(expected *PeerInfo) {
                if expected == nil {
                        assert.Panics(t, func() { pp.PopMax() })
                } else {
                        assert.Equal(t, *expected, pp.PopMax())
                }
        }
-       min := func(expected *Peer) {
+       min := func(expected *PeerInfo) {
                i, ok := pp.DeleteMin()
                if expected == nil {
                        assert.False(t, ok)