loopback-datasource-juggler/test/mixins.test.js

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

  });

});