{-# LANGUAGE RankNTypes #-}
module Utils
  ( assertBool
  , assertEqual
  , assertFailure
  , assertException
  ) where

import GHC.Stack
import qualified Test.Tasty.HUnit as T

import Effectful
import Control.Exception

assertBool :: (HasCallStack, IOE :> es) => String -> Bool -> Eff es ()
assertBool msg p = liftIO $ T.assertBool msg p

assertEqual
  :: (HasCallStack, Eq a, Show a, IOE :> es)
  => String
  -> a
  -> a
  -> Eff es ()
assertEqual msg expected given = liftIO $ T.assertEqual msg expected given

assertFailure :: (HasCallStack, IOE :> es) => String -> Eff es a
assertFailure msg = liftIO $ T.assertFailure msg

assertException :: forall e es a
                . (HasCallStack, Exception e, IOE :> es)
                => IO a
                -> (e -> Bool)
                -> Eff es ()
assertException action expectedClassifier = do
  r <- liftIO $ try action
  case r of
      Right _ ->
        assertFailure "The action did not raise an exception"
      Left e ->
        assertBool "The expected exception doesn't match what was given" (expectedClassifier e)