reheat/old_test/connection/index.test.js

A red hot Node.js ORM for RethinkDB.

/*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);
		});
	}
};