Expose tracker RPC module

3 files changed, 10 insertions, 3 deletions

diff --git a/tests/Network/BitTorrent/Exchange/BlockSpec.hs b/tests/Network/BitTorrent/Exchange/BlockSpec.hs
index 158a30b4..f4301452 100644
--- a/tests/Network/BitTorrent/Exchange/BlockSpec.hs
+++ b/tests/Network/BitTorrent/Exchange/BlockSpec.hs
@@ -1,5 +1,6 @@
 module Network.BitTorrent.Exchange.BlockSpec (spec) where
 import Control.Applicative
+import Data.Maybe
 import Test.Hspec
 import Test.QuickCheck
 
@@ -12,8 +13,14 @@ instance Arbitrary a => Arbitrary (Block a) where
 instance Arbitrary BlockIx where
   arbitrary = BlockIx <$> arbitrary <*> arbitrary <*> arbitrary
 
+instance Arbitrary Bucket where
+  arbitrary = error "arbitrary: block bucket"
+
+instance Show Bucket where
+  show = error "show: bucket"
+
 spec :: Spec
 spec = do
   describe "bucket" $ do
     it "render to piece when it is full" $ property $ \ bkt ->
-      if full bkt then isJust (toPiece bkt)
\ No newline at end of file
+      full bkt == isJust (toPiece bkt)
\ No newline at end of file
diff --git a/tests/Network/BitTorrent/Tracker/RPC/HTTPSpec.hs b/tests/Network/BitTorrent/Tracker/RPC/HTTPSpec.hs
index 6ff68cdf..37029b75 100644
--- a/tests/Network/BitTorrent/Tracker/RPC/HTTPSpec.hs
+++ b/tests/Network/BitTorrent/Tracker/RPC/HTTPSpec.hs
@@ -1,4 +1,4 @@
-module Network.BitTorrent.Tracker.RPC.HTTPSpec (spec) where
+module Network.BitTorrent.Tracker.RPC.HTTPSpec (spec, trackerURIs) where
 
 import Control.Monad
 import Control.Monad.Trans
diff --git a/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs b/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
index 53babbb3..e46f15b6 100644
--- a/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
+++ b/tests/Network/BitTorrent/Tracker/RPC/UDPSpec.hs
@@ -1,5 +1,5 @@
 {-# LANGUAGE RecordWildCards #-}
-module Network.BitTorrent.Tracker.RPC.UDPSpec (spec) where
+module Network.BitTorrent.Tracker.RPC.UDPSpec (spec, trackerURIs) where
 
 import Control.Monad
 import Data.Default