Use "Walker" class instead of traverse()

for Ember 1.13 compat
mainmatter · Jan 23, 2017 · ff661e3 · ff661e3
1 parent fb17788
commit ff661e3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/transform-test-selector-params-to-hash-pairs.js b/transform-test-selector-params-to-hash-pairs.js
@@ -12,10 +12,10 @@ function isTestSelectorParam(param) {
 
 TransformTestSelectorParamsToHashPairs.prototype.transform = function(ast) {
   var b = this.syntax.builders;
-  var traverse = this.syntax.traverse;
+  var walker = new this.syntax.Walker();
 
-  traverse(ast, {
-    MustacheStatement: function(node) {
+  walker.visit(ast, function(node) {
+    if (node.type === 'MustacheStatement' || node.type === 'BlockStatement') {
       var testSelectorParams = [];
       var otherParams = [];
 
@@ -24,7 +24,7 @@ TransformTestSelectorParamsToHashPairs.prototype.transform = function(ast) {
           testSelectorParams.push(param);
         } else {
           otherParams.push(param);
-        };
+        }
       });
 
       node.params = otherParams;