3 files changed, 378 insertions, 0 deletions

diff --git a/test-src/DataencQC.hs b/test-src/DataencQC.hs
new file mode 100644
index 0000000..d35c09c
--- /dev/null
+++ b/test-src/DataencQC.hs
@@ -0,0 +1,137 @@
+{-# OPTIONS_GHC -XTemplateHaskell #-}
+
+{-
+ - Copyright : (c) 2007 Magnus Therning
+ - License   : BSD3
+ -}
+
+module DataencQC
+    where
+
+import Test.Framework.TH
+
+import Data.Maybe
+import Data.Word
+import Test.QuickCheck
+import Test.Framework.Providers.QuickCheck2
+
+import qualified Codec.Binary.Uu as Uu
+import qualified Codec.Binary.Uu as Xx
+import qualified Codec.Binary.Base85 as Base85
+import qualified Codec.Binary.Base64 as Base64
+import qualified Codec.Binary.Base64Url as Base64Url
+import qualified Codec.Binary.Base32 as Base32
+import qualified Codec.Binary.Base32Hex as Base32Hex
+import qualified Codec.Binary.Base16 as Base16
+import qualified Codec.Binary.Yenc as Yenc
+import qualified Codec.Binary.QuotedPrintable as QP
+import qualified Codec.Binary.PythonString as Py
+import qualified Codec.Binary.Url as Url
+
+-- {{{1 uuencode properties
+prop_uuEncode ws = ws == (fromJust . Uu.decode . Uu.encode) ws
+    where types = ws::[Word8]
+
+prop_uuChop n ws = s == (Uu.unchop . Uu.chop n) s
+    where
+        types = (n :: Int, ws::[Word8])
+        s = Uu.encode ws
+
+prop_uuCombined n ws = ws == fromJust (Uu.decode $ Uu.unchop $ Uu.chop n $ Uu.encode ws)
+    where types = (n::Int, ws::[Word8])
+
+-- {{{1 xxencode properties
+prop_xxEncode ws = ws == (fromJust . Xx.decode . Xx.encode) ws
+    where types = ws::[Word8]
+
+prop_xxChop n ws = s == (Xx.unchop . Xx.chop n) s
+    where
+        types = (n:: Int, ws::[Word8])
+        s = Xx.encode ws
+
+prop_xxCombined n ws = ws == fromJust (Xx.decode $ Xx.unchop $ Xx.chop n $ Xx.encode ws)
+    where types = (n::Int, ws::[Word8])
+
+-- {{{1 base85 properties
+prop_base85Encode ws = ws == fromJust (Base85.decode $ Base85.encode ws)
+    where types = ws::[Word8]
+
+prop_base85Chop n s = s == Base85.unchop (Base85.chop n s)
+    where types = (n::Int, s::String)
+
+-- {{{1 base64 properties
+prop_base64Encode ws = ws == fromJust (Base64.decode $ Base64.encode ws)
+    where types = ws::[Word8]
+
+prop_base64Chop n s = s == Base64.unchop (Base64.chop n s)
+    where types = (n::Int, s::String)
+
+-- {{{1 base64url properties
+prop_base64UrlEncode ws = ws == fromJust (Base64Url.decode $ Base64Url.encode ws)
+    where types = ws::[Word8]
+
+prop_base64UrlChop n s = s == Base64Url.unchop (Base64Url.chop n s)
+    where types = (n::Int, s::String)
+
+-- {{{1 base32
+prop_base32Encode ws = ws == fromJust (Base32.decode $ Base32.encode ws)
+    where types = ws::[Word8]
+
+prop_base32Chop n s = s == Base32.unchop (Base32.chop n s)
+    where types = (n::Int, s::String)
+
+-- {{{1 base32hex
+prop_base32HexEncode ws = ws == fromJust (Base32Hex.decode $ Base32Hex.encode ws)
+    where types = ws::[Word8]
+
+prop_base32HexChop n s = s == Base32Hex.unchop (Base32Hex.chop n s)
+    where types = (n::Int, s::String)
+
+-- {{{1 base16
+prop_base16Encode ws = ws == fromJust (Base16.decode $ Base16.encode ws)
+    where types = ws::[Word8]
+
+prop_base16Chop n s = s == Base16.unchop (Base16.chop n s)
+    where types = (n::Int, s::String)
+
+-- {{{1 yEncoding
+prop_yencEncode ws = ws == fromJust (Yenc.decode $ Yenc.encode ws)
+    where types = ws ::[Word8]
+
+prop_yencChop n ws = ws == Yenc.unchop (Yenc.chop n ws)
+    where types = (n::Int, ws :: [Word8])
+
+-- {{{1 qp
+prop_qpEncode ws = ws == fromJust (QP.decode $ QP.encode ws)
+    where types = ws :: [Word8]
+
+prop_qpChop n ws = s == (QP.unchop . QP.chop n) s
+    where
+        types = (n::Int, ws::[Word8])
+        s = QP.encode ws
+
+prop_qpCombined n ws = ws == fromJust (QP.decode $ QP.unchop $ QP.chop n $ QP.encode ws)
+    where types = (n::Int, ws::[Word8])
+
+-- {{{1 py
+prop_pyEncode ws = ws == fromJust (Py.decode $ Py.encode ws)
+    where types = ws :: [Word8]
+
+prop_pyChop n s = s == Py.unchop (Py.chop n s)
+    where types = (n :: Int, s :: String)
+
+prop_pyCombined n ws = ws == fromJust (runAll ws)
+    where runAll = Py.decode . Py.unchop . Py.chop n . Py.encode
+
+-- {{{1 url
+prop_urlEncode ws = ws == fromJust (Url.decode $ Url.encode ws)
+    where types = ws :: [Word8]
+
+prop_urlChop n s = s == Url.unchop (Url.chop n s)
+    where types = (n :: Int, s :: String)
+
+prop_urlCombined n ws = ws == fromJust (runAll ws)
+    where runAll = Url.decode . Url.unchop . Url.chop n . Url.encode
+
+-- {{{1 all the tests
+allTests = $(testGroupGenerator)
diff --git a/test-src/DataencUT.hs b/test-src/DataencUT.hs
new file mode 100644
index 0000000..af91a0f
--- /dev/null
+++ b/test-src/DataencUT.hs
@@ -0,0 +1,225 @@
+{-# OPTIONS_GHC -XTemplateHaskell #-}
+{-
+ - Copyright : (c) 2007 Magnus Therning
+ - License   : BSD3
+ -}
+
+module DataencUT
+    where
+
+import Test.HUnit
+import Control.Monad
+import System.Exit
+import Data.Maybe
+import qualified Test.Framework.Providers.API as TFAPI
+import Test.Framework.TH
+import Test.Framework.Providers.HUnit
+
+import Codec.Binary.DataEncoding
+import qualified Codec.Binary.Yenc as Yenc
+
+-- {{{1 checkAssertions
+checkAssertions (suite, desc, enc, dec, codec) = do
+    enc @=? encode codec dec
+    dec @=? fromJust (decode codec enc)
+
+-- {{{1 uuencode tests
+uuTestData =
+    [ ("uu", "empty", "", [], uu)
+    , ("uu", "\\0", "``", [0], uu)
+    , ("uu", "\\255", "_P", [255], uu)
+    , ("uu", "AA", "04$", [65, 65], uu)
+    , ("uu", "AAA", "04%!", [65, 65, 65], uu)
+    , ("uu", "AAAA", "04%!00", [65, 65, 65, 65], uu)
+    , ("uu", "Example", "17AA;7!L90", [69,120,97,109,112,108,101], uu)
+    ]
+case_uuTests = mapM_ checkAssertions uuTestData
+
+case_uuTests2 = do
+    "EI2" @=? unchop uu (chop uu 1 "EI2")
+    "EI3-" @=? unchop uu (chop uu 1 "EI3-")
+    "EI3-EE" @=? unchop uu (chop uu 1 "EI3-EE")
+    [0..255] @=? fromJust (decode uu $ unchop uu $ chop uu 1 $ encode uu [0..255])
+    [0..255] @=? fromJust (decode uu $ unchop uu $ chop uu 61 $ encode uu [0..255])
+    [0..255] @=? fromJust (decode uu $ unchop uu $ chop uu 100 $ encode uu [0..255])
+
+case_uuTestsFail = do
+    Nothing @=? decode uu "A"
+    Nothing @=? decode uu "aa"
+
+-- {{{1 xxencode tests
+xxTestData =
+    [ ("xx", "empty", "", [], xx)
+    , ("xx", "\\0", "++", [0], xx)
+    , ("xx", "\\255", "zk", [255], xx)
+    , ("xx", "AA", "EI2", [65, 65], xx)
+    , ("xx", "AAA", "EI3-", [65, 65, 65], xx)
+    , ("xx", "AAAA", "EI3-EE", [65, 65, 65, 65], xx)
+    , ("xx", "Example", "FLVVPL-gNE", [69,120,97,109,112,108,101], xx)
+    ]
+case_xxTest = mapM_ checkAssertions xxTestData
+
+case_xxTests2 = do
+    "EI2" @=? unchop xx (chop xx 1 "EI2")
+    "EI3-" @=? unchop xx (chop xx 1 "EI3-")
+    "EI3-EE" @=? unchop xx (chop xx 1 "EI3-EE")
+    [0..255] @=? fromJust (decode xx $ unchop xx $ chop xx 1 $ encode xx [0..255])
+    [0..255] @=? fromJust (decode xx $ unchop xx $ chop xx 61 $ encode xx [0..255])
+    [0..255] @=? fromJust (decode xx $ unchop xx $ chop xx 100 $ encode xx [0..255])
+
+case_xxTestsFail = do
+    Nothing @=? decode xx "A"
+    Nothing @=? decode xx "''"
+
+-- {{{1 base85 tests
+base85TestData =
+    [ ("base85", "empty", "", [], base85)
+    , ("base85", "f", "Ac", [102], base85)
+    , ("base85", "fo", "Ao@", [102,111], base85)
+    , ("base85", "foo", "AoDS", [102,111,111], base85)
+    , ("base85", "foob", "AoDTs", [102,111,111,98], base85)
+    , ("base85", "fooba", "AoDTs@/", [102,111,111,98,97], base85)
+    , ("base85", "foobar", "AoDTs@<)", [102,111,111,98,97,114], base85)
+    , ("base85", "\0", "!!", [0], base85)
+    , ("base85", "foob\0\0\0\0ar", "AoDTszEW", [102,111,111,98,0,0,0,0,114], base85)
+    , ("base85", "Example", "7<i6XE,9(", [69,120,97,109,112,108,101], base85)
+    , ("base85", "zeros", "z", [0, 0, 0, 0], base85)
+    , ("base85", "spaces", "y", [0x20, 0x20, 0x20, 0x20], base85)
+    ]
+case_base85Tests = mapM_ checkAssertions base85TestData
+
+case_base85TestsFail = do
+    Nothing @=? decode base85 "A"
+    Nothing @=? decode base85 "!z"
+    Nothing @=? decode base85 "!z!"
+    Nothing @=? decode base85 "!z!z"
+
+-- {{{1 base64 tests
+base64TestData =
+    [ ("base64", "empty", "", [], base64)
+    , ("base64", "f", "Zg==", [102], base64)
+    , ("base64", "fo", "Zm8=", [102,111], base64)
+    , ("base64", "foo", "Zm9v", [102,111,111], base64)
+    , ("base64", "foob", "Zm9vYg==", [102,111,111,98], base64)
+    , ("base64", "fooba", "Zm9vYmE=", [102,111,111,98,97], base64)
+    , ("base64", "foobar", "Zm9vYmFy", [102,111,111,98,97,114], base64)
+    , ("base64", "\0", "AA==", [0], base64)
+    , ("base64", "\255", "/w==", [255], base64)
+    , ("base64", "Example", "RXhhbXBsZQ==", [69,120,97,109,112,108,101], base64)
+    ]
+case_base64Tests = mapM_ checkAssertions base64TestData
+
+case_base64TestsFail = do
+    Nothing @=? decode base64 "A"
+    Nothing @=? decode base64 "!!"
+
+-- {{{1 base64url tests
+base64UrlTestData =
+    [ ("base64url", "empty", "", [], base64Url)
+    , ("base64url", "\0", "AA==", [0], base64Url)
+    , ("base64url", "\255", "_w==", [255], base64Url)
+    , ("base64url", "Example", "RXhhbXBsZQ==", [69,120,97,109,112,108,101], base64Url)
+    ]
+case_base64UrlTests = mapM_ checkAssertions base64UrlTestData
+
+-- {{{1 base32 tests
+base32TestData =
+    [ ("base32", "empty", "", [], base32)
+    , ("base32", "f", "MY======", [102], base32)
+    , ("base32", "fo", "MZXQ====", [102,111], base32)
+    , ("base32", "foo", "MZXW6===", [102,111,111], base32)
+    , ("base32", "foob", "MZXW6YQ=", [102,111,111,98], base32)
+    , ("base32", "fooba", "MZXW6YTB", [102,111,111,98,97], base32)
+    , ("base32", "foobar", "MZXW6YTBOI======", [102,111,111,98,97,114], base32)
+    ]
+case_base32Tests = mapM_ checkAssertions base32TestData
+
+case_base32TestsFail = do
+    Nothing @=? decode base32 "A"
+    Nothing @=? decode base32 "gh"
+
+-- {{{1 base32hex tests
+base32HexTestData =
+    [ ("base32hex", "empty", "", [], base32Hex)
+    , ("base32hex", "f", "CO======", [102], base32Hex)
+    , ("base32hex", "fo", "CPNG====", [102,111], base32Hex)
+    , ("base32hex", "foo", "CPNMU===", [102,111,111], base32Hex)
+    , ("base32hex", "foob", "CPNMUOG=", [102,111,111,98], base32Hex)
+    , ("base32hex", "fooba", "CPNMUOJ1", [102,111,111,98,97], base32Hex)
+    , ("base32hex", "foobar", "CPNMUOJ1E8======", [102,111,111,98,97,114], base32Hex)
+    ]
+case_base32HexTests = mapM_ checkAssertions base32HexTestData
+
+case_base32HexTestsFail = do
+    Nothing @=? decode base32Hex "A"
+    Nothing @=? decode base32Hex "gh"
+
+-- {{{1 base16 (hex)
+base16TestData =
+    [ ("base16", "empty", "", [], base16)
+    , ("base16", "f", "66", [102], base16)
+    , ("base16", "fo", "666F", [102,111], base16)
+    , ("base16", "foo", "666F6F", [102,111,111], base16)
+    , ("base16", "foob", "666F6F62", [102,111,111,98], base16)
+    , ("base16", "fooba", "666F6F6261", [102,111,111,98,97], base16)
+    , ("base16", "foobar", "666F6F626172", [102,111,111,98,97,114], base16)
+    ]
+case_base16Tests = mapM_ checkAssertions base16TestData
+
+case_base16TestsFail = do
+    Nothing @=? decode base16 "A"
+    Nothing @=? decode base16 "GH"
+
+-- {{{1 yEncoding
+case_yencTests = do
+    [] @=? Yenc.encode []
+    Just [] @=? Yenc.decode []
+    [0x90] @=? Yenc.encode [0x66]
+    Just [0x66] @=? Yenc.decode [0x90]
+    [0x90, 0x99, 0x99, 0x8c, 0x8b, 0x9c] @=? Yenc.encode [0x66, 0x6f, 0x6f, 0x62, 0x61, 0x72]
+    Just [0x66, 0x6f, 0x6f, 0x62, 0x61, 0x72] @=? Yenc.decode [0x90, 0x99, 0x99, 0x8c, 0x8b, 0x9c]
+    [0x3d, 0x40, 0x01] @=? Yenc.encode [0xd6, 0xd7]
+    Just [0xd6, 0xd7] @=? Yenc.decode [0x3d, 0x40, 0x01]
+    [0x3d, 0x40, 0x3d, 0x4a, 0x3d, 0x4d, 0x3d, 0x7d] @=? Yenc.encode [0xd6, 0xe0, 0xe3, 0x13]
+    Just [0xd6, 0xe0, 0xe3, 0x13] @=? Yenc.decode [0x3d, 0x40, 0x3d, 0x4a, 0x3d, 0x4d, 0x3d, 0x7d]
+    [[0x3d, 0x40], [0x01, 0x3d, 0x4a]] @=? Yenc.chop 2 [0x3d, 0x40, 0x01, 0x3d, 0x4a]
+
+-- {{{1 quoted-printable
+qpTestData =
+    [ ("qp", "empty", "", [], qp)
+    , ("qp", "foo=bar", "foo=3Dbar", [102,111,111,61,98,97,114], qp)
+    ]
+case_qpTests = mapM_ checkAssertions qpTestData
+
+case_qpTestsSucc = do
+    ["foo=","=3D=","bar"] @=? chop qp 4 "foo=3Dbar"
+
+case_qpTestsFail = do
+    Nothing @=? decode qp "=4"
+    Nothing @=? decode qp "=G"
+
+-- {{{1 python string
+pyTestData =
+    [ ("py", "empty", "", [], py)
+    , ("py", "<0x00><0x1f><0x20><0x7e><0x7f><0xff>", "\\x00\\x1F ~\\x7F\\xFF", [0x00, 0x1f, 0x20, 0x7e, 0x7f, 0xff], py)
+    , ("py", "\"\'\\", "\\\"\\'\\\\", [34, 39, 92], py)
+    ]
+case_pyTests = mapM_ checkAssertions pyTestData
+
+case_pyTestsFail = do
+    Nothing @=? decode py "\\z"
+
+-- {{{1 url encoding
+urlTestData =
+    [ ("url", "empty", "", [], url)
+    , ("url", "aA", "aA", [97, 65], url)
+    , ("url", "~ ", "~%20", [126, 0x20], url)
+    ]
+case_urlTests = mapM_ checkAssertions urlTestData
+
+case_urlTestsFail = do
+    Nothing @=? decode url "%ga"
+    Nothing @=? decode url "%%"
+
+-- {{{1 all the tests
+allTests = $(testGroupGenerator)
diff --git a/test-src/Test.hs b/test-src/Test.hs
new file mode 100644
index 0000000..2155bdf
--- /dev/null
+++ b/test-src/Test.hs
@@ -0,0 +1,16 @@
+{-
+ - Copyright : (c) 2009 Magnus Therning
+ - License   : BSD3
+ -}
+
+module Main
+    where
+
+import Test.Framework
+
+import qualified DataencQC as DQC
+import qualified DataencUT as DUT
+
+tests = [ DQC.allTests , DUT.allTests ]
+
+main = defaultMain tests