From b189b24a4f6f3afb5856db9634a55cbf6ca86375 Mon Sep 17 00:00:00 2001
From: Fabian Keller <bluenote10@users.noreply.github.com>
Date: Sun, 18 Jun 2017 00:01:27 +0200
Subject: [PATCH] fixes #25 (#26)

---
 .gitignore                           |  1 +
 src/karaxdsl.nim                     | 10 ++++++++--
 tests/compiler_tests.nim             | 18 ++++++++++++++++++
 tests/{diffDomTests.nims => nim.cfg} |  0
 tests/tester.nim                     |  1 +
 5 files changed, 28 insertions(+), 2 deletions(-)
 create mode 100644 tests/compiler_tests.nim
 rename tests/{diffDomTests.nims => nim.cfg} (100%)

diff --git a/.gitignore b/.gitignore
index 67d9b34..42b07a7 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
 nimcache/
+/tests/tester
\ No newline at end of file
diff --git a/src/karaxdsl.nim b/src/karaxdsl.nim
index a3558c8..e3e8290 100644
--- a/src/karaxdsl.nim
+++ b/src/karaxdsl.nim
@@ -47,12 +47,18 @@ proc tcall2(n, tmpContext: NimNode): NimNode =
     assert n.len == result.len
     if L > 0:
       result[L-1] = tcall2(result[L-1], tmpContext)
-  of nnkStmtList, nnkStmtListExpr, nnkWhenStmt, nnkIfStmt, nnkCaseStmt,
-     nnkTryStmt, nnkFinally:
+  of nnkStmtList, nnkStmtListExpr, nnkWhenStmt, nnkIfStmt, nnkTryStmt,
+     nnkFinally:
     # recurse for every child:
     result = copyNimNode(n)
     for x in n:
       result.add tcall2(x, tmpContext)
+  of nnkCaseStmt:
+    # recurse for children, but don't add call for case ident
+    result = copyNimNode(n)
+    result.add n[0]
+    for i in 1 ..< n.len:
+      result.add tcall2(n[i], tmpContext)
   of nnkVarSection, nnkLetSection, nnkConstSection:
     result = n
   of nnkCallKinds:
diff --git a/tests/compiler_tests.nim b/tests/compiler_tests.nim
new file mode 100644
index 0000000..4352cab
--- /dev/null
+++ b/tests/compiler_tests.nim
@@ -0,0 +1,18 @@
+include karaxprelude
+
+block:
+  type
+    TestEnum {.pure.} = enum
+      ValA, ValB
+
+  var e = TestEnum.ValA
+
+  proc renderDom(): VNode {.used.} =
+    result = buildHtml():
+      tdiv():
+        case e
+        of TestEnum.ValA:
+          tdiv: text "A"
+        of TestEnum.ValB:
+          tdiv: text "B"
+
diff --git a/tests/diffDomTests.nims b/tests/nim.cfg
similarity index 100%
rename from tests/diffDomTests.nims
rename to tests/nim.cfg
diff --git a/tests/tester.nim b/tests/tester.nim
index 5dc7426..482cad3 100644
--- a/tests/tester.nim
+++ b/tests/tester.nim
@@ -8,6 +8,7 @@ proc exec(cmd: string) =
 
 proc main =
   exec("nim js tests/diffDomTests.nim")
+  exec("nim js tests/compiler_tests.nim")
   exec("nim js examples/todoapp/todoapp.nim")
   exec("nim js examples/scrollapp/scrollapp.nim")
   exec("nim js examples/mediaplayer/playerapp.nim")