node-query/test/adapters/mysql2_test.js

/* eslint-env node, mocha */
'use strict';

// Load the test base
const reload = require('require-reload')(require);
reload.emptyCache();
const testBase = reload('../base');
const expect = testBase.expect;
const promiseTestRunner = testBase.promiseTestRunner;
const testRunner = testBase.testRunner;

// Load the test config file
let adapterName = 'mysql2';
const config = testBase.config[adapterName];

// Set up the query builder object
let nodeQuery = reload('../../lib/NodeQuery')(config);
let qb = nodeQuery.getQuery();

suite('Mysql2 adapter tests -', () => {
	test('nodeQuery.getQuery = nodeQuery.init', () => {
		expect(nodeQuery.getQuery())
			.to.be.deep.equal(qb);
	});

	// --------------------------------------------------------------------------
	// Callback Tests
	// --------------------------------------------------------------------------
	testRunner(qb, (err, result, done) => {
		expect(err).is.not.ok;
		expect(result.rows).is.an('array');
		expect(result.columns).is.an('array');
		expect(result.rowCount()).to.not.be.undefined;
		expect(result.columnCount()).to.not.be.undefined;
		done();
	});
	test('Callback - Select with function and argument in WHERE clause', done => {
		qb.select('id')
			.from('create_test')
			.where('id', 'CEILING(SQRT(88))')
			.get((err, rows) => {
				expect(err).is.not.ok;
				return done();
			});
	});
	test('Callback - Test Truncate', done => {
		qb.truncate('create_test', (err, res) => {
			return done(err);
		});
	});
	test('Callback - Test Insert Batch', done => {
		let data = [
			{
				id: 5441,
				key: 3,
				val: new Buffer('7')
			}, {
				id: 891,
				key: 34,
				val: new Buffer('10 o\'clock')
			}, {
				id: 481,
				key: 403,
				val: new Buffer('97')
			}
		];

		qb.insertBatch('create_test', data, (err, res) => {
			expect(err).is.not.ok;
			return done(err);
		});
	});

	// ---------------------------------------------------------------------------
	// Promise Tests
	// ---------------------------------------------------------------------------
	promiseTestRunner(qb);
	test('Promise - Select with function and argument in WHERE clause', () => {
		let promise = qb.select('id')
			.from('create_test')
			.where('id', 'CEILING(SQRT(88))')
			.get();

		return expect(promise).to.be.fulfilled;
	});
	test('Test Truncate', () => {
		let promise = qb.truncate('create_test');
		return expect(promise).to.be.fullfilled;
	});
	test('Test Insert Batch', () => {
		let data = [
			{
				id: 5442,
				key: 4,
				val: new Buffer('7')
			}, {
				id: 892,
				key: 35,
				val: new Buffer('10 o\'clock')
			}, {
				id: 482,
				key: 404,
				val: 97
			}
		];

		return expect(qb.insertBatch('create_test', data)).to.be.fulfilled;
	});

	suiteTeardown(() => {
		qb.end();
	});
});