var falafel = require('../');
var test = require('tape');

test('for loop', function (t) {
    t.plan(3);
    
    var src = '(' + function () {
        var sum = 0;
        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++');
            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);
    //t.equal(res, 222 + 4 + 6 + 8);
});