1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
{-# LANGUAGE DeriveGeneric #-}
{-# OPTIONS -fno-warn-unused-binds #-}
module Main (main) where
import Control.Applicative
import qualified Data.ByteString as B
import qualified Data.ByteString.Lazy as L
import Test.Framework (defaultMain)
import Test.Framework.Providers.QuickCheck2 (testProperty)
import Test.QuickCheck
import GHC.Generics
import Data.BEncode
instance Arbitrary B.ByteString where
arbitrary = fmap B.pack arbitrary
instance Arbitrary BValue where
arbitrary = frequency
[ (50, BInteger <$> arbitrary)
, (40, BString <$> arbitrary)
, (5, BList <$> (arbitrary `suchThat` ((10 >) . length)))
]
prop_EncDec :: BValue -> Bool
prop_EncDec x = case decode (L.toStrict (encode x)) of
Left _ -> False
Right x' -> x == x'
data List a = Cons a (List a) | Nil
deriving (Show, Eq, Generic)
instance BEncodable a => BEncodable (List a)
instance Arbitrary a => Arbitrary (List a) where
arbitrary = frequency
[ (90, pure Nil)
, (10, Cons <$> arbitrary <*> arbitrary)
]
data FileInfo = FileInfo
{ fiLength :: !Integer
, fiPath :: [B.ByteString]
, fiMD5Sum :: B.ByteString
} deriving (Show, Eq, Generic)
instance BEncodable FileInfo
instance Arbitrary FileInfo where
arbitrary = FileInfo <$> arbitrary <*> arbitrary <*> arbitrary
data T a = T
prop_bencodable :: Eq a => BEncodable a => T a -> a -> Bool
prop_bencodable _ x = decoded (L.toStrict (encoded x)) == Right x
-- All tests are (encode >>> decode = id)
main :: IO ()
main = defaultMain
[ testProperty "BEncode" prop_EncDec
, testProperty "generic recordless" $
prop_bencodable (T :: T (List Int))
, testProperty "generic records" $
prop_bencodable (T :: T FileInfo)
]
|
