jugglingdb
Version:
Node.js ORM for every database: redis, mysql, mongodb, postgres, sqlite, ...
410 lines (356 loc) • 13 kB
JavaScript
// This test written in mocha+should.js
const should = require('./init.js');
let j = require('../'),
Schema = j.Schema,
AbstractClass = j.AbstractClass,
Hookable = j.Hookable,
db, User;
describe('hooks', function() {
before(function(done) {
db = getSchema();
User = db.define('User', {
email: { type: String, index: true, limit: 100 },
name: String,
password: String,
state: String
});
db.automigrate(done);
});
describe('behavior', function() {
it('should allow to break flow in case of error', function(done) {
const Model = db.define('Model');
Model.beforeCreate = function(next, data) {
next(new Error('Fail'));
};
Model.create(function(err, model) {
should.not.exist(model);
should.exist(err);
done();
});
});
});
describe('initialize', function() {
afterEach(function() {
User.afterInitialize = null;
});
it('should be triggered on new', function(done) {
User.afterInitialize = function() {
done();
};
new User;
});
it('should be triggered on create', function(done) {
let user;
User.afterInitialize = function() {
if (this.name === 'Nickolay') {
this.name += ' Rozental';
}
};
User.create({ name: 'Nickolay' }, function(err, u) {
u.id.should.be.ok;
u.name.should.equal('Nickolay Rozental');
done();
});
});
});
describe('create', function() {
afterEach(removeHooks('Create'));
it('should be triggered on create', function(done) {
addHooks('Create', done);
User.create();
});
it('should not be triggered on new', function() {
User.beforeCreate = function(next) {
should.fail('This should not be called');
next();
};
const u = new User;
});
it('should be triggered on new+save', function(done) {
addHooks('Create', done);
(new User).save();
});
it('afterCreate should not be triggered on failed create', function(done) {
const old = User.schema.adapter.create;
User.schema.adapter.create = function(modelName, id, cb) {
cb(new Error('error'));
};
User.afterCreate = function() {
throw new Error('shouldn\'t be called');
};
User.create(function(err, user) {
User.schema.adapter.create = old;
done();
});
});
});
describe('save', function() {
afterEach(removeHooks('Save'));
it('should be triggered on create', function(done) {
addHooks('Save', done);
User.create();
});
it('should be triggered on new+save', function(done) {
addHooks('Save', done);
(new User).save();
});
it('should be triggered on updateAttributes', function(done) {
User.create(function(err, user) {
addHooks('Save', done);
user.updateAttributes({ name: 'Anatoliy' });
});
});
it('should be triggered on save', function(done) {
User.create(function(err, user) {
addHooks('Save', done);
user.name = 'Hamburger';
user.save();
});
});
it('should save full object', function(done) {
User.create(function(err, user) {
User.beforeSave = function(next, data) {
data.should.have.keys('id', 'name', 'email',
'password', 'state');
done();
};
user.save();
});
});
it('should save actual modifications to database', function(done) {
User.beforeSave = function(next, data) {
data.password = 'hash';
next();
};
User.destroyAll(function() {
User.create({
email: 'james.bond@example.com',
password: '53cr3t'
}, function() {
User.findOne({
where: { email: 'james.bond@example.com' }
}, function(err, jb) {
jb.password.should.equal('hash');
done();
});
});
});
});
it('should save actual modifications on updateAttributes', function(done) {
User.beforeSave = function(next, data) {
data.password = 'hash';
next();
};
User.destroyAll(function() {
User.create({
email: 'james.bond@example.com'
}, function(err, u) {
u.updateAttribute('password', 'new password', function(e, u) {
should.not.exist(e);
should.exist(u);
u.password.should.equal('hash');
User.findOne({
where: { email: 'james.bond@example.com' }
}, function(err, jb) {
jb.password.should.equal('hash');
done();
});
});
});
});
});
});
describe('update', function() {
afterEach(removeHooks('Update'));
it('should not be triggered on create', function() {
User.beforeUpdate = function(next) {
should.fail('This should not be called');
next();
};
User.create();
});
it('should not be triggered on new+save', function() {
User.beforeUpdate = function(next) {
should.fail('This should not be called');
next();
};
(new User).save();
});
it('should be triggered on updateAttributes', function(done) {
User.create(function(err, user) {
addHooks('Update', done);
user.updateAttributes({ name: 'Anatoliy' });
});
});
it('should be triggered on save', function(done) {
User.create(function(err, user) {
addHooks('Update', done);
user.name = 'Hamburger';
user.save();
});
});
it('should update limited set of fields', function(done) {
User.create(function(err, user) {
User.beforeUpdate = function(next, data) {
data.should.have.keys('name', 'email');
done();
};
user.updateAttributes({ name: 1, email: 2 });
});
});
it('should not trigger after-hook on failed save', function(done) {
User.afterUpdate = function() {
should.fail('afterUpdate shouldn\'t be called');
};
User.create(function(err, user) {
const save = User.schema.adapter.save;
User.schema.adapter.save = function(modelName, id, cb) {
User.schema.adapter.save = save;
cb(new Error('Error'));
};
user.save(function(err) {
done();
});
});
});
});
describe('destroy', function() {
afterEach(removeHooks('Destroy'));
it('should be triggered on destroy', function(done) {
let hook = 'not called';
User.beforeDestroy = function(next) {
hook = 'called';
next();
};
User.afterDestroy = function(next) {
hook.should.eql('called');
next();
};
User.create(function(err, user) {
user.destroy(done);
});
});
it('should not trigger after-hook on failed destroy', function(done) {
const destroy = User.schema.adapter.destroy;
User.schema.adapter.destroy = function(modelName, id, cb) {
cb(new Error('error'));
};
User.afterDestroy = function() {
should.fail('afterDestroy shouldn\'t be called');
};
User.create(function(err, user) {
user.destroy(function(err) {
User.schema.adapter.destroy = destroy;
done();
});
});
});
});
describe('lifecycle', function() {
let life = [], user;
before(function(done) {
User.beforeSave = function(d) {life.push('beforeSave'); d();};
User.beforeCreate = function(d) {life.push('beforeCreate'); d();};
User.beforeUpdate = function(d) {life.push('beforeUpdate'); d();};
User.beforeDestroy = function(d) {life.push('beforeDestroy'); d();};
User.beforeValidate = function(d) {life.push('beforeValidate'); d();};
User.afterInitialize = function() {life.push('afterInitialize'); };
User.afterSave = function(d) {life.push('afterSave'); d();};
User.afterCreate = function(d) {life.push('afterCreate'); d();};
User.afterUpdate = function(d) {life.push('afterUpdate'); d();};
User.afterDestroy = function(d) {life.push('afterDestroy'); d();};
User.afterValidate = function(d) {life.push('afterValidate'); d();};
User.create(function(e, u) {
user = u;
life = [];
done();
});
});
beforeEach(function() {
life = [];
});
it('should describe create sequence', function(done) {
User.create(function() {
life.should.eql([
'afterInitialize',
'beforeValidate',
'afterValidate',
'beforeCreate',
'beforeSave',
'afterSave',
'afterCreate'
]);
done();
});
});
it('should describe new+save sequence', function(done) {
const u = new User;
u.save(function() {
life.should.eql([
'afterInitialize',
'beforeValidate',
'afterValidate',
'beforeCreate',
'beforeSave',
'afterSave',
'afterCreate'
]);
done();
});
});
it('should describe updateAttributes sequence', function(done) {
user.updateAttributes({ name: 'Antony' }, function() {
life.should.eql([
'beforeValidate',
'afterValidate',
'beforeSave',
'beforeUpdate',
'afterUpdate',
'afterSave',
]);
done();
});
});
it('should describe isValid sequence', function(done) {
should.not.exist(
user.constructor._validations,
'Expected user to have no validations, but she have');
user.isValid(function(valid) {
valid.should.be.true;
life.should.eql([
'beforeValidate',
'afterValidate'
]);
done();
});
});
it('should describe destroy sequence', function(done) {
user.destroy(function() {
life.should.eql([
'beforeDestroy',
'afterDestroy'
]);
done();
});
});
});
});
function addHooks(name, done) {
let called = false, random = String(Math.floor(Math.random() * 1000));
User['before' + name] = function(next, data) {
called = true;
data.email = random;
next();
};
User['after' + name] = function(next) {
(new Boolean(called)).should.equal(true);
this.email.should.equal(random);
done();
};
}
function removeHooks(name) {
return function() {
User['after' + name] = null;
User['before' + name] = null;
};
}