loopback/test/util/model-tests.js

var async = require('async');
var describe = require('./describe');
var loopback = require('../../');
var ACL = loopback.ACL;
var Change = loopback.Change;
var PersistedModel = loopback.PersistedModel;
var RemoteObjects = require('strong-remoting');

module.exports = function defineModelTestsWithDataSource(options) {

  describe('Model Tests', function() {

    var User;
    var dataSource;

    if (options.beforeEach) {
      beforeEach(options.beforeEach);
    }

    beforeEach(function() {
      var test = this;

      // setup a model / datasource
      dataSource = this.dataSource || loopback.createDataSource(options.dataSource);

      var extend = PersistedModel.extend;

      // create model hook
      PersistedModel.extend = function() {
        var extendedModel = extend.apply(PersistedModel, arguments);

        if (options.onDefine) {
          options.onDefine.call(test, extendedModel);
        }

        return extendedModel;
      };

      User = PersistedModel.extend('UtilUser', {
        id: { id: true, type: String, defaultFn: 'guid' },
        'first': String,
        'last': String,
        'age': Number,
        'password': String,
        'gender': String,
        'domain': String,
        'email': String
      }, {
        trackChanges: true
      });

      User.attachTo(dataSource);
      User.handleChangeError = function(err) {
        console.warn('WARNING: unhandled change-tracking error');
        console.warn(err);
      };
    });

    describe('Model.validatesPresenceOf(properties...)', function() {
      it('Require a model to include a property to be considered valid', function() {
        User.validatesPresenceOf('first', 'last', 'age');
        var joe = new User({first: 'joe'});
        assert(joe.isValid() === false, 'model should not validate');
        assert(joe.errors.last, 'should have a missing last error');
        assert(joe.errors.age, 'should have a missing age error');
      });
    });

    describe('Model.validatesLengthOf(property, options)', function() {
      it('Require a property length to be within a specified range', function() {
        User.validatesLengthOf('password', {min: 5, message: {min: 'Password is too short'}});
        var joe = new User({password: '1234'});
        assert(joe.isValid() === false, 'model should not be valid');
        assert(joe.errors.password, 'should have password error');
      });
    });

    describe('Model.validatesInclusionOf(property, options)', function() {
      it('Require a value for `property` to be in the specified array', function() {
        User.validatesInclusionOf('gender', {in: ['male', 'female']});
        var foo = new User({gender: 'bar'});
        assert(foo.isValid() === false, 'model should not be valid');
        assert(foo.errors.gender, 'should have gender error');
      });
    });

    describe('Model.validatesExclusionOf(property, options)', function() {
      it('Require a value for `property` to not exist in the specified array', function() {
        User.validatesExclusionOf('domain', {in: ['www', 'billing', 'admin']});
        var foo = new User({domain: 'www'});
        var bar = new User({domain: 'billing'});
        var bat = new User({domain: 'admin'});
        assert(foo.isValid() ===  false);
        assert(bar.isValid() === false);
        assert(bat.isValid() === false);
        assert(foo.errors.domain, 'model should have a domain error');
        assert(bat.errors.domain, 'model should have a domain error');
        assert(bat.errors.domain, 'model should have a domain error');
      });
    });

    describe('Model.validatesNumericalityOf(property, options)', function() {
      it('Require a value for `property` to be a specific type of `Number`', function() {
        User.validatesNumericalityOf('age', {int: true});
        var joe = new User({age: 10.2});
        assert(joe.isValid() === false);
        var bob = new User({age: 0});
        assert(bob.isValid() === true);
        assert(joe.errors.age, 'model should have an age error');
      });
    });

    describe('myModel.isValid()', function() {
      it('Validate the model instance', function() {
        User.validatesNumericalityOf('age', {int: true});
        var user = new User({first: 'joe', age: 'flarg'});
        var valid = user.isValid();
        assert(valid === false);
        assert(user.errors.age, 'model should have age error');
      });

      it('Asynchronously validate the model', function(done) {
        User.validatesNumericalityOf('age', {int: true});
        var user = new User({first: 'joe', age: 'flarg'});
        user.isValid(function(valid) {
          assert(valid === false);
          assert(user.errors.age, 'model should have age error');
          done();
        });
      });
    });

    describe('Model.create([data], [callback])', function() {
      it('Create an instance of Model with given data and save to the attached data source', function(done) {
        User.create({first: 'Joe', last: 'Bob'}, function(err, user) {
          assert(user instanceof User);
          done();
        });
      });
    });

    describe('model.save([options], [callback])', function() {
      it('Save an instance of a Model to the attached data source', function(done) {
        var joe = new User({first: 'Joe', last: 'Bob'});
        joe.save(function(err, user) {
          assert(user.id);
          assert(!err);
          assert(!user.errors);
          done();
        });
      });
    });

    describe('model.updateAttributes(data, [callback])', function() {
      it('Save specified attributes to the attached data source', function(done) {
        User.create({first: 'joe', age: 100}, function(err, user) {
          assert(!err);
          assert.equal(user.first, 'joe');

          user.updateAttributes({
            first: 'updatedFirst',
            last: 'updatedLast'
          }, function(err, updatedUser) {
            assert(!err);
            assert.equal(updatedUser.first, 'updatedFirst');
            assert.equal(updatedUser.last, 'updatedLast');
            assert.equal(updatedUser.age, 100);
            done();
          });
        });
      });
    });

    describe('Model.upsert(data, callback)', function() {
      it('Update when record with id=data.id found, insert otherwise', function(done) {
        User.upsert({first: 'joe', id: 7}, function(err, user) {
          assert(!err);
          assert.equal(user.first, 'joe');

          User.upsert({first: 'bob', id: 7}, function(err, updatedUser) {
            assert(!err);
            assert.equal(updatedUser.first, 'bob');
            done();
          });
        });
      });
    });

    describe('model.destroy([callback])', function() {
      it('Remove a model from the attached data source', function(done) {
        User.create({first: 'joe', last: 'bob'}, function(err, user) {
          User.findById(user.id, function(err, foundUser) {
            if (err) return done(err);
            assert.equal(user.id, foundUser.id);
            User.deleteById(foundUser.id, function(err) {
              if (err) return done(err);
              User.find({ where: { id: user.id } }, function(err, found) {
                if (err) return done(err);
                assert.equal(found.length, 0);
                done();
              });
            });
          });
        });
      });
    });

    describe('Model.deleteById(id, [callback])', function() {
      it('Delete a model instance from the attached data source', function(done) {
        User.create({first: 'joe', last: 'bob'}, function(err, user) {
          User.deleteById(user.id, function(err) {
            User.findById(user.id, function(err, notFound) {
              assert.equal(notFound, null);
              done();
            });
          });
        });
      });
    });

    describe('Model.findById(id, callback)', function() {
      it('Find an instance by id', function(done) {
        User.create({first: 'michael', last: 'jordan', id: 23}, function() {
          User.findById(23, function(err, user) {
            assert.equal(user.id, 23);
            assert.equal(user.first, 'michael');
            assert.equal(user.last, 'jordan');
            done();
          });
        });
      });
    });

    describe('Model.count([query], callback)', function() {
      it('Query count of Model instances in data source', function(done) {
        (new TaskEmitter())
          .task(User, 'create', {first: 'jill', age: 100})
          .task(User, 'create', {first: 'bob', age: 200})
          .task(User, 'create', {first: 'jan'})
          .task(User, 'create', {first: 'sam'})
          .task(User, 'create', {first: 'suzy'})
          .on('done', function() {
            User.count({age: {gt: 99}}, function(err, count) {
              assert.equal(count, 2);
              done();
            });
          });
      });
    });

  });

};