bittorrent: fix check-bitfield, add a test.

factor · Jan 29, 2020 · 2300311 · 2300311
1 parent 6aa8f64
commit 2300311
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/extra/bittorrent/bittorrent-tests.factor b/extra/bittorrent/bittorrent-tests.factor
@@ -1,4 +1,9 @@
-USING: bittorrent io.sockets tools.test ;
+USING: bittorrent sequences io.sockets tools.test ;
+
+
+{ { t f t t f t f t } } [
+    8 <iota> [ B{ 0b10110101 } check-bitfield ] map
+] unit-test
 
 {
     {

diff --git a/extra/bittorrent/bittorrent.factor b/extra/bittorrent/bittorrent.factor
@@ -40,7 +40,7 @@ torrent-port [ 6881 ] initialize
     '[ _ _ [ set-bit ] [ clear-bit ] if ] change-nth ;
 
 : check-bitfield ( n bitfield -- ? )
-    [ bitfield-index swap ] dip nth bit? ;
+    [ bitfield-index swap ] dip nth swap bit? ;
 
 
 ! http