46 lines
1.5 KiB
Haskell
46 lines
1.5 KiB
Haskell
|
module Main where
|
||
|
|
|
||
|
|
import Test.HUnit
|
||
|
|
import Encoding.Base64 (enc64, dec64)
|
||
|
|
import System.Exit (exitFailure, exitSuccess)
|
||
|
|
|
||
|
|
-- Test cases for enc64 function
|
||
|
|
testEnc64 :: Test
|
||
|
|
testEnc64 = TestCase $ do
|
||
|
|
assertEqual "for (enc64 \"hello\")," "aGVsbG8=" (enc64 "hello")
|
||
|
|
assertEqual "for (enc64 \"Haskell\")," "SGFza2VsbA==" (enc64 "Haskell")
|
||
|
|
assertEqual "for (enc64 \"\x00\x01\x02\")," "AAEC" (enc64 "\x00\x01\x02")
|
||
|
|
assertEqual "for (enc64 \"😂\")," "8J+Ygg==" (enc64 "😂")
|
||
|
|
|
||
|
|
-- Test cases for dec64 function
|
||
|
|
testDec64 :: Test
|
||
|
|
testDec64 = TestCase $ do
|
||
|
|
assertEqual "for (dec64 \"aGVsbG8=\")," "hello" (dec64 "aGVsbG8=")
|
||
|
|
assertEqual "for (dec64 \"SGFza2VsbA==\")," "Haskell" (dec64 "SGFza2VsbA==")
|
||
|
|
assertEqual "for (dec64 \"AAEC\")," "\x00\x01\x02" (dec64 "AAEC")
|
||
|
|
assertEqual "for (dec64 \"8J+Ygg==\")," "😂" (dec64 "😂")
|
||
|
|
|
||
|
|
tests :: Test
|
||
|
|
tests = TestList [TestLabel "Test enc64" testEnc64,
|
||
|
|
TestLabel "Test dec64" testDec64]
|
||
|
|
|
||
|
|
-- main :: IO Counts
|
||
|
|
-- main = runTestTT tests >>= \counts -> print counts >> runTestTTAndExit tests
|
||
|
|
|
||
|
|
-- runAndPrint :: Test -> IO Counts
|
||
|
|
-- runAndPrint t = do
|
||
|
|
-- counts <- runTestTT t
|
||
|
|
-- let label = case t of
|
||
|
|
-- TestLabel l _ -> l
|
||
|
|
-- _ -> "Unnamed test."
|
||
|
|
-- putStrLn $ label ++ ": " ++ if errors counts + failures counts == 0 then "[OK]" else "[FAIL]"
|
||
|
|
-- return counts
|
||
|
|
|
||
|
|
main :: IO ()
|
||
|
|
main = do
|
||
|
|
counts <- runTestTT tests
|
||
|
|
print counts
|
||
|
|
if errors counts + failures counts == 0
|
||
|
|
then exitSuccess
|
||
|
|
else exitFailure
|