fix tests and traverser

yeonjuan · Nov 18, 2024 · 69ad912 · 69ad912
1 parent f41b47e
commit 69ad912
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 354 deletions.
diff --git a/packages/template-parser/lib/traverser.js b/packages/template-parser/lib/traverser.js
@@ -60,16 +60,14 @@ const visitorKeys = {
 function traverse(node, visitors) {
   const visitor = visitors[node.type];
   visitor && visitor(node);
-  const nextKeys = node[visitorKeys[node.type]];
-  if (!Array.isArray(nextKeys) || nextKeys.length === 0) {
-    return;
-  }
+  const nextKeys = visitorKeys[node.type];
+
   nextKeys.forEach((key) => {
     const next = node[key];
     if (Array.isArray(next)) {
-      next.forEach((n) => traverse(n, visitor));
-    } else {
-      traverse(next, visitor);
+      next.forEach((n) => traverse(n, visitors));
+    } else if (next) {
+      traverse(next, visitors);
     }
   });
 }

diff --git a/packages/template-parser/tests/__snapshots__/template-parser.test.js.snap b/packages/template-parser/tests/__snapshots__/template-parser.test.js.snap