// This test written in mocha+should.js var should = require('./init.js'); var jdb = require('../'); var ModelBuilder = jdb.ModelBuilder; var DataSource = jdb.DataSource; var Memory = require('../lib/connectors/memory'); var modelBuilder = new ModelBuilder(); var mixins = modelBuilder.mixins; function timestamps(Model, options) { Model.defineProperty('createdAt', { type: Date }); Model.defineProperty('updatedAt', { type: Date }); var originalBeforeSave = Model.beforeSave; Model.beforeSave = function(next, data) { Model.applyTimestamps(data, this.isNewRecord()); if (data.createdAt) { this.createdAt = data.createdAt; } if (data.updatedAt) { this.updatedAt = data.updatedAt; } if (originalBeforeSave) { originalBeforeSave.apply(this, arguments); } else { next(); } }; Model.applyTimestamps = function(data, creation) { data.updatedAt = new Date(); if (creation) { data.createdAt = data.updatedAt; } }; } mixins.define('TimeStamp', timestamps); describe('Model class', function () { it('should define mixins', function() { mixins.define('Example', function(Model, options) { Model.prototype.example = function() { return options; }; }); mixins.define('Demo', function(Model, options) { Model.demoMixin = options.value; }); mixins.define('Multi', function(Model, options) { Model.multiMixin = Model.multiMixin || {}; Model.multiMixin[options.key] = options.value; }); }); it('should apply a mixin class', function() { var Address = modelBuilder.define('Address', { street: { type: 'string', required: true }, city: { type: 'string', required: true } }); var memory = new DataSource('mem', {connector: Memory}, modelBuilder); var Item = memory.createModel('Item', { name: 'string' }, { mixins: { Address: true } }); var properties = Item.definition.properties; properties.street.should.eql({ type: String, required: true }); properties.city.should.eql({ type: String, required: true }); }); it('should apply mixins', function(done) { var memory = new DataSource('mem', {connector: Memory}, modelBuilder); var Item = memory.createModel('Item', { name: 'string' }, { mixins: { TimeStamp: true, Demo: { value: true }, Multi: [ { key: 'foo', value: 'bar' }, { key: 'fox', value: 'baz' } ] } }); Item.mixin('Example', { foo: 'bar' }); Item.demoMixin.should.be.true; Item.multiMixin.foo.should.equal('bar'); Item.multiMixin.fox.should.equal('baz'); var properties = Item.definition.properties; properties.createdAt.should.eql({ type: Date }); properties.updatedAt.should.eql({ type: Date }); Item.create({ name: 'Item 1' }, function(err, inst) { inst.createdAt.should.be.a.date; inst.updatedAt.should.be.a.date; inst.example().should.eql({ foo: 'bar' }); done(); }); }); describe('#mixin()', function () { var Person, Author, Address; beforeEach(function () { Address = modelBuilder.define('Address', { street: { type: 'string', required: true }, city: { type: 'string', required: true } }); var memory = new DataSource('mem', {connector: Memory}, modelBuilder); Person = memory.createModel('Person', { name: 'string' }); Author = memory.createModel('Author', { name: 'string' }); }); it('should register mixin class into _mixins', function () { Person.mixin(Address); Person._mixins.should.containEql(Address); }); it('should NOT share mixins registry', function () { Person.mixin(Address); Author._mixins.should.not.containEql(Address); }); it('should able to mixin same class', function () { Person.mixin(Address); Author.mixin(Address); Author._mixins.should.containEql(Address); }); }); });