module Test.Hspec.Runner (
Specs, hspec, hspecX, hspecB, hHspec, hHspecWithFormat, describe, it, toExitCode
) where
import Test.Hspec.Internal
import Test.Hspec.Core (EvaluatedSpec, Specs)
import Test.Hspec.Formatters
import Test.Hspec.Formatters.Internal
import System.IO
import System.Exit
runFormatter :: Formatter -> Spec -> FormatM EvaluatedSpec
runFormatter formatter = go 0 [] . unSpec
where
go nesting groups (SpecGroup group xs) = do
exampleGroupStarted formatter nesting group
ys <- mapM (go (succ nesting) (group : groups)) xs
return (SpecGroup group ys)
go nesting groups (SpecExample requirement e) = do
result <- liftIO $ safeEvaluateExample e
case result of
Success -> do
increaseSuccessCount
exampleSucceeded formatter nesting requirement
Fail err -> do
increaseFailCount
exampleFailed formatter nesting requirement err
n <- getFailCount
addFailMessage $ failureDetails groups requirement err n
Pending reason -> do
increasePendingCount
examplePending formatter nesting requirement reason
return (SpecExample requirement result)
failureDetails :: [String] -> String -> String -> Int -> String
failureDetails groups requirement err i =
show i ++ ") " ++ groups_ ++ requirement ++ " FAILED" ++ err_
where
err_
| null err = ""
| otherwise = "\n" ++ err
groups_ = case groups of
[x] -> x ++ " "
_ -> concatMap (++ " - ") (reverse groups)
hspecX :: Specs -> IO a
hspecX ss = hspecB ss >>= exitWith . toExitCode
hspecB :: Specs -> IO Bool
hspecB ss = success `fmap` hspec ss
where
success :: [EvaluatedSpec] -> Bool
success = not . failure
failure :: [EvaluatedSpec] -> Bool
failure = any p
where
p (SpecGroup _ xs) = any p xs
p (SpecExample _ x) = isFailure x
isFailure :: Result -> Bool
isFailure (Fail _) = True
isFailure _ = False
hspec :: Specs -> IO [EvaluatedSpec]
hspec = hHspec stdout
hHspec :: Handle -> Specs -> IO [EvaluatedSpec]
hHspec h specs = do
useColor <- hIsTerminalDevice h
hHspecWithFormat specdoc useColor h specs
hHspecWithFormat :: Formatter -> Bool -> Handle -> Specs -> IO [EvaluatedSpec]
hHspecWithFormat formatter useColor h ss = runFormatM useColor h $ do
specList <- mapM (runFormatter formatter) ss
failedFormatter formatter
footerFormatter formatter
return specList
toExitCode :: Bool -> ExitCode
toExitCode True = ExitSuccess
toExitCode False = ExitFailure 1