{-# LANGUAGE ConstraintKinds #-}
{-# LANGUAGE DefaultSignatures #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE ImplicitParams #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE TypeSynonymInstances #-}

-- | This package provides the ability to run a Checklist of several
-- "checks" during a single test.  A "bad" check does not immediately
-- result in a test failure; at the end of the test (passed or failed
-- due to primary testing), all failed checks are reported (and any
-- failed checks will result in an overall test failure at the end.
--
-- This type of checking can be very useful when needing to test
-- various aspects of an operation that is complex to setup, has
-- multiple effects, or where the checks are related such that knowing
-- about the multiple failures makes debugging easier.
--
-- An alternative approach is to have some sort of common preparation
-- code and use a separate test for each item.  This module simply
-- provides a convenient method to collate related items under the
-- aegis of a single test.
--
-- This package also provides the 'checkValues' function which can be
-- used to check a number of derived values from a single input value
-- via a checklist.  This can be used to independently verify a number
-- of record fields of a data structure or to validate related
-- operations performed from a single input.
--
-- See the documentation for 'check' and 'checkValues' for examples of
-- using this library.  The tests in the source package also provide
-- additional examples of usage.

module Test.Tasty.Checklist
  (
    -- * Checklist testing context
    withChecklist
  , CanCheck
  -- * Performing or Disabling checks
  , check
  , discardCheck
  -- * Type-safe multi-check specifications
  -- $checkValues
  -- $setup
  , checkValues
  , DerivedVal(Val, Got)
  -- * Error reporting
  , CheckResult
  , ChecklistFailures
  -- * Displaying tested values
  , TestShow(testShow)
  , testShowList
  )
where

import           Control.Exception ( evaluate )
import           Control.Monad ( join, unless )
import           Control.Monad.Catch
import           Control.Monad.IO.Class ( MonadIO, liftIO )
import           Data.IORef
import qualified Data.List as List
import qualified Data.Parameterized.Context as Ctx
import           Data.Text ( Text )
import qualified Data.Text as T
import           System.IO ( hFlush, hPutStrLn, stdout, stderr )


-- | The ChecklistFailures exception is thrown if any checks have
-- failed during testing.

data ChecklistFailures = ChecklistFailures Text [CheckResult]

-- | The internal 'CheckResult' captures the failure information for a check

data CheckResult = CheckFailed Text Text  -- check name from user, fail message

instance Exception ChecklistFailures

instance Show CheckResult where
  show :: CheckResult -> String
show (CheckFailed Text
what Text
msg) =
    String
"Failed check of " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> Text -> String
T.unpack Text
what String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
" with " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> Text -> String
T.unpack Text
msg

instance Show ChecklistFailures where
  show :: ChecklistFailures -> String
show (ChecklistFailures Text
topMsg [CheckResult]
fails) =
    String
"ERROR: " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> Text -> String
T.unpack Text
topMsg String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
"\n  " String -> ShowS
forall a. Semigroup a => a -> a -> a
<>
    Int -> String
forall a. Show a => a -> String
show ([CheckResult] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [CheckResult]
fails) String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
" checks failed in this checklist:\n  -" String -> ShowS
forall a. Semigroup a => a -> a -> a
<>
    String -> [String] -> String
forall a. [a] -> [[a]] -> [a]
List.intercalate String
"\n  -" (CheckResult -> String
forall a. Show a => a -> String
show (CheckResult -> String) -> [CheckResult] -> [String]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [CheckResult]
fails)

-- | A convenient Constraint to apply to functions that will perform
-- checks (i.e. call 'check' one or more times)

type CanCheck = (?checker :: IORef [CheckResult])


-- | This should be used to wrap the test that contains checks.  This
-- initializes the environment needed for the checks to run, and on
-- exit from the test, reports any (and all) failed checks as a test
-- failure.

withChecklist :: (MonadIO m, MonadMask m)
              => Text -> (CanCheck => m a) -> m a
              -- KWQ: if used TestTree instead of `m a`, could wrap the test like expected failure does, then update the Result with the check failures to get better output integration.
withChecklist :: Text -> (CanCheck => m a) -> m a
withChecklist Text
topMsg CanCheck => m a
t = do
  IORef [CheckResult]
checks <- IO (IORef [CheckResult]) -> m (IORef [CheckResult])
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (IO (IORef [CheckResult]) -> m (IORef [CheckResult]))
-> IO (IORef [CheckResult]) -> m (IORef [CheckResult])
forall a b. (a -> b) -> a -> b
$ [CheckResult] -> IO (IORef [CheckResult])
forall a. a -> IO (IORef a)
newIORef [CheckResult]
forall a. Monoid a => a
mempty
  a
r <- (let ?checker = checks in m a
CanCheck => m a
t)
       m a -> m () -> m a
forall (m :: * -> *) a b. MonadCatch m => m a -> m b -> m a
`onException` (IO () -> m ()
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (IO () -> m ()) -> IO () -> m ()
forall a b. (a -> b) -> a -> b
$
                       do [CheckResult]
cs <- [CheckResult] -> [CheckResult]
forall a. [a] -> [a]
List.reverse ([CheckResult] -> [CheckResult])
-> IO [CheckResult] -> IO [CheckResult]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> IORef [CheckResult] -> IO [CheckResult]
forall a. IORef a -> IO a
readIORef IORef [CheckResult]
checks
                          Bool -> IO () -> IO ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
unless ([CheckResult] -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null [CheckResult]
cs) (IO () -> IO ()) -> IO () -> IO ()
forall a b. (a -> b) -> a -> b
$ do
                            Handle -> IO ()
hFlush Handle
stdout
                            Handle -> String -> IO ()
hPutStrLn Handle
stderr String
""
                            let pfx :: String
pfx = String
"        WARN "
                            (CheckResult -> IO ()) -> [CheckResult] -> IO ()
forall (t :: * -> *) (m :: * -> *) a b.
(Foldable t, Monad m) =>
(a -> m b) -> t a -> m ()
mapM_ (Handle -> String -> IO ()
hPutStrLn Handle
stderr (String -> IO ())
-> (CheckResult -> String) -> CheckResult -> IO ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (String
pfx String -> ShowS
forall a. Semigroup a => a -> a -> a
<>) ShowS -> (CheckResult -> String) -> CheckResult -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. CheckResult -> String
forall a. Show a => a -> String
show) [CheckResult]
cs
                            Handle -> IO ()
hFlush Handle
stderr
                     )

  -- If t failed, never get here:
  IO () -> m ()
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (IO () -> m ()) -> IO () -> m ()
forall a b. (a -> b) -> a -> b
$ do
    [CheckResult]
collected <- [CheckResult] -> [CheckResult]
forall a. [a] -> [a]
List.reverse ([CheckResult] -> [CheckResult])
-> IO [CheckResult] -> IO [CheckResult]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> IORef [CheckResult] -> IO [CheckResult]
forall a. IORef a -> IO a
readIORef IORef [CheckResult]
checks
    Bool -> IO () -> IO ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
unless ([CheckResult] -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null [CheckResult]
collected) (IO () -> IO ()) -> IO () -> IO ()
forall a b. (a -> b) -> a -> b
$
      ChecklistFailures -> IO ()
forall (m :: * -> *) e a. (MonadThrow m, Exception e) => e -> m a
throwM (Text -> [CheckResult] -> ChecklistFailures
ChecklistFailures Text
topMsg [CheckResult]
collected)
  a -> m a
forall (m :: * -> *) a. Monad m => a -> m a
return a
r

-- | This is used to run a check within the code.  The first argument
-- is the "name" of this check, the second is a function that takes a
-- value and returns 'True' if the value is OK, or 'False' if the
-- value fails the check.  The last argument is the value to check.
--
-- >>> :set -XOverloadedStrings
-- >>> import Test.Tasty
-- >>> import Test.Tasty.HUnit
-- >>> :{
-- >>> defaultMain $ testCase "odd numbers" $ withChecklist "odds" $ do
-- >>>  let three = 3 :: Int
-- >>>  check "three is odd" odd three
-- >>>  check "two is odd" odd (2 :: Int)
-- >>>  check "7 + 3 is odd" odd $ 7 + three
-- >>>  check "7 is odd" odd (7 :: Int)
-- >>> :}
-- odd numbers: FAIL
--   Exception: ERROR: odds
--     2 checks failed in this checklist:
--     -Failed check of two is odd with 2
--     -Failed check of 7 + 3 is odd with 10
-- <BLANKLINE>
-- 1 out of 1 tests failed (...s)
-- *** Exception: ExitFailure 1
--
-- Any check failures are also printed to stdout (and omitted from the
-- above for clarity).  This is so that those failures are reported
-- even if a more standard test assertion is used that prevents
-- completion of the checklist.  Thus, if an @assertEqual "values"
-- three 7@ had been added to the above, that would have been the only
-- actual (and immediate) fail for the test, but any failing 'check's
-- appearing before that @assertEqual@ would still have printed.

check :: (CanCheck, TestShow a, MonadIO m)
      => Text -> (a -> Bool) -> a -> m ()
check :: Text -> (a -> Bool) -> a -> m ()
check Text
what a -> Bool
eval a
val = do
  Bool
r <- IO Bool -> m Bool
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (IO Bool -> m Bool) -> IO Bool -> m Bool
forall a b. (a -> b) -> a -> b
$ Bool -> IO Bool
forall a. a -> IO a
evaluate (a -> Bool
eval a
val)
  Bool -> m () -> m ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
unless Bool
r (m () -> m ()) -> m () -> m ()
forall a b. (a -> b) -> a -> b
$ do
    let chk :: CheckResult
chk = Text -> Text -> CheckResult
CheckFailed Text
what (Text -> CheckResult) -> Text -> CheckResult
forall a b. (a -> b) -> a -> b
$ String -> Text
T.pack (String -> Text) -> String -> Text
forall a b. (a -> b) -> a -> b
$ a -> String
forall v. TestShow v => v -> String
testShow a
val
    IO () -> m ()
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (IO () -> m ()) -> IO () -> m ()
forall a b. (a -> b) -> a -> b
$ IORef [CheckResult] -> ([CheckResult] -> [CheckResult]) -> IO ()
forall a. IORef a -> (a -> a) -> IO ()
modifyIORef CanCheck
IORef [CheckResult]
?checker (CheckResult
chkCheckResult -> [CheckResult] -> [CheckResult]
forall a. a -> [a] -> [a]
:)


-- | Sometimes checks are provided in common testing code, often in
-- setup/preparation for the main tests.  In some cases, the check is
-- not applicable for that particular test.  This function can be used
-- to discard any pending failures for the associated named check.
--
-- This is especially useful when a common code block is used to
-- perform a set of checks: if a few of the common checks are not
-- appropriate for the current situation, 'discardCheck' can be used
-- to throw away the results of those checks by matching on the check
-- name.

discardCheck :: (CanCheck, MonadIO m) => Text -> m ()
discardCheck :: Text -> m ()
discardCheck Text
what = do
  let isCheck :: Text -> CheckResult -> Bool
isCheck Text
n (CheckFailed Text
n' Text
_) = Text
n Text -> Text -> Bool
forall a. Eq a => a -> a -> Bool
== Text
n'
  IO () -> m ()
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (IO () -> m ()) -> IO () -> m ()
forall a b. (a -> b) -> a -> b
$ IORef [CheckResult] -> ([CheckResult] -> [CheckResult]) -> IO ()
forall a. IORef a -> (a -> a) -> IO ()
modifyIORef CanCheck
IORef [CheckResult]
?checker ((CheckResult -> Bool) -> [CheckResult] -> [CheckResult]
forall a. (a -> Bool) -> [a] -> [a]
filter (Bool -> Bool
not (Bool -> Bool) -> (CheckResult -> Bool) -> CheckResult -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> CheckResult -> Bool
isCheck Text
what))

----------------------------------------------------------------------

-- $checkValues
--
-- Implementing a number of discrete 'check' calls can be tedious,
-- especially when they are validating different aspects of the same
-- result value.  To facilitate this, the 'checkValues' function can
-- be used along with a type-safe list of checks to perform.
--
-- To demonstrate this, first consider the following sample program,
-- which has code that generates a complex @Struct@ value, along with
-- tests for various fields in that @Struct@.

-- $setup
-- >>> :set -XPatternSynonyms
-- >>> :set -XOverloadedStrings
-- >>>
-- >>> import Data.Parameterized.Context ( pattern Empty, pattern (:>) )
-- >>> import Test.Tasty.Checklist
-- >>> import Test.Tasty
-- >>> import Test.Tasty.HUnit
-- >>>
-- >>> :{
-- >>> data Struct = MyStruct { foo :: Int, bar :: Char, baz :: String }
-- >>>
-- >>> instance Show Struct where
-- >>>    show s = baz s <> " is " <> show (foo s) <> [bar s]
-- >>> instance TestShow Struct where testShow = show
-- >>>
-- >>> someFun :: Int -> Struct
-- >>> someFun n = MyStruct (n * 6)
-- >>>               (if n * 6 == 42 then '!' else '?')
-- >>>               "The answer to the universe"
-- >>>
-- >>> oddAnswer :: Struct -> Bool
-- >>> oddAnswer = odd . foo
-- >>>
-- >>> test = testCase "someFun result" $
-- >>>    withChecklist "results for someFun" $
-- >>>    someFun 3 `checkValues`
-- >>>         (Empty
-- >>>         :> Val "foo" foo 42
-- >>>         :> Val "baz field" baz "The answer to the universe"
-- >>>         :> Val "shown" show "The answer to the universe is 42!"
-- >>>         :> Val "odd answer" oddAnswer False
-- >>>         :> Got "even answer" (not . oddAnswer)
-- >>>         :> Val "double-checking foo" foo 42
-- >>>         )
-- >>> :}
--
-- This code will be used below to demonstrate various advanced
-- checklist capabilities.

-- | The 'checkValues' is a checklist that tests various values that
-- can be derived from the input value.  The input value is provided,
-- along with an 'Data.Parameterized.Context.Assignment' list of
-- extraction functions and the expected result value (and name) of
-- that extraction.  Each extraction is performed as a check within
-- the checklist.
--
-- This is convenient to gather together a number of validations on a
-- single datatype and represent them economically.
--
-- One example is testing the fields of a record structure, given the
-- above code:
--
-- >>> defaultMain test
-- someFun result: FAIL
--   Exception: ERROR: results for someFun
--     3 checks failed in this checklist:
--     -Failed check of foo on input <<The answer to the universe is 18?>> expected <<42>> but failed with 18
--     -Failed check of shown on input <<The answer to the universe is 18?>> expected <<"The answer to the universe is 42!">> but failed with "The answer to the universe is 18?"
--     -Failed check of double-checking foo on input <<The answer to the universe is 18?>> expected <<42>> but failed with 18
-- <BLANKLINE>
-- 1 out of 1 tests failed (...s)
-- *** Exception: ExitFailure 1
--
-- In this case, several of the values checked were correct, but more
-- than one was wrong.  Helpfully, this test output lists /all/ the
-- wrong answers for the single input provided.

checkValues :: CanCheck
            => TestShow dType
            => dType -> Ctx.Assignment (DerivedVal dType) idx ->  IO ()
checkValues :: dType -> Assignment (DerivedVal dType) idx -> IO ()
checkValues dType
got Assignment (DerivedVal dType) idx
expF =
  IO (IO ()) -> IO ()
forall (m :: * -> *) a. Monad m => m (m a) -> m a
join (IO (IO ()) -> IO ()) -> IO (IO ()) -> IO ()
forall a b. (a -> b) -> a -> b
$ () -> IO ()
forall a. a -> IO a
evaluate (() -> IO ()) -> IO () -> IO (IO ())
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (forall tp. Index idx tp -> DerivedVal dType tp -> IO ())
-> Assignment (DerivedVal dType) idx -> IO ()
forall k (m :: * -> *) (ctx :: Ctx k) (f :: k -> *).
Applicative m =>
(forall (tp :: k). Index ctx tp -> f tp -> m ())
-> Assignment f ctx -> m ()
Ctx.traverseWithIndex_ (dType -> Index idx tp -> DerivedVal dType tp -> IO ()
forall dType (idx :: Ctx *) valType.
(CanCheck, TestShow dType) =>
dType -> Index idx valType -> DerivedVal dType valType -> IO ()
chkValue dType
got) Assignment (DerivedVal dType) idx
expF


chkValue :: CanCheck
         => TestShow dType
         => dType -> Ctx.Index idx valType -> DerivedVal dType valType -> IO ()
chkValue :: dType -> Index idx valType -> DerivedVal dType valType -> IO ()
chkValue dType
got Index idx valType
_idx = \case
  (Val Text
txt dType -> valType
fld valType
v) ->
    let r :: valType
r = dType -> valType
fld dType
got
        msg :: Text
msg = Text
txt Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
" on input <<" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
ti Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
">> expected <<" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
tv Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
">> but failed"
        ti :: Text
ti = String -> Text
T.pack (dType -> String
forall v. TestShow v => v -> String
testShow dType
got)
        tv :: Text
tv = String -> Text
T.pack (valType -> String
forall v. TestShow v => v -> String
testShow valType
v)
    in Text -> (valType -> Bool) -> valType -> IO ()
forall a (m :: * -> *).
(CanCheck, TestShow a, MonadIO m) =>
Text -> (a -> Bool) -> a -> m ()
check Text
msg (valType
v valType -> valType -> Bool
forall a. Eq a => a -> a -> Bool
==) valType
r
  (Got Text
txt dType -> Bool
fld) ->
    let r :: Bool
r = dType -> Bool
fld dType
got
        msg :: Text
msg = Text
txt Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
" on input <<" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
ti Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
">>"
        ti :: Text
ti = String -> Text
T.pack (dType -> String
forall v. TestShow v => v -> String
testShow dType
got)
    in Text -> (Bool -> Bool) -> Bool -> IO ()
forall a (m :: * -> *).
(CanCheck, TestShow a, MonadIO m) =>
Text -> (a -> Bool) -> a -> m ()
check Text
msg (Bool
True Bool -> Bool -> Bool
forall a. Eq a => a -> a -> Bool
==) Bool
r

-- | Each entry in the 'Data.Parameterized.Context.Assignment' list
-- for 'checkValues' should be one of these 'DerivedVal' values.

data DerivedVal i d where

  -- | Val allows specification of a description string, an extraction
  -- function, and the expected value to be extracted.  The
  -- 'checkValues' function will add a Failure if the expected value is
  -- not obtained.
  Val :: (TestShow d, Eq d) => Text -> (i -> d) -> d -> DerivedVal i d

  -- | Got allow specification of a description string and an
  -- extraction function.  The 'checkValues' function will add a
  -- Failure if the extraction result is False.
  --
  -- > Val "what" f True === Got "what" f
  --
  Got :: Text -> (i -> Bool) -> DerivedVal i Bool

----------------------------------------------------------------------

-- | The 'TestShow' class is defined to provide a way for the various
-- data objects tested by this module to be displayed when tests fail.
-- The default 'testShow' will use a 'Show' instance, but this can be
-- overridden if there are alternate ways t o display a particular
-- object (e.g. pretty-printing, etc.)

class TestShow v where
  testShow :: v -> String
  default testShow :: Show v => v -> String
  testShow = v -> String
forall a. Show a => a -> String
show

-- Some TestShow instances using Show for regular datatypes
instance TestShow ()
instance TestShow Bool
instance TestShow Int
instance TestShow Integer
instance TestShow Float
instance TestShow Char
instance TestShow String

instance (TestShow a, TestShow b) => TestShow (a,b) where
  testShow :: (a, b) -> String
testShow (a
a,b
b) = String
"(" String -> ShowS
forall a. Semigroup a => a -> a -> a
<> a -> String
forall v. TestShow v => v -> String
testShow a
a String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
", " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> b -> String
forall v. TestShow v => v -> String
testShow b
b String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
")"
instance (TestShow a, TestShow b, TestShow c) => TestShow (a,b,c) where
  testShow :: (a, b, c) -> String
testShow (a
a,b
b,c
c) = String
"(" String -> ShowS
forall a. Semigroup a => a -> a -> a
<> a -> String
forall v. TestShow v => v -> String
testShow a
a String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
", " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> b -> String
forall v. TestShow v => v -> String
testShow b
b String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
", " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> c -> String
forall v. TestShow v => v -> String
testShow c
c String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
")"

-- | A helper function for defining a testShow for lists of items.
--
-- > instance TestShow [Int] where testShow = testShowList

testShowList :: TestShow v => [v] -> String
testShowList :: [v] -> String
testShowList  [v]
l = String
"[ " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> (String -> [String] -> String
forall a. [a] -> [[a]] -> [a]
List.intercalate String
", " (v -> String
forall v. TestShow v => v -> String
testShow (v -> String) -> [v] -> [String]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [v]
l)) String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
" ]"