Green OKs and red FAILs test helper :)

colorfulAsserts.hs

{-# LANGUAGE OverloadedStrings #-}
module Main where

import Lib

import GHC.List (take)
import Data.Text as T
import System.Console.ANSI
import System.Exit (exitFailure)

assert :: String -> Bool -> IO Bool
assert 
    testTitle
    test =
        do
            let printColorfulResult txt col = do
                    putStr "["
                    setSGR [SetColor Foreground Vivid col]
                    putStr txt
                    setSGR [Reset]
                    putStr "] "
                
            if test
                then printColorfulResult " OK " Green
                else printColorfulResult "FAIL" Red

            putStrLn testTitle
            return test

main :: IO ()
main = do
    putStrLn ""
    putStrLn "------------------------------"
    putStrLn "Running tests now..."
    
    result <-
        sequence [ 
            (assert "Success example" True),
            (assert "Fail example" False)
        ]

    putStrLn "------------------------------"
    if and $ result
       then putStrLn "Success!"
       else exitFailure