Fix test

3 files changed, 25 insertions, 7 deletions

diff --git a/tests/Data/Torrent/ClientSpec.hs b/tests/Data/Torrent/ClientSpec.hs
index 0ff74f6f..4bc881c3 100644
--- a/tests/Data/Torrent/ClientSpec.hs
+++ b/tests/Data/Torrent/ClientSpec.hs
@@ -27,4 +27,7 @@ spec = do
       clientInfo "OP0123xxxxxxxxxxxxxx" `shouldBe` "Opera-123"
 
     it "decode ML donkey style peer id" $ do
-      clientInfo "-ML2.7.2-xxxxxxxxxxx" `shouldBe` "MLdonkey-0"
\ No newline at end of file
+      clientInfo "-ML2.7.2-xxxxxxxxxxx" `shouldBe` "MLdonkey-0"
+
+-- TODO XBT, Bits on Wheels, Queen Bee, BitTyrant, TorrenTopia,
+-- BitSpirit, Rufus, G3 Torrent, FlashGet
\ No newline at end of file
diff --git a/tests/Data/Torrent/MagnetSpec.hs b/tests/Data/Torrent/MagnetSpec.hs
index 5adc6df7..3ecf8705 100644
--- a/tests/Data/Torrent/MagnetSpec.hs
+++ b/tests/Data/Torrent/MagnetSpec.hs
@@ -3,6 +3,7 @@ module Data.Torrent.MagnetSpec (spec) where
 
 import Control.Applicative
 import Data.Maybe
+import Data.Monoid
 import Test.Hspec
 import Test.QuickCheck
 import Test.QuickCheck.Instances ()
@@ -23,7 +24,7 @@ instance Arbitrary URI where
 instance Arbitrary Magnet where
   arbitrary = Magnet <$> arbitrary <*> arbitrary
     <*> arbitrary <*> arbitrary <*> arbitrary <*> arbitrary
-    <*> arbitrary <*> arbitrary <*> pure (error "arbitrary magnet")
+    <*> arbitrary <*> arbitrary <*> pure mempty
 
 magnetEncoding :: Magnet -> Bool
 magnetEncoding m = parseMagnet (renderMagnet m) == Just m
diff --git a/tests/Data/Torrent/MetainfoSpec.hs b/tests/Data/Torrent/MetainfoSpec.hs
index 297b28f1..636bb6b1 100644
--- a/tests/Data/Torrent/MetainfoSpec.hs
+++ b/tests/Data/Torrent/MetainfoSpec.hs
@@ -7,11 +7,13 @@ import Data.ByteString as BS
 import Data.ByteString.Lazy as BL
 import Data.BEncode
 import Data.Maybe
+import Data.Time
 import Network.URI
 import Test.Hspec
 import Test.QuickCheck
 import Test.QuickCheck.Instances ()
 
+import Data.Torrent.Piece
 import Data.Torrent.Layout
 import Data.Torrent
 
@@ -23,8 +25,8 @@ import Data.Torrent
 data T a = T
 
 prop_properBEncode :: Show a => BEncode a => Eq a
-                   => T a -> a -> Bool
-prop_properBEncode _ expected = actual == Right expected
+                   => T a -> a -> IO ()
+prop_properBEncode _ expected = actual `shouldBe` Right expected
   where
     actual = decode $ BL.toStrict $ encode expected
 
@@ -48,13 +50,25 @@ instance Arbitrary LayoutInfo where
     , MultiFile  <$> arbitrary <*> arbitrary
     ]
 
+instance Arbitrary HashArray where
+  arbitrary = HashArray <$> arbitrary
+
+instance Arbitrary PieceInfo where
+  arbitrary = PieceInfo <$> arbitrary <*> arbitrary
+
 instance Arbitrary InfoDict where
-  arbitrary = undefined
+  arbitrary = infoDictionary <$> arbitrary <*> arbitrary <*> arbitrary
+
+pico :: Gen (Maybe NominalDiffTime)
+pico = oneof
+  [ pure Nothing
+  , (Just . fromIntegral) <$> (arbitrary :: Gen Int)
+  ]
 
 instance Arbitrary Torrent where
   arbitrary = Torrent <$> arbitrary
-                 <*> arbitrary <*> arbitrary <*> arbitrary
-                 <*> arbitrary <*> arbitrary <*> arbitrary
+                 <*> arbitrary <*> arbitrary    <*> arbitrary
+                 <*> pico      <*> arbitrary    <*> arbitrary
                  <*> arbitrary <*> pure Nothing <*> arbitrary
 
 {-----------------------------------------------------------------------