@@ -1,339 +1,116 @@
1−{-# LANGUAGE CPP, FlexibleInstances, FlexibleContexts, TupleSections #-}
1+module Main where
3−module Main ( main ) where
5−import Data.Binary ( Binary(..) )
3+import Data.Binary ( encode )
6−import Data.Binary.Get ( runGet, runGetIncremental, pushChunks, Decoder(..) )
4+import Data.Binary.Get ( runGet )
7import Data.Binary.Put ( runPut )
5import Data.Binary.Put ( runPut )
9−import Data.Binary.Bits
7+import Bits
10−import Data.Binary.Bits.Get
11−import Data.Binary.Bits.Put
12−import Data.Binary.Bits.BitOrder
14−import qualified Data.ByteString as B
15−import qualified Data.ByteString.Lazy as L
17−import Control.Applicative
18import Data.Bits
9import Data.Bits
19import Data.Word
10import Data.Word
20import Foreign.Storable
11import Foreign.Storable
21−import Data.Traversable (traverse)
12+import System.Random
22−import Data.Foldable (traverse_)
24−import Test.Framework.Providers.QuickCheck2 ( testProperty )
25−import Test.Framework.Runners.Console ( defaultMain )
26−import Test.Framework ( Test, testGroup )
27−import Test.QuickCheck
30−main = defaultMain tests
34− [ testGroup "Internal test functions"
35− [ testProperty "prop_bitreq" prop_bitreq ]
37− , testGroup "Custom test cases"
38− [ testProperty "prop_composite_case" prop_composite_case ]
40− , testGroup "getByteString"
41− [ testProperty "prop_getByteString_negative" prop_getByteString_negative
42− , testProperty "prop_putByteString_getByteString" (prop_putByteString_getByteString :: BitOrder -> B.ByteString -> Property)
43− , testProperty "prop_putByteString_getByteString_many" (prop_putByteString_getByteString_many :: BitOrder -> [B.ByteString] -> Property)
46− , testGroup "getLazyByteString"
47− [ testProperty "getLazyByteString == getByteString"
48− prop_getLazyByteString_equal_to_ByteString
49− , testProperty "getLazyByteString == getByteString (with shift)"
50− prop_getLazyByteString_equal_to_ByteString2
53− , testGroup "isEmpty"
54− [ testProperty "prop_isEmptyOfEmptyEmpty" prop_isEmptyOfEmptyEmpty
55− , testProperty "prop_isEmptyOfNonEmptyEmpty" prop_isEmptyOfNonEmptyEmpty
56− , testProperty "prop_isEmptyOfConsumedEmpty" prop_isEmptyOfConsumedEmpty
57− , testProperty "prop_isEmptyOfNotConsumedNotEmpty" prop_isEmptyOfNotConsumedNotEmpty
61− [ testProperty "monadic fail" prop_fail ]
63− , testGroup "prop_bitput_with_get_from_binary"
64− [ testProperty "Word8" (prop_bitput_with_get_from_binary :: W [Word8] -> Property)
65− , testProperty "Word16" (prop_bitput_with_get_from_binary :: W [Word16] -> Property)
66− , testProperty "Word32" (prop_bitput_with_get_from_binary :: W [Word32] -> Property)
67− , testProperty "Word64" (prop_bitput_with_get_from_binary :: W [Word64] -> Property)
70− , testGroup "prop_bitget_with_put_from_binary"
71− [ testProperty "Word8" (prop_bitget_with_put_from_binary :: W [Word8] -> Property)
72− , testProperty "Word16" (prop_bitget_with_put_from_binary :: W [Word16] -> Property)
73− , testProperty "Word32" (prop_bitget_with_put_from_binary :: W [Word32] -> Property)
74− , testProperty "Word64" (prop_bitget_with_put_from_binary :: W [Word64] -> Property)
77− , testGroup "prop_compare_put_with_naive"
78− [ testProperty "Word8" (prop_compare_put_with_naive :: W [Word8] -> Property)
79− , testProperty "Word16" (prop_compare_put_with_naive :: W [Word16] -> Property)
80− , testProperty "Word32" (prop_compare_put_with_naive :: W [Word32] -> Property)
81− , testProperty "Word64" (prop_compare_put_with_naive :: W [Word64] -> Property)
84− , testGroup "prop_compare_get_with_naive"
85− [ testProperty "Word8" (prop_compare_get_with_naive:: W [Word8] -> Property)
86− , testProperty "Word16" (prop_compare_get_with_naive:: W [Word16] -> Property)
87− , testProperty "Word32" (prop_compare_get_with_naive:: W [Word32] -> Property)
88− , testProperty "Word64" (prop_compare_get_with_naive:: W [Word64] -> Property)
91− , testGroup "prop_put_with_bitreq"
92− [ testProperty "Word8" (prop_putget_with_bitreq :: BitOrder -> W Word8 -> Property)
93− , testProperty "Word16" (prop_putget_with_bitreq :: BitOrder -> W Word16 -> Property)
94− , testProperty "Word32" (prop_putget_with_bitreq :: BitOrder -> W Word32 -> Property)
95− , testProperty "Word64" (prop_putget_with_bitreq :: BitOrder -> W Word64 -> Property)
98− , testGroup "prop_putget_list_simple"
99− [ testProperty "Bool" (prop_putget_list_simple :: W [Bool] -> Property)
100− , testProperty "Word8" (prop_putget_list_simple :: W [Word8] -> Property)
101− , testProperty "Word16" (prop_putget_list_simple :: W [Word16] -> Property)
102− , testProperty "Word32" (prop_putget_list_simple :: W [Word32] -> Property)
103− , testProperty "Word64" (prop_putget_list_simple :: W [Word64] -> Property)
106− , testGroup "prop_putget_applicative_with_bitreq"
14+import BitsGet
107− [ testProperty "Word8" (prop_putget_applicative_with_bitreq :: W [(Word8,Word8,Word8)] -> Property)
15+import BitsPut
108− , testProperty "Word16" (prop_putget_applicative_with_bitreq :: W [(Word16,Word16,Word16)] -> Property)
109− , testProperty "Word32" (prop_putget_applicative_with_bitreq :: W [(Word32,Word32,Word32)] -> Property)
110− , testProperty "Word64" (prop_putget_applicative_with_bitreq :: W [(Word64,Word64,Word64)] -> Property)
113− , testGroup "prop_putget_list_with_bitreq"
17+import Test.QuickCheck
114− [ testProperty "Word8" (prop_putget_list_with_bitreq :: W [Word8] -> Property)
115− , testProperty "Word16" (prop_putget_list_with_bitreq :: W [Word16] -> Property)
116− , testProperty "Word32" (prop_putget_list_with_bitreq :: W [Word32] -> Property)
117− , testProperty "Word64" (prop_putget_list_with_bitreq :: W [Word64] -> Property)
119− , testGroup "prop_bitget_bytestring_interspersed"
120− [ testProperty "Word8" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word8 -> [B.ByteString] -> Property)
121− , testProperty "Word16" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word16 -> [B.ByteString] -> Property)
122− , testProperty "Word32" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word32 -> [B.ByteString] -> Property)
123− , testProperty "Word64" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word64 -> [B.ByteString] -> Property)
125− , testGroup "Simulate programs"
126− [ testProperty "primitive" prop_primitive
127− , testProperty "many primitives in sequence" prop_program
131−prop_isEmptyOfEmptyEmpty :: Bool
132−prop_isEmptyOfEmptyEmpty = runGet (runBitGet isEmpty) L.empty
134−prop_isEmptyOfNonEmptyEmpty :: L.ByteString -> Property
135−prop_isEmptyOfNonEmptyEmpty bs =
136− not (L.null bs) ==> not (runGet (runBitGet isEmpty) bs)
138−prop_isEmptyOfConsumedEmpty :: L.ByteString -> Property
139−prop_isEmptyOfConsumedEmpty bs =
141− runGet (runBitGet (getByteString n >> isEmpty)) bs
142− where n = fromIntegral $ L.length bs
144−prop_isEmptyOfNotConsumedNotEmpty :: L.ByteString -> Int -> Property
145−prop_isEmptyOfNotConsumedNotEmpty bs n =
146− (fromIntegral n) < L.length bs && not (L.null bs) ==>
147− not (runGet (runBitGet (getByteString n >> isEmpty)) bs)
149−prop_getLazyByteString_equal_to_ByteString :: L.ByteString -> Int -> Property
150−prop_getLazyByteString_equal_to_ByteString bs n =
151− (fromIntegral n) <= L.length bs ==>
152− runGet (runBitGet (getLazyByteString (fromIntegral n))) bs ==
153− (L.fromChunks . (:[]) $ runGet (runBitGet (getByteString n)) bs)
155−prop_getLazyByteString_equal_to_ByteString2 :: L.ByteString -> Int -> Property
156−prop_getLazyByteString_equal_to_ByteString2 bs n =
157− (L.length bs > 1) && (fromIntegral n) < L.length bs ==>
158− runGet (runBitGet (getWord8 2 >> getLazyByteString (fromIntegral n))) bs ==
159− (L.fromChunks . (:[]) $ runGet (runBitGet (getWord8 2 >> getByteString n)) bs)
161−prop_getByteString_negative :: Int -> Property
19+main = do
162−prop_getByteString_negative n =
20+ quickCheck prop_bitreq
163− n < 1 ==>
21+ quickCheck prop_composite_case
164− runGet (runBitGet (getByteString n)) L.empty == B.empty
166−prop_putByteString_getByteString :: BitOrder -> B.ByteString -> Property
23+ quickCheck (prop_putget_with_bitreq :: Word8 -> Property)
167−prop_putByteString_getByteString bo bs = property $ bs' == bs
24+ quickCheck (prop_putget_with_bitreq :: Word16 -> Property)
170− w = runPut (runBitPut (withBitOrder bo (putByteString bs)))
171− bs' = runGet (runBitGet (withBitOrder bo (getByteString n))) w
173−prop_putByteString_getByteString_many :: BitOrder -> [B.ByteString] -> Property
26+ quickCheck (prop_putget_list_simple :: [Bool] -> Property)
174−prop_putByteString_getByteString_many bo bs = property $ bs' == bs
27+ quickCheck (prop_putget_list_simple :: [Word8] -> Property)
175− where
28+ quickCheck (prop_putget_list_simple :: [Word16] -> Property)
176− n = fmap B.length bs
177− w = runPut (runBitPut (withBitOrder bo (traverse_ putByteString bs)))
178− bs' = runGet (runBitGet (withBitOrder bo (traverse getByteString n))) w
30+ quickCheck (prop_putget_list_with_bitreq :: [Word8] -> Property)
31+ quickCheck (prop_putget_list_with_bitreq :: [Word16] -> Property)
33+ -- these tests use the R structure
35+ -- quickCheck prop_Word32_from_2_Word16
36+ -- quickCheck prop_Word32_from_Word8_and_Word16
181−prop_putget_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => BitOrder -> W a -> Property
38+prop_putget_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => a -> Property
182−prop_putget_with_bitreq bo (W w) = property $
39+prop_putget_with_bitreq w = property $
183 -- write all words with as many bits as it's required
40 -- write all words with as many bits as it's required
184 let p = putBits (bitreq w) w
41 let p = putBits (bitreq w) w
185 g = getBits (bitreq w)
42 g = getBits (bitreq w)
186− lbs = runPut (runBitPut (withBitOrder bo p))
43+ lbs = runPut (runBitPut p)
187− w' = runGet (runBitGet (withBitOrder bo g)) lbs
44+ w' = runGet (runBitGetSimple g) lbs
188 in w == w'
45 in w == w'
190−-- | Write a list of items. Each item is written with the maximum amount of
191−-- bits, i.e. 8 for Word8, 16 for Word16, etc.
192−prop_putget_list_simple :: (BinaryBit a, Eq a, Storable a) => W [a] -> Property
47+prop_putget_list_simple :: (BinaryBit a, Eq a, Storable a) => [a] -> Property
193−prop_putget_list_simple (W ws) = property $
48+prop_putget_list_simple ws = property $
194 let s = sizeOf (head ws) * 8
49 let s = sizeOf (head ws) * 8
195− p = mapM_ (putBits s) ws
50+ p = mapM_ (\v -> putBits s v) ws
196 g = mapM (const (getBits s)) ws
51 g = mapM (const (getBits s)) ws
197 lbs = runPut (runBitPut p)
52 lbs = runPut (runBitPut p)
198− ws' = runGet (runBitGet g) lbs
53+ ws' = runGet (runBitGetSimple g) lbs
199 in ws == ws'
54 in ws == ws'
201−-- | Write a list of items. Each item is written with exactly as many bits
202−-- as required. Then read it back.
203−prop_putget_list_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => W [a] -> Property
56+prop_putget_list_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => [a] -> Property
204−prop_putget_list_with_bitreq (W ws) = property $
57+prop_putget_list_with_bitreq ws = property $
205 -- write all words with as many bits as it's required
58 -- write all words with as many bits as it's required
206 let p = mapM_ (\v -> putBits (bitreq v) v) ws
59 let p = mapM_ (\v -> putBits (bitreq v) v) ws
207 g = mapM getBits bitlist
60 g = mapM getBits bitlist
208 lbs = runPut (runBitPut p)
61 lbs = runPut (runBitPut p)
209− ws' = runGet (runBitGet g) lbs
62+ ws' = runGet (runBitGetSimple g) lbs
210 in ws == ws'
63 in ws == ws'
212 bitlist = map bitreq ws
65 bitlist = map bitreq ws
214−prop_putget_applicative_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => W [(a,a,a)] -> Property
215−prop_putget_applicative_with_bitreq (W ts) = property $
216− let p = mapM_ (\(a,b,c) -> do putBits (bitreq a) a
217− putBits (bitreq b) b
218− putBits (bitreq c) c) ts
219− g = mapM (\(a,b,c) -> (,,) <$> getBits a <*> getBits b <*> getBits c) bitlist
220− lbs = runPut (runBitPut p)
221− ts' = runGet (runBitGet g) lbs
224− bitlist = map (\(a,b,c) -> (bitreq a, bitreq b, bitreq c)) ts
226−-- | Write bits using this library, and read them back using the binary
228−prop_bitput_with_get_from_binary :: (BinaryBit a, Binary a, Storable a, Eq a) => W [a] -> Property
229−prop_bitput_with_get_from_binary (W ws) = property $
230− let s = sizeOf (head ws) * 8
231− p = mapM_ (putBits s) ws
232− g = mapM (const get) ws
233− lbs = runPut (runBitPut p)
237−-- | Write bits using the binary library, and read them back using this
239−prop_bitget_with_put_from_binary :: (BinaryBit a, Binary a, Storable a, Eq a) => W [a] -> Property
240−prop_bitget_with_put_from_binary (W ws) = property $
241− let s = sizeOf (head ws) * 8
243− g = mapM (const (getBits s)) ws
245− ws' = runGet (runBitGet g) lbs
248−-- | Write each 'ByteString' with a variable sized value as a separator.
249−prop_bitget_bytestring_interspersed :: (BinaryBit a, Binary a, Num a, Ord a, Bits a) => BitOrder -> W a -> [B.ByteString] -> Property
250−prop_bitget_bytestring_interspersed bo (W ws) bss = property $
251− let p = mapM_ (\bs -> putBits (bitreq ws) ws >> putByteString bs) bss
252− g = mapM (\bs -> (,) <$> getBits (bitreq ws) <*> getByteString (B.length bs)) bss
253− lbs = runPut (runBitPut (withBitOrder bo p))
254− r = runGet (runBitGet (withBitOrder bo g)) lbs
255− in map (ws,) bss == r
258−prop_fail :: L.ByteString -> String -> Property
259−prop_fail lbs errMsg0 = forAll (choose (0, 8 * L.length lbs)) $ \len ->
260− let (bytes,bits) = len `divMod` 8
261− expectedBytesConsumed
263− | otherwise = bytes + 1
264− p = do _ <- getByteString (fromIntegral bytes)
265− _ <- getBits (fromIntegral bits) :: BitGet Word8
267− r = runGetIncremental (runBitGet p) `pushChunks` lbs
269− Fail remainingBS pos errMsg ->
270− and [ L.fromChunks [remainingBS] == L.drop expectedBytesConsumed lbs
271− , pos == expectedBytesConsumed
276−-- | number of bits required to write @v@
67+-- number of bits required to write 'v'
277−bitreq :: (Num b, Num a, Bits a, Ord a) => a -> b
68+bitreq :: (Num b, Bits a, Ord a) => a -> b
278bitreq v = fromIntegral . head $ [ req | (req, top) <- bittable, v <= top ]
69bitreq v = fromIntegral . head $ [ req | (req, top) <- bittable, v <= top ]
280−bittable :: (Bits a, Num a) => [(Integer, a)]
71+bittable :: Bits a => [(Integer, a)]
281bittable = [ (fromIntegral x, (1 `shiftL` x) - 1) | x <- [1..64] ]
72bittable = [ (fromIntegral x, (1 `shiftL` x) - 1) | x <- [1..64] ]
283−prop_bitreq :: W Word64 -> Property
74+prop_bitreq :: Word64 -> Property
284−prop_bitreq (W w) = property $
75+prop_bitreq w = property $
285 ( w == 0 && bitreq w == 1 )
76 ( w == 0 && bitreq w == 1 )
286− || bitreq w == bitreq (w `shiftR` 1) + 1
77+ || bitreq w == (bitreq (w `shiftR` 1)) + 1
288−prop_composite_case :: Bool -> W Word16 -> Property
79+prop_composite_case :: Bool -> Word16 -> Property
289−prop_composite_case b (W w) = w < 0x8000 ==>
80+prop_composite_case b w = w < 0x8000 ==>
290 let p = do putBool b
81 let p = do putBool b
291− putWord16 15 w
82+ putWord16be 15 w
292 g = do v <- getBool
83 g = do v <- getBool
293 case v of
84 case v of
294− True -> getWord16 15
85+ True -> getWord16be 15
295 False -> do
86 False -> do
296 msb <- getWord8 7
87 msb <- getWord8 7
297 lsb <- getWord8 8
88 lsb <- getWord8 8
298 return ((fromIntegral msb `shiftL` 8) .|. fromIntegral lsb)
89 return ((fromIntegral msb `shiftL` 8) .|. fromIntegral lsb)
299 lbs = runPut (runBitPut p)
90 lbs = runPut (runBitPut p)
300− w' = runGet (runBitGet g) lbs
91+ w' = runGet (runBitGetSimple g) lbs
301 in w == w'
92 in w == w'
303−prop_compare_put_with_naive :: (Bits a, BinaryBit a, Ord a, Num a) => W [a] -> Property
94+{-
304−prop_compare_put_with_naive (W ws) = property $
95+prop_Word32_from_Word8_and_Word16 :: Word8 -> Word16 -> Property
305− let pn = mapM_ (\v -> naive_put (bitreq v) v) ws
96+prop_Word32_from_Word8_and_Word16 w8 w16 = property $
306− p = mapM_ (\v -> putBits (bitreq v) v) ws
97+ let p = RWord32be 24
307− lbs_n = runPut (runBitPut pn)
98+ w' = runGet (get p) lbs
308− lbs = runPut (runBitPut p)
99+ in w0 == w'
309− in lbs_n == lbs
100+ where
310−
101+ lbs = runPut (putWord8 w8 >> putWord16be w16)
311−prop_compare_get_with_naive :: (Bits a, BinaryBit a, Ord a, Num a) => W [a] -> Property
102+ w0 = ((fromIntegral w8) `shiftL` 16) .|. fromIntegral w16
312−prop_compare_get_with_naive (W ws) = property $
103+
313− let gn = mapM (\v -> naive_get (bitreq v)) ws
104+prop_Word32_from_2_Word16 :: Word16 -> Word16 -> Property
314− g = mapM (\v -> getBits (bitreq v)) ws
105+prop_Word32_from_2_Word16 w1 w2 = property $
315− p = mapM_ (\v -> naive_put (bitreq v) v) ws
106+ let p = RWord32be 32
316− lbs = runPut (runBitPut p)
317− rn = runGet (runBitGet gn) lbs
107+ w' = runGet (get p) lbs
318− r = runGet (runBitGet g ) lbs
108+ in w0 == w'
319− -- we must help our compiler to resolve the types of 'gn' and 'g'
109+ where
320− _types = rn == ws && r == ws
110+ lbs = encode w0
321− in rn == r
111+ w0 = ((fromIntegral w1) `shiftL` 16) .|. fromIntegral w2
323−-- | Write one bit at a time until the full word has been written
324−naive_put :: (Bits a) => Int -> a -> BitPut ()
325−naive_put n w = mapM_ (\b -> putBool (testBit w b)) [n-1,n-2..0]
327−-- | Read one bit at a time until we've reconstructed the whole word
328−naive_get :: (Bits a, Num a) => Int -> BitGet a
330− let loop 0 acc = return acc
334− False -> loop (n-1) (acc `shiftL` 1)
335− True -> loop (n-1) ((acc `shiftL` 1) + 1)
338shrinker :: (Num a, Ord a, Bits a) => a -> [a]
115shrinker :: (Num a, Ord a, Bits a) => a -> [a]
339shrinker 0 = []
116shrinker 0 = []
@@ -346,187 +123,27 @@ shrinker w = [ w `shiftR` 1 -- try to make everything roughly half size
347 n = bitreq w
124 n = bitreq w
349−data W a = W { unW :: a } deriving (Show, Eq, Ord)
351−arbitraryW :: (Arbitrary (W a)) => Gen a
352−arbitraryW = unW <$> arbitrary
354−shrinkW :: (Arbitrary (W a)) => a -> [a]
355−shrinkW x = unW <$> shrink (W x)
357−instance Arbitrary (W Bool) where
126+instance Arbitrary Word8 where
358− arbitrary = W <$> arbitrary
359− shrink = map W <$> shrink . unW
361−instance Arbitrary (W Word8) where
362− arbitrary = W <$> choose (minBound, maxBound)
363− shrink = map W . shrinker . unW
365−instance Arbitrary (W Word16) where
366− arbitrary = W <$> choose (minBound, maxBound)
127+ arbitrary = choose (minBound, maxBound)
367− shrink = map W . shrinker . unW
128+ shrink = shrinker
369−instance Arbitrary (W Word32) where
370− arbitrary = W <$> choose (minBound, maxBound)
371− shrink = map W . shrinker . unW
373−instance Arbitrary (W Word64) where
374− arbitrary = W <$> choose (minBound, maxBound)
375− shrink = map W . shrinker . unW
377−instance Arbitrary B.ByteString where
130+instance Arbitrary Word16 where
378− arbitrary = B.pack <$> arbitrary
131+ arbitrary = choose (minBound, maxBound)
379− shrink bs = B.pack <$> shrink (B.unpack bs)
132+ shrink = shrinker
381−instance Arbitrary L.ByteString where
134+instance Arbitrary Word32 where
382− arbitrary = L.fromChunks <$> arbitrary
135+ arbitrary = choose (minBound, maxBound)
383− shrink bs = L.fromChunks <$> shrink (L.toChunks bs)
136+ shrink = shrinker
385−instance (Arbitrary (W a)) => Arbitrary (W [a]) where
138+instance Arbitrary Word64 where
386− arbitrary = W . map unW <$> arbitrary
139+ arbitrary = choose (minBound, maxBound)
387− shrink = map (W . map unW) <$> mapM shrink . map W . unW
140+ shrink = shrinker
389−instance (Arbitrary (W a), Arbitrary (W b)) => Arbitrary (W (a,b)) where
142+integralRandomR :: (Integral a, RandomGen g) => (a,a) -> g -> (a,g)
390− arbitrary = (W .) . (,) <$> arbitraryW <*> arbitraryW
143+integralRandomR (a,b) g = case randomR (fromIntegral a :: Integer,
391− shrink (W (a,b)) = (W .) . (,) <$> shrinkW a <*> shrinkW b
144+ fromIntegral b :: Integer) g of
145+ (x,g) -> (fromIntegral x, g)
393−instance (Arbitrary (W a), Arbitrary (W b), Arbitrary (W c)) => Arbitrary (W (a,b,c)) where
394− arbitrary = ((W .) .) . (,,) <$> arbitraryW <*> arbitraryW <*> arbitraryW
395− shrink (W (a,b,c)) = ((W .) .) . (,,) <$> shrinkW a <*> shrinkW b <*> shrinkW c
403− | BS Int B.ByteString
404− | LBS Int L.ByteString
409−type Program = [Primitive]
411−instance Arbitrary Primitive where
414− let (maxBits, _) = (\w -> (finiteBitSize w, c undefined w)) undefined
415− bits <- choose (0, maxBits)
416− n <- choose (0, fromIntegral (2^bits-1))
419− [ Bool <$> arbitrary
424− , Skip <$> choose (0, 3000)
425− , do n <- choose (0,10)
427− return (BS n (B.pack cs))
428− , do n <- choose (0,10)
430− return (LBS n (L.pack cs))
434− let snk c x = map (\x' -> c (bitreq x') x') (shrinker x) in
436− Bool b -> if b then [Bool False] else []
438− W16 _ x -> snk W16 x
439− W32 _ x -> snk W32 x
440− W64 _ x -> snk W64 x
441− Skip x -> Skip <$> shrink x
442− BS _ bs -> let ws = B.unpack bs in map (\ws' -> BS (length ws') (B.pack ws')) (shrink ws)
443− LBS _ lbs -> let ws = L.unpack lbs in map (\ws' -> LBS (length ws') (L.pack ws')) (shrink ws)
446−instance Arbitrary BitOrder where
447− arbitrary = elements [BB, LB, LL, BL]
448− shrink LL = [BB,LB,BL]
454−prop_primitive :: Primitive -> Property
455−prop_primitive prim = property $
456− let p = putPrimitive prim
457− g = getPrimitive prim
458− lbs = runPut (runBitPut p)
459− r = runGet (runBitGet g) lbs
462−prop_program :: Program -> Property
463−prop_program program = property $
464− let p = mapM_ putPrimitive program
465− g = verifyProgram (8 * fromIntegral (L.length lbs)) program
466− lbs = runPut (runBitPut p)
467− r = runGet (runBitGet g) lbs
470−putPrimitive :: Primitive -> BitPut ()
474− W8 n x -> putWord8 n x
475− W16 n x -> putWord16 n x
476− W32 n x -> putWord32 n x
477− W64 n x -> putWord64 n x
478− Skip n -> skipBits n
479− BS _ bs -> putByteString bs
480− LBS _ lbs -> mapM_ putByteString (L.toChunks lbs)
481− IsEmpty -> return ()
483−getPrimitive :: Primitive -> BitGet Primitive
486− Bool _ -> Bool <$> getBool
487− W8 n _ -> W8 n <$> getWord8 n
488− W16 n _ -> W16 n <$> getWord16 n
489− W32 n _ -> W32 n <$> getWord32 n
490− W64 n _ -> W64 n <$> getWord64 n
491− Skip n -> skipBits n >> return (Skip n)
492− BS n _ -> BS n <$> getByteString n
493− LBS n _ -> LBS n <$> getLazyByteString n
494− IsEmpty -> isEmpty >> return IsEmpty
496−getPrimitiveSize :: Primitive -> Int
497−getPrimitiveSize p = case p of
508−verifyProgram :: Int -> Program -> BitGet Bool
509−verifyProgram totalLength ps0 = go 0 ps0
511− go _ [] = return True
515− let expected = pos == totalLength
517− if expected == actual
519− else error $ "isEmpty returned wrong value, expected "
520− ++ show expected ++ " but got " ++ show actual
521− _ -> check p (getPrimitive p) >> go (pos + getPrimitiveSize p) ps
526− else error $ "Roundtrip error: Expected "
527− ++ show x ++ " but got " ++ show y
530instance Random Word where
147instance Random Word where
531 randomR = integralRandomR
148 randomR = integralRandomR
532 random = randomR (minBound,maxBound)
149 random = randomR (minBound,maxBound)
@@ -546,4 +163,3 @@ instance Random Word32 where
546instance Random Word64 where
163instance Random Word64 where
547 randomR = integralRandomR
164 randomR = integralRandomR
548 random = randomR (minBound,maxBound)
165 random = randomR (minBound,maxBound)