flow like the river

BACK_BACK/node_modules/falafel/test/for.js

@@ -0,0 +1,30 @@
+var falafel = require('../');
+var test = require('tape');
+
+test('for loop', function (t) {
+    t.plan(7);
+    
+    var src = '(function () {'
+        + 'var sum = 0;'
+        + 'for (var i = 0; i < 10; i++)'
+            + 'sum += i;'
+        + 'if (true)'
+            + 'for (var i = 0; i < 10; i++)'
+                + 'sum += i;'
+        + 'return sum;'
+    + '})()';
+    
+    var output = falafel(src, function (node) {
+        if (node.type === 'ForStatement') {
+            t.equal(node.update.source(), 'i++');
+            t.equal(node.update.type, "UpdateExpression");
+            node.update.update('i+=2');
+        }
+        if (node.type === 'UpdateExpression') {
+            t.equal(node.source(), 'i++');
+        }
+    });
+    
+    var res = Function('return ' + output)();
+    t.equal(res, 2 + 4 + 6 + 8 + 2 + 4 + 6 + 8);
+});