based/test/Main.hs

module Main where

import Test.HUnit
import Encoding.Base64 (enc64, dec64)
import Encoding.Base32 (enc32, dec32)
import Encoding.Base16 (enc16, dec16)
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 "8J+Ygg==")

testDec32 :: Test
testDec32 = TestCase $ do
    assertEqual "for (dec32 \"JBSWY3DPFQQFO33SNRSCC===\")," "Hello, World!" (dec32 "JBSWY3DPFQQFO33SNRSCC===")
    assertEqual "for (dec32 \"JBQXG23FNRWA====\")," "Haskell" (dec32 "JBQXG23FNRWA====")
    -- assertEqual "for (dec32 \"AAEC\")," "\x00\x01\x02" (dec32 "AAEC")
    assertEqual "for (dec32 \"6CPZRAQ=\")," "😂" (dec32 "6CPZRAQ=")

testEnc32 :: Test
testEnc32 = TestCase $ do
    assertEqual "for (enc32 \"Hello, World!\")," "JBSWY3DPFQQFO33SNRSCC===" (enc32 "Hello, World!")
    assertEqual "for (enc32 \"Haskell\")," "JBQXG23FNRWA====" (enc32 "Haskell")
    assertEqual "for (enc32 \"😂\")," "6CPZRAQ=" (enc32 "😂")

testEnc16 :: Test
testEnc16 = TestCase $ do
    assertEqual "for (enc16 \"Hello, World!\")," "48656C6C6F2C20576F726C6421" (enc16 "Hello, World!")
    assertEqual "for (enc16 \"Haskell\")," "4861736B656C6C" (enc16 "Haskell")
    assertEqual "for (enc16 \"😂\")," "F09F9882" (enc16 "😂")

testDec16 :: Test
testDec16 = TestCase $ do
    assertEqual "for (dec16 \"48656C6C6F2C20576F726C6421\")," "Hello, World!" (dec16 "48656C6C6F2C20576F726C6421")
    assertEqual "for (dec16 \"4861736B656C6C\")," "Haskell" (dec16 "4861736B656C6C")
    assertEqual "for (dec16 \"F09F9882\")," "😂" (dec16 "F09F9882")

tests :: Test
tests = TestList [TestLabel "Test enc64" testEnc64,
                  TestLabel "Test dec64" testDec64,
                  TestLabel "Test enc32" testEnc32,
                  TestLabel "Test dec32" testDec32,
                  TestLabel "Test enc16" testEnc16,
                  TestLabel "Test dec16" testDec16]


-- 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