After each (#6)

nuunit.nu

@@ -47,18 +47,23 @@ export def discover-tests [module testImportScript] {
     {
       id: ($it.index + 1)
       name: $it.item
-      exec: $'($testImportScript)
+      exec: $"($testImportScript)
               try {
-                if (scope commands | where name == "before each" | is-empty) {
+                if \(scope commands | where name == 'before each' | is-empty) {
                   null
                 } else {
                   before each
                 }
                 | ($it.item)
+                | if \(scope commands | where name == 'after each' | is-empty) {
+                  null
+                } else {
+                  after each
+                }
               } catch {|err|
                 print -e $err.debug
                 exit 1
-              }'
+              }"
     }
   }
 }

test-spec.nu

@@ -106,6 +106,14 @@ export def "test that the before each passes context to a test" [] {
   | verify json results
 }
 
+export def "test that the after each passes context to a test" [] {
+  use tests/test-spec-with-after-each.nu "verify json results"
+  let specFile = "tests/test-spec-with-after-each.nu"
+
+  run-test-spec $specFile
+  | verify json results
+}
+
 def run-test-spec [specFile] {
   (^$nu.current-exe --no-config-file nuunit.nu --test-spec-module-name $specFile --as-json)
   | from json

tests/test-spec-with-after-each.nu

@@ -0,0 +1,29 @@
+
+export def "test that after each gets context from test" [] {
+  let context = { some: stuff, that: is, useful: for, multiple: tests }
+  $context
+}
+
+export def "test that after each context gets passed to multiple tests" [] {
+  let context = { some: stuff, that: is, useful: for, multiple: tests }
+  $context
+}
+
+export def "after each" [] {
+  let actual = $in
+  let expected = { some: stuff, that: is, useful: for, multiple: tests }
+  use std assert
+
+  print "after each happened"
+
+  assert equal $expected $actual
+}
+
+export def "verify json results" [] {
+  let results = $in
+  use std assert
+
+  assert ($results | get exit_code | all {|it| $it == 0})
+  assert ($results | get stdout | all {|it| $it =~ "after each happened"})
+}
+

tests/test-spec-with-before-each.nu

@@ -24,7 +24,6 @@ export def "verify json results" [] {
   let results = $in
   use std assert
 
-  print ($results | get stdout)
-  assert equal 0 ($results | get exit_code | first)
+  assert ($results | get exit_code | all {|it| $it == 0})
 }