Fix test

1 files changed, 19 insertions, 5 deletions

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
 
 {-----------------------------------------------------------------------