deps/npm/node_modules/readable-stream/node_modules/inline-process-browser/node_modules/falafel/test/for.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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);
});