Fixed build of testsuite.

3 files changed, 6 insertions, 4 deletions

diff --git a/tests/Network/BitTorrent/DHT/RoutingSpec.hs b/tests/Network/BitTorrent/DHT/RoutingSpec.hs
index aeccff5f..3077a52a 100644
--- a/tests/Network/BitTorrent/DHT/RoutingSpec.hs
+++ b/tests/Network/BitTorrent/DHT/RoutingSpec.hs
@@ -1,4 +1,5 @@
 {-# LANGUAGE ScopedTypeVariables #-}
+{-# LANGUAGE FlexibleContexts    #-}
 module Network.BitTorrent.DHT.RoutingSpec (spec) where
 import Control.Applicative
 import Control.Monad.State
@@ -73,4 +74,4 @@ spec = do
       let t1 = runSimulation e (T.insert n t)
       let t2 = runSimulation e (T.insert n t >>= T.insert n)
       t1 `shouldBe` t2
--}
\ No newline at end of file
+-}
diff --git a/tests/Network/BitTorrent/Exchange/BitfieldSpec.hs b/tests/Network/BitTorrent/Exchange/BitfieldSpec.hs
index 234965fa..1ba772f6 100644
--- a/tests/Network/BitTorrent/Exchange/BitfieldSpec.hs
+++ b/tests/Network/BitTorrent/Exchange/BitfieldSpec.hs
@@ -1,13 +1,14 @@
 {-# OPTIONS -fno-warn-orphans #-}
 module Network.BitTorrent.Exchange.BitfieldSpec (spec) where
 import Control.Applicative
+import Data.ByteString.Arbitrary
 import Test.Hspec
 import Test.QuickCheck
 
 import Network.BitTorrent.Exchange.Bitfield
 
 instance Arbitrary Bitfield where
-  arbitrary = fromBitmap <$> arbitrary
+  arbitrary = fromBitmap . fromABS <$> arbitrary
 
 spec :: Spec
-spec = return ()
\ No newline at end of file
+spec = return ()
diff --git a/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs b/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
index 3a3dddf5..73acb3fa 100644
--- a/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
+++ b/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
@@ -18,7 +18,7 @@ import Network.BitTorrent.Tracker.RPC.UDP
 
 
 validateInfo :: AnnounceQuery -> AnnounceInfo -> Expectation
-validateInfo _ Message.Failure {..} = error "validateInfo: failure"
+validateInfo _ Message.Failure {} = error "validateInfo: failure"
 validateInfo AnnounceQuery {..}  AnnounceInfo {..} = do
     respComplete    `shouldSatisfy` isJust
     respIncomplete  `shouldSatisfy` isJust