feat(Lib/unittest): TestCase: addCleanup,tearDown,setUp,run

Author: litlighilit

src/pylib/Lib/unittest/case_py.nim

@@ -11,9 +11,12 @@ export TypeError
 
 import ./case_py/[
   types,
-  util
+  util,
+  prc_main,
+  prc_cleanup,
+  meth,
 ]
-export types
+export types, meth, prc_main, prc_cleanup
 
 template gen1(name, op){.dirty.} =
   bind check

src/pylib/Lib/unittest/case_py/meth.nim

@@ -0,0 +1,8 @@
+
+import ./types
+
+method setup*(self: TestCase){.base.} =
+  discard
+
+method teardown*(self: TestCase){.base.} =
+  discard

src/pylib/Lib/unittest/case_py/prc_cleanup.nim

@@ -0,0 +1,25 @@
+
+import std/macros
+import ./types
+
+macro addCleanup*(self: TestCase; cleanup: proc,
+    argsOrKws: varargs[untyped]) =
+  let attr = newDotExpr(self, ident("private.cleanups"))
+  let cb =
+    if argsOrKws.len == 0: cleanup
+    else: quote do:
+      proc() = `cleanup`(`argsOrKws`)
+  result = newCall("add", attr, cb)
+
+proc doCleanups*(self: TestCase) =
+  for cleanup in self.`private.cleanups`:
+    cleanup()
+
+when isMainModule:
+  var self = newTestCase()
+  var glb = 0
+  self.addCleanup(proc() = glb += 1)
+  self.addCleanup(proc(x, y: int) = glb += x+y, 1, y=3)
+  assert glb == 0
+  self.doCleanups()
+  assert glb == 5

src/pylib/Lib/unittest/case_py/prc_main.nim

@@ -0,0 +1,26 @@
+
+import std/macros
+import ./[types, prc_cleanup, meth]
+
+
+macro run*[T: TestCase](self: T) =
+  result = newStmtList()
+  when not declared(`T.dunder.dict.keys`):
+    error "TestCase must be subclassed via `class` pysugar"
+  else:
+    template call(self, meth): NimNode =
+      newCall(bindSym(astToStr(meth)), self)
+    result.add call(self, setup)
+    for fn in `T.dunder.dict.keys`():
+      if fn.startsWith("test_"):
+        let fnIdent = ident(fn)
+        result.add quote do:
+          `fnIdent`(`self`)
+    result.add call(self, teardown)
+
+    let doCleanupsId = bindSym("doCleanups")
+    result = quote do:
+      try:
+        `result`
+      finally:
+        `self`.`doCleanupsId`()

src/pylib/Lib/unittest/case_py/types.nim

@@ -1,6 +1,10 @@
 
-
 type
+  Cleanup = proc ()
   TestCase* = ref object of RootObj
+    `private.testMethodName`*: string
+    `private.cleanups`*: seq[Cleanup]
 
-func newTestCase*: TestCase = TestCase()
+func newTestCase*(methodName="runTest"): TestCase =
+  result = TestCase()
+  result.`private.testMethodName` = methodName