Fix request heap pop bug

[btrtrc.git] / bep40_test.go

diff --git a/bep40_test.go b/bep40_test.go
index 4c82a5adb21978b26f6551e255284a8629ca90f5..48d5fdda857deb35152b12524f208c8c8ec6b438 100644 (file)
--- a/bep40_test.go
+++ b/bep40_test.go
@@ -8,20 +8,27 @@ import (
 )
 
 func TestBep40Priority(t *testing.T) {
-       assert.EqualValues(t, 0xec2d7224, bep40Priority(
-               ipPort{net.ParseIP("123.213.32.10"), 0},
-               ipPort{net.ParseIP("98.76.54.32"), 0},
+       assert.EqualValues(t, peerPriority(0xec2d7224), bep40PriorityIgnoreError(
+               IpPort{IP: net.ParseIP("123.213.32.10"), Port: 0},
+               IpPort{IP: net.ParseIP("98.76.54.32"), Port: 0},
        ))
-       assert.EqualValues(t, 0xec2d7224, bep40Priority(
-               ipPort{net.ParseIP("98.76.54.32"), 0},
-               ipPort{net.ParseIP("123.213.32.10"), 0},
+       assert.EqualValues(t, peerPriority(0xec2d7224), bep40PriorityIgnoreError(
+               IpPort{IP: net.ParseIP("98.76.54.32"), Port: 0},
+               IpPort{IP: net.ParseIP("123.213.32.10"), Port: 0},
        ))
-       assert.Equal(t, peerPriority(0x99568189), bep40Priority(
-               ipPort{net.ParseIP("123.213.32.10"), 0},
-               ipPort{net.ParseIP("123.213.32.234"), 0},
+       assert.Equal(t, peerPriority(0x99568189), bep40PriorityIgnoreError(
+               IpPort{IP: net.ParseIP("123.213.32.10"), Port: 0},
+               IpPort{IP: net.ParseIP("123.213.32.234"), Port: 0},
        ))
-       assert.EqualValues(t, "\x00\x00\x00\x00", bep40PriorityBytes(
-               ipPort{net.ParseIP("123.213.32.234"), 0},
-               ipPort{net.ParseIP("123.213.32.234"), 0},
+       assert.Equal(t, peerPriority(0x2b41d456), bep40PriorityIgnoreError(
+               IpPort{IP: net.ParseIP("206.248.98.111"), Port: 0},
+               IpPort{IP: net.ParseIP("142.147.89.224"), Port: 0},
        ))
+       assert.EqualValues(t, "\x00\x00\x00\x00", func() []byte {
+               b, _ := bep40PriorityBytes(
+                       IpPort{IP: net.ParseIP("123.213.32.234"), Port: 0},
+                       IpPort{IP: net.ParseIP("123.213.32.234"), Port: 0},
+               )
+               return b
+       }())
 }