From 25174e0cf448c9a80da2170e4961fac0598d8425 Mon Sep 17 00:00:00 2001
From: rhiannon morris <rhi@rhiannon.website>
Date: Mon, 17 Jul 2023 20:00:40 +0200
Subject: [PATCH] print totals at the end

---
 TAP.idr | 54 +++++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 41 insertions(+), 13 deletions(-)

diff --git a/TAP.idr b/TAP.idr
index 83e6c6f..c70a544 100644
--- a/TAP.idr
+++ b/TAP.idr
@@ -226,9 +226,18 @@ record RunnerEnv where
   ||| whether to include control codes for colours
   color  : Bool
 
+public export
+record Results where
+  constructor Res
+  pass, fail, skip, todo : Nat
+
+public export
+zeroRes : Results
+zeroRes = Res 0 0 0 0
+
 private
 Runner : Type -> Type
-Runner = ReaderT RunnerEnv IO
+Runner = ReaderT RunnerEnv $ StateT Results IO
 
 ||| print some lines at the current indent level
 private
@@ -285,24 +294,41 @@ private
 putOk : Bool -> Nat -> String -> Runner ()
 putOk ok = putOk' (okCol ok) ok
 
+private
+putComment : String -> Runner ()
+putComment str = putIndentLines [!(col Magenta "# ") ++ str]
+
 ||| print a TAP version line
 private
 putVersion : TAPVersion -> Runner ()
 putVersion ver = putColor Cyan "TAP version \{show ver}"
 
+||| print comments at the end counting passes, failures, skips, and todos
+private
+putFooter : Runner ()
+putFooter = do
+  res <- get
+  putComment "passed:  \{show res.pass}"
+  putComment "failed:  \{show res.fail}"
+  putComment "skipped: \{show res.skip}"
+  putComment "todo:    \{show res.todo}"
+
 ||| run a test, print its line, and return whether it succeeded
 private
 run1' : (Nat, TestBase) -> Runner Bool
 run1' (index, test) = do
   res <- liftIO test.run
   case res of
-       Tried ok info => do
-         putOk ok index test.label
-         local {indent $= plus 2} $ putIndentLines $ toLines info
-       Skip reason =>
-         putOk' Yellow True index "\{test.label}  # skip \{reason}"
-       Todo reason =>
-         putOk' Yellow True index "\{test.label}  # todo \{reason}"
+    Tried ok info => do
+      modify $ \s : Results => if ok then {pass $= S} s else {fail $= S} s
+      putOk ok index test.label
+      local {indent $= plus 2} $ putIndentLines $ toLines info
+    Skip reason => do
+      modify $ \s : Results => {skip $= S} s
+      putOk' Yellow True index "\{test.label}  # skip \{reason}"
+    Todo reason => do
+      modify $ \s : Results => {todo $= S} s
+      putOk' Yellow True index "\{test.label}  # todo \{reason}"
   pure $ toBool res
 
 parameters (skipNotes : Bool)
@@ -317,7 +343,7 @@ parameters (skipNotes : Bool)
       putOk res index label
       pure res
     run' (_, Note note) = do
-      unless skipNotes $ putIndentLines [!(col Magenta "# ") ++ note]
+      unless skipNotes $ putComment note
       pure True
 
     private
@@ -359,10 +385,12 @@ main' : Options -> List Test -> IO ExitCode
 main' opts tests = do
   let tests = filterMatch opts.pattern $
               case opts.version of V13 => flatten tests; V14 => tests
-  let act = do putVersion opts.version; runList opts.skipComments tests
-  pure $ if !(runReaderT (RE 0 opts.color) act)
-            then ExitSuccess
-            else ExitFailure 70
+  res <- evalStateT zeroRes $ runReaderT (RE 0 opts.color) $ do
+    putVersion opts.version
+    res <- runList opts.skipComments tests
+    putFooter
+    pure res
+  pure $ if res then ExitSuccess else ExitFailure 70
 
 ||| run tests and exit with an appropriate code
 export