jmdobry/reheat

/*jshint loopfunc:true*/

var Connection = require('../../../build/instrument/lib/connection'),
    r = require('rethinkdb'),
    connection,
    mout = require('mout'),
    tableName = 'connection';

exports.ConnectionIntegration = {
    setUp: function (cb) {
        connection = new Connection();
        connection.run(r.dbList())
            .then(function (dbList) {
                if (mout.array.contains(dbList, 'test')) {
                    return connection.run(r.dbDrop('test'));
                } else {
                    return null;
                }
            })
            .then(function () {
                return connection.drain(function () {
                    connection.destroyAllNow();

                    connection = new Connection();
                    return connection.run(r.tableCreate(tableName), function (err) {
                        if (err) {
                            cb(err);
                        } else {
                            cb();
                        }
                    });
                });
            })
            .catch(cb)
            .error(cb);
    },

    tearDown: function (cb) {
        connection.run(r.dbList())
            .then(function (dbList) {
                if (mout.array.contains(dbList, 'test')) {
                    return connection.run(r.dbDrop('test'));
                }
                return null;
            })
            .then(function () {
                connection.drain(function () {
                    connection.destroyAllNow();
                    cb();
                });
            })
            .catch(cb)
            .error(cb);
    },
    insert: function (test) {
        test.expect(4);

        connection.run(r.table(tableName).insert({
            test: 'insert'
        }, { return_vals: true }), function (err, cursor) {
            test.ifError(err);
            test.equal(cursor.new_val.test, 'insert');
            test.ok(cursor.new_val.hasOwnProperty('id'));
            test.equal(cursor.inserted, 1);
            test.done();
        });
    },
    manyInsert: function (test) {
        test.expect(4003);

        connection.configure({
            max: 100
        }, function (err) {
            test.ifError(err);
            var completed = 0;
            for (var i = 0; i < 1000; i++) {
                connection.run(r.table(tableName).insert({
                    test: 'manyInsert'
                }, { return_vals: true }), function (err, cursor) {
                    test.ifError(err);
                    test.equal(cursor.new_val.test, 'manyInsert');
                    test.ok(cursor.new_val.hasOwnProperty('id'));
                    test.equal(cursor.inserted, 1);
                    completed += 1;
                });
            }
            var interval;
            setTimeout(function () {
                interval = setInterval(function () {
                    if (completed === 1000) {
                        connection.run(r.table(tableName).count(), function (err, count) {
                            test.ifError(err);
                            test.ok(count >= 1000);
                            clearInterval(interval);
                            test.done();
                        });
                    }
                }, 100);
            }, 200);
        });
    }
};