After each (#6)

Author: NonlinearFruit

nuunit.nu

@@ -55,6 +55,11 @@ export def discover-tests [module testImportScript] {
                   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})
 }