eeee054f1e
Now it's possible to change the character encoding while de-/encoding. Also, it's possible to use any data structure as a source or target of the de-/encoding process. darcs-hash:20090221203100-a4fee-6da31f2e37c30a3f5cd5f10af71984209488bb0b
77 lines
2.6 KiB
Haskell
77 lines
2.6 KiB
Haskell
{-# LANGUAGE ExistentialQuantification #-}
|
|
module Test.Tester where
|
|
|
|
import Data.Encoding
|
|
import Test.HUnit
|
|
import Data.Word
|
|
import Data.Char
|
|
import qualified Data.ByteString as BS
|
|
import qualified Data.ByteString.Lazy as LBS
|
|
import Test.QuickCheck hiding (Testable)
|
|
|
|
data EncodingTest
|
|
= forall enc. (Encoding enc,Show enc) =>
|
|
EncodingTest enc String [Word8]
|
|
| forall enc. (Encoding enc,Show enc) =>
|
|
DecodingError enc [Word8] DecodingException
|
|
| forall enc. (Encoding enc,Show enc) =>
|
|
EncodingError enc String EncodingException
|
|
|
|
instance Testable EncodingTest where
|
|
test (EncodingTest enc src trg)
|
|
= TestList
|
|
[TestLabel (show enc ++ " encoding")
|
|
(TestCase $ encodeStrictByteStringExplicit enc src
|
|
@?= Right (BS.pack trg))
|
|
,TestLabel (show enc ++ " decoding")
|
|
(TestCase $ decodeStrictByteStringExplicit enc (BS.pack trg)
|
|
@=? Right src)
|
|
]
|
|
test (DecodingError enc src ex)
|
|
= TestLabel (show enc ++ " decoding error")
|
|
(TestCase $ decodeStrictByteStringExplicit enc (BS.pack src) @=? Left ex)
|
|
|
|
|
|
charGen :: Gen Char
|
|
charGen = let
|
|
ascii = choose (0x00,0x7F) >>= return.chr
|
|
oneByte = choose (0x80,0xFF) >>= return.chr
|
|
twoByte = choose (0x0100,0xFFFF) >>= return.chr
|
|
threeByte = choose (0x010000,0x10FFFF) >>= return.chr
|
|
in frequency [(40,ascii),(30,oneByte),(20,twoByte),(10,threeByte)]
|
|
|
|
instance Arbitrary Char where
|
|
arbitrary = charGen
|
|
coarbitrary x = id
|
|
|
|
instance Arbitrary Word8 where
|
|
arbitrary = choose (0x00,0xFF::Int) >>= return.fromIntegral
|
|
coarbitrary x = id
|
|
|
|
quickCheckEncoding :: Encoding enc => enc -> IO ()
|
|
quickCheckEncoding e = do
|
|
quickCheck (encodingIdentity e)
|
|
quickCheck (decodingIdentity e)
|
|
|
|
encodingIdentity :: Encoding enc => enc -> String -> Property
|
|
encodingIdentity e str
|
|
= trivial (null str)
|
|
$ case encoded of
|
|
Left err -> trivial True True
|
|
Right res -> case decodeStrictByteStringExplicit e res of
|
|
Left err -> property False
|
|
Right res' -> property (str==res')
|
|
where
|
|
encoded = encodeStrictByteStringExplicit e str
|
|
|
|
decodingIdentity :: Encoding enc => enc -> [Word8] -> Property
|
|
decodingIdentity e wrd
|
|
= trivial (null wrd)
|
|
$ case decoded of
|
|
Left err -> trivial True True
|
|
Right res -> case encodeStrictByteStringExplicit e res of
|
|
Left err -> property False
|
|
Right res' -> property (bstr==res')
|
|
where
|
|
bstr = BS.pack wrd
|
|
decoded = decodeStrictByteStringExplicit e bstr |