Fix #244

[btrtrc.git] / bep40_test.go

diff --git a/bep40_test.go b/bep40_test.go
index 4c82a5adb21978b26f6551e255284a8629ca90f5..78102149c1789323e9a899a88833e7f92b55d26c 100644 (file)
--- a/bep40_test.go
+++ b/bep40_test.go
@@ -8,20 +8,24 @@ import (
 )
 
 func TestBep40Priority(t *testing.T) {
-       assert.EqualValues(t, 0xec2d7224, bep40Priority(
+       assert.EqualValues(t, 0xec2d7224, bep40PriorityIgnoreError(
                ipPort{net.ParseIP("123.213.32.10"), 0},
                ipPort{net.ParseIP("98.76.54.32"), 0},
        ))
-       assert.EqualValues(t, 0xec2d7224, bep40Priority(
+       assert.EqualValues(t, 0xec2d7224, bep40PriorityIgnoreError(
                ipPort{net.ParseIP("98.76.54.32"), 0},
                ipPort{net.ParseIP("123.213.32.10"), 0},
        ))
-       assert.Equal(t, peerPriority(0x99568189), bep40Priority(
+       assert.Equal(t, peerPriority(0x99568189), bep40PriorityIgnoreError(
                ipPort{net.ParseIP("123.213.32.10"), 0},
                ipPort{net.ParseIP("123.213.32.234"), 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.EqualValues(t, "\x00\x00\x00\x00", func() []byte {
+               b, _ := bep40PriorityBytes(
+                       ipPort{net.ParseIP("123.213.32.234"), 0},
+                       ipPort{net.ParseIP("123.213.32.234"), 0},
+               )
+               return b
+       }())
+
 }