+ Add initial tests for bitfields.

2 files changed, 36 insertions, 17 deletions

diff --git a/tests/encoding.hs b/tests/Encoding.hs
index 405d2eb1..bd452975 100644
--- a/tests/encoding.hs
+++ b/tests/Encoding.hs
@@ -1,15 +1,14 @@
 {-# LANGUAGE StandaloneDeriving #-}
 {-# LANGUAGE GeneralizedNewtypeDeriving #-}
 {-# OPTIONS -fno-warn-orphans #-}
-module Main (main) where
+module Encoding where
 
 import Control.Applicative
 import Data.Word
 import           Data.ByteString (ByteString)
 import qualified Data.ByteString as B
 import Data.Serialize
-
-import Test.Framework (defaultMain, Test)
+import Test.Framework (Test)
 import Test.Framework.Providers.QuickCheck2 (testProperty)
 import Test.QuickCheck
 
@@ -85,17 +84,3 @@ test_scrape_url = zipWith mkTest [1 :: Int ..] (check `map` tests)
       ]
 
     mkTest i = testProperty ("scrape test #" ++ show i)
-
-main :: IO ()
-main = defaultMain $
-       [ testProperty "Message encode <-> decode" $
-            prop_encoding (T :: T Message)
-
-       , testProperty "PeerID encode <-> decode" $
-            prop_encoding (T :: T PeerID)
-
-       , testProperty "Handshake encode <-> decode" $
-            prop_encoding (T :: T Handshake)
-       ] ++ test_scrape_url ++
-       [
-       ]
\ No newline at end of file
diff --git a/tests/Main.hs b/tests/Main.hs
new file mode 100644
index 00000000..0b158cb9
--- /dev/null
+++ b/tests/Main.hs
@@ -0,0 +1,34 @@
+module Main (main) where
+
+import Test.Framework (defaultMain)
+import Test.Framework.Providers.QuickCheck2 (testProperty)
+
+import Encoding
+
+import Network.BitTorrent
+
+prop_bitfieldDiff0 :: Bitfield -> Bool
+prop_bitfieldDiff0 b = (b `difference` empty (8 * bitfieldByteCount b)) == b
+
+prop_bitfieldDiff1 :: Bitfield -> Bool
+prop_bitfieldDiff1 b = em `difference` b == em
+  where
+    em = empty (8 * bitfieldByteCount b)
+
+
+main :: IO ()
+main = defaultMain $
+       [ testProperty "Message encode <-> decode" $
+            prop_encoding (T :: T Message)
+
+       , testProperty "PeerID encode <-> decode" $
+            prop_encoding (T :: T PeerID)
+
+       , testProperty "Handshake encode <-> decode" $
+            prop_encoding (T :: T Handshake)
+       ]
+       ++ test_scrape_url ++
+       [
+         testProperty "bitfield `difference` empty bitfield" prop_bitfieldDiff0
+       , testProperty "empty bitfield `difference` bitfield" prop_bitfieldDiff1
+       ]
\ No newline at end of file