// Copyright IBM Corp. 2013,2016. All Rights Reserved. // Node module: loopback // This file is licensed under the MIT License. // License text available at https://opensource.org/licenses/MIT 'use strict'; var assert = require('assert'); var async = require('async'); var describe = require('./util/describe'); var loopback = require('../'); var ACL = loopback.ACL; var defineModelTestsWithDataSource = require('./util/model-tests'); var PersistedModel = loopback.PersistedModel; var Promise = require('bluebird'); var TaskEmitter = require('strong-task-emitter'); var request = require('supertest'); var expect = require('./helpers/expect'); describe('Model / PersistedModel', function() { defineModelTestsWithDataSource({ dataSource: { connector: loopback.Memory, }, }); describe('Model.validatesUniquenessOf(property, options)', function() { it('Ensure the value for `property` is unique', function(done) { var User = PersistedModel.extend('ValidatedUser', { 'first': String, 'last': String, 'age': Number, 'password': String, 'gender': String, 'domain': String, 'email': String, }); var dataSource = loopback.createDataSource({ connector: loopback.Memory, }); User.attachTo(dataSource); User.validatesUniquenessOf('email', {message: 'email is not unique'}); var joe = new User({email: 'joe@joe.com'}); var joe2 = new User({email: 'joe@joe.com'}); joe.save(function() { joe2.save(function(err) { assert(err, 'should get a validation error'); assert(joe2.errors.email, 'model should have email error'); done(); }); }); }); }); describe('Model.attachTo(dataSource)', function() { it('Attach a model to a [DataSource](#data-source)', function() { var MyModel = loopback.createModel('my-model', {name: String}); var dataSource = loopback.createDataSource({ connector: loopback.Memory, }); MyModel.attachTo(dataSource); MyModel.find(function(err, results) { assert(results.length === 0, 'should have data access methods after attaching to a data source'); }); }); }); }); describe.onServer('Remote Methods', function() { var User, Post, dataSource, app; beforeEach(function() { app = loopback({localRegistry: true, loadBuiltinModels: true}); app.set('remoting', {errorHandler: {debug: true, log: false}}); User = app.registry.createModel('user', { id: {id: true, type: String, defaultFn: 'guid'}, 'first': String, 'last': String, 'age': Number, 'password': String, 'gender': String, 'domain': String, 'email': String, }, { trackChanges: true, }); Post = app.registry.createModel('post', { id: {id: true, type: String, defaultFn: 'guid'}, title: String, content: String, }, { trackChanges: true, }); dataSource = app.dataSource('db', {connector: 'memory'}); app.model(User, {dataSource: 'db'}); app.model(Post, {dataSource: 'db'}); User.hasMany(Post); User.login = function(username, password, fn) { if (username === 'foo' && password === 'bar') { fn(null, 123); } else { throw new Error('bad username and password!'); } }; User.remoteMethod('login', { accepts: [ {arg: 'username', type: 'string', required: true}, {arg: 'password', type: 'string', required: true}, ], returns: {arg: 'sessionId', type: 'any', root: true}, http: {path: '/sign-in', verb: 'get'}, }); app.use(loopback.rest()); }); describe('Model.create(data, callback)', function() { it('creates model', function(done) { var anObject = {first: 'June'}; request(app) .post('/users') // sends an object .send(anObject) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); expect(res.body).to.have.property('id'); expect(res.body).to.have.property('first', 'June'); done(); }); }); // batch create must be tested with a remote request because there are // coercion being done on strong-remoting side it('creates array of models', function(done) { var arrayOfObjects = [ {first: 'John'}, {first: 'Jane'}, ]; request(app) .post('/users') // sends an array of objects .send(arrayOfObjects) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); expect(res.body.length).to.eql(2); expect(res.body).to.have.deep.property('[0].first', 'John'); expect(res.body).to.have.deep.property('[1].first', 'Jane'); done(); }); }); it('creates related models', function(done) { User.create({first: 'Bob'}, function(err, res) { expect(res).to.have.property('id'); var aPost = {title: 'A story', content: 'Once upon a time'}; request(app) .post('/users/' + res.id + '/posts') .send(aPost) .expect('Content-Type', /json/) .expect(200) .end(function(err, result) { if (err) return done(err); expect(result.body).to.have.property('id'); expect(result.body).to.have.property('title', aPost.title); expect(result.body).to.have.property('content', aPost.content); done(); }); }); }); it('creates array of hasMany models', function(done) { User.create({first: 'Bob'}, function(err, res) { expect(res).to.have.property('id'); var twoPosts = [ {title: 'One story', content: 'Content #1'}, {title: 'Two story', content: 'Content #2'}, ]; request(app) .post('/users/' + res.id + '/posts') .send(twoPosts) .expect('Content-Type', /json/) .expect(200) .end(function(err, result) { if (err) return done(err); expect(result.body.length).to.eql(2); expect(result.body).to.have.deep.property('[0].title', 'One story'); expect(result.body).to.have.deep.property('[1].title', 'Two story'); done(); }); }); }); it('rejects array of obj input for hasOne relation', function(done) { var Friend = app.registry.createModel('friend', {name: String}); app.model(Friend, {dataSource: 'db'}); User.hasOne(Friend); User.create({first: 'Bob'}, function(err, res) { expect(res).to.have.property('id'); var twoFriends = [ {name: 'bob'}, {name: 'rob'}, ]; request(app) .post('/users/' + res.id + '/friend') .send(twoFriends) .expect('Content-Type', /json/) .expect(400) .end(function(err, result) { if (err) return done(err); var resError = result.body.error; expect(resError.message).to.match(/value(.*?)not(.*?)object(\.?)/i); done(); }); }); }); }); // destoryAll is not exposed as a remoteMethod by default describe('Model.destroyAll(callback)', function() { it('Delete all Model instances from data source', function(done) { (new TaskEmitter()) .task(User, 'create', {first: 'jill'}) .task(User, 'create', {first: 'bob'}) .task(User, 'create', {first: 'jan'}) .task(User, 'create', {first: 'sam'}) .task(User, 'create', {first: 'suzy'}) .on('done', function() { User.count(function(err, count) { User.destroyAll(function() { User.count(function(err, count) { assert.equal(count, 0); done(); }); }); }); }); }); }); describe('Model.upsertWithWhere(where, data, callback)', function() { it('Updates when a Model instance is retreived from data source', function(done) { var taskEmitter = new TaskEmitter(); taskEmitter .task(User, 'create', {first: 'jill', second: 'pill'}) .task(User, 'create', {first: 'bob', second: 'sob'}) .on('done', function() { User.upsertWithWhere({second: 'pill'}, {second: 'jones'}, function(err, user) { if (err) return done(err); var id = user.id; User.findById(id, function(err, user) { if (err) return done(err); assert.equal(user.second, 'jones'); done(); }); }); }); }); it('Creates when no Model instance is retreived from data source', function(done) { var taskEmitter = new TaskEmitter(); taskEmitter .task(User, 'create', {first: 'simon', second: 'somers'}) .on('done', function() { User.upsertWithWhere({first: 'somers'}, {first: 'Simon'}, function(err, user) { if (err) return done(err); var id = user.id; User.findById(id, function(err, user) { if (err) return done(err); assert.equal(user.first, 'Simon'); done(); }); }); }); }); }); describe('Example Remote Method', function() { it('Call the method using HTTP / REST', function(done) { request(app) .get('/users/sign-in?username=foo&password=bar') .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); assert.equal(res.body, 123); done(); }); }); it('Converts null result of findById to 404 Not Found', function(done) { request(app) .get('/users/not-found') .expect(404) .end(function(err, res) { if (err) return done(err); var errorResponse = res.body.error; assert(errorResponse); assert.equal(errorResponse.code, 'MODEL_NOT_FOUND'); done(); }); }); it('Call the findById with filter.fields using HTTP / REST', function(done) { request(app) .post('/users') .send({first: 'x', last: 'y'}) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); var userId = res.body.id; assert(userId); request(app) .get('/users/' + userId + '?filter[fields]=first') .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); assert.equal(res.body.first, 'x', 'first should be x'); assert(res.body.last === undefined, 'last should not be present'); done(); }); }); }); it('Call the findById with filter.include using HTTP / REST', function(done) { request(app) .post('/users') .send({first: 'x', last: 'y'}) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); var userId = res.body.id; assert(userId); request(app) .post('/users/' + userId + '/posts') .send({title: 'T1', content: 'C1'}) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); var post = res.body; request(app) .get('/users/' + userId + '?filter[include]=posts') .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); assert.equal(res.body.first, 'x', 'first should be x'); assert.equal(res.body.last, 'y', 'last should be y'); assert.deepEqual(post, res.body.posts[0]); done(); }); }); }); }); }); describe('Model.beforeRemote(name, fn)', function() { it('Run a function before a remote method is called by a client', function(done) { var hookCalled = false; User.beforeRemote('create', function(ctx, user, next) { hookCalled = true; next(); }); // invoke save request(app) .post('/users') .send({data: {first: 'foo', last: 'bar'}}) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); assert(hookCalled, 'hook wasnt called'); done(); }); }); it('Does not stop the hook chain after returning a promise', function(done) { var hooksCalled = []; User.beforeRemote('create', function() { hooksCalled.push('first'); return Promise.resolve(); }); User.beforeRemote('create', function(ctx, user, next) { hooksCalled.push('second'); next(); }); // invoke save request(app) .post('/users') .send({data: {first: 'foo', last: 'bar'}}) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); expect(hooksCalled).to.eql(['first', 'second']); done(); }); }); }); describe('Model.afterRemote(name, fn)', function() { it('Run a function after a remote method is called by a client', function(done) { var beforeCalled = false; var afterCalled = false; User.beforeRemote('create', function(ctx, user, next) { assert(!afterCalled); beforeCalled = true; next(); }); User.afterRemote('create', function(ctx, user, next) { assert(beforeCalled); afterCalled = true; next(); }); // invoke save request(app) .post('/users') .send({data: {first: 'foo', last: 'bar'}}) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); assert(beforeCalled, 'before hook was not called'); assert(afterCalled, 'after hook was not called'); done(); }); }); }); describe('Model.afterRemoteError(name, fn)', function() { it('runs the function when method fails', function(done) { var actualError = 'hook not called'; User.afterRemoteError('login', function(ctx, next) { actualError = ctx.error; next(); }); request(app).get('/users/sign-in?username=bob&password=123') .end(function(err, res) { if (err) return done(err); expect(actualError) .to.have.property('message', 'bad username and password!'); done(); }); }); }); describe('Remote Method invoking context', function() { describe('ctx.req', function() { it('The express ServerRequest object', function(done) { var hookCalled = false; User.beforeRemote('create', function(ctx, user, next) { hookCalled = true; assert(ctx.req); assert(ctx.req.url); assert(ctx.req.method); assert(ctx.res); assert(ctx.res.write); assert(ctx.res.end); next(); }); // invoke save request(app) .post('/users') .send({data: {first: 'foo', last: 'bar'}}) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); assert(hookCalled); done(); }); }); }); describe('ctx.res', function() { it('The express ServerResponse object', function(done) { var hookCalled = false; User.beforeRemote('create', function(ctx, user, next) { hookCalled = true; assert(ctx.req); assert(ctx.req.url); assert(ctx.req.method); assert(ctx.res); assert(ctx.res.write); assert(ctx.res.end); next(); }); // invoke save request(app) .post('/users') .send({data: {first: 'foo', last: 'bar'}}) .expect('Content-Type', /json/) .expect(200) .end(function(err, res) { if (err) return done(err); assert(hookCalled); done(); }); }); }); }); describe('Model.hasMany(Model)', function() { it('Define a one to many relationship', function(done) { var Book = dataSource.createModel('book', {title: String, author: String}); var Chapter = dataSource.createModel('chapter', {title: String}); // by referencing model Book.hasMany(Chapter); Book.create({title: 'Into the Wild', author: 'Jon Krakauer'}, function(err, book) { // using 'chapters' scope for build: var c = book.chapters.build({title: 'Chapter 1'}); book.chapters.create({title: 'Chapter 2'}, function() { c.save(function() { Chapter.count({bookId: book.id}, function(err, count) { assert.equal(count, 2); book.chapters({where: {title: 'Chapter 1'}}, function(err, chapters) { assert.equal(chapters.length, 1); assert.equal(chapters[0].title, 'Chapter 1'); done(); }); }); }); }); }); }); }); describe('Model.properties', function() { it('Normalized properties passed in originally by loopback.createModel()', function() { var props = { s: String, n: {type: 'Number'}, o: {type: 'String', min: 10, max: 100}, d: Date, g: loopback.GeoPoint, }; var MyModel = loopback.createModel('foo', props); Object.keys(MyModel.definition.properties).forEach(function(key) { var p = MyModel.definition.properties[key]; var o = MyModel.definition.properties[key]; assert(p); assert(o); assert(typeof p.type === 'function'); if (typeof o === 'function') { // the normalized property // should match the given property assert( p.type.name === o.name || p.type.name === o ); } }); }); }); describe('Model.extend()', function() { it('Create a new model by extending an existing model', function() { var User = loopback.PersistedModel.extend('test-user', { email: String, }); User.foo = function() { return 'bar'; }; User.prototype.bar = function() { return 'foo'; }; var MyUser = User.extend('my-user', { a: String, b: String, }); assert.equal(MyUser.prototype.bar, User.prototype.bar); assert.equal(MyUser.foo, User.foo); var user = new MyUser({ email: 'foo@bar.com', a: 'foo', b: 'bar', }); assert.equal(user.email, 'foo@bar.com'); assert.equal(user.a, 'foo'); assert.equal(user.b, 'bar'); }); }); describe('Model.extend() events', function() { it('create isolated emitters for subclasses', function() { var User1 = loopback.createModel('User1', { 'first': String, 'last': String, }); var User2 = loopback.createModel('User2', { 'name': String, }); var user1Triggered = false; User1.once('x', function(event) { user1Triggered = true; }); var user2Triggered = false; User2.once('x', function(event) { user2Triggered = true; }); assert(User1.once !== User2.once); assert(User1.once !== loopback.Model.once); User1.emit('x', User1); assert(user1Triggered); assert(!user2Triggered); }); }); describe('Model.checkAccessTypeForMethod(remoteMethod)', function() { shouldReturn('create', ACL.WRITE); shouldReturn('updateOrCreate', ACL.WRITE); shouldReturn('upsertWithWhere', ACL.WRITE); shouldReturn('upsert', ACL.WRITE); shouldReturn('exists', ACL.READ); shouldReturn('findById', ACL.READ); shouldReturn('find', ACL.READ); shouldReturn('findOne', ACL.READ); shouldReturn('destroyById', ACL.WRITE); shouldReturn('deleteById', ACL.WRITE); shouldReturn('removeById', ACL.WRITE); shouldReturn('count', ACL.READ); shouldReturn('unkown-model-method', ACL.EXECUTE); function shouldReturn(methodName, expectedAccessType) { describe(methodName, function() { it('should return ' + expectedAccessType, function() { var remoteMethod = {name: methodName}; assert.equal( User._getAccessTypeForMethod(remoteMethod), expectedAccessType ); }); }); } }); describe('Model.getChangeModel()', function() { it('Get the Change Model', function() { var UserChange = User.getChangeModel(); var change = new UserChange(); assert(change instanceof app.registry.getModel('Change')); }); }); describe('Model.getSourceId(callback)', function() { it('Get the Source Id', function(done) { User.getSourceId(function(err, id) { assert.equal('memory-user', id); done(); }); }); }); describe('Model.checkpoint(callback)', function() { it('Create a checkpoint', function(done) { var Checkpoint = User.getChangeModel().getCheckpointModel(); var tasks = [ getCurrentCheckpoint, checkpoint, ]; var result, current; async.series(tasks, function(err) { if (err) return done(err); assert.equal(result, current + 1); done(); }); function getCurrentCheckpoint(cb) { Checkpoint.current(function(err, cp) { current = cp; cb(err); }); } function checkpoint(cb) { User.checkpoint(function(err, cp) { result = cp.seq; cb(err); }); } }); }); describe('Model._getACLModel()', function() { it('should return the subclass of ACL', function() { var Model = require('../').Model; var originalValue = Model._ACL(); var acl = ACL.extend('acl'); Model._ACL(null); // Reset the ACL class for the base model var model = Model._ACL(); Model._ACL(originalValue); // Reset the value back assert.equal(model, acl); }); }); describe('PersistedModel remote methods', function() { it('includes all aliases', function() { var app = loopback(); var model = PersistedModel.extend('PersistedModelForAliases'); app.dataSource('db', {connector: 'memory'}); app.model(model, {dataSource: 'db'}); // this code is used by loopback-sdk-angular codegen var metadata = app.handler('rest') .adapter .getClasses() .filter(function(c) { return c.name === model.modelName; })[0]; var methodNames = []; metadata.methods.forEach(function(method) { methodNames.push(method.name); var aliases = method.sharedMethod.aliases; if (method.name.indexOf('prototype.') === 0) { aliases = aliases.map(function(alias) { return 'prototype.' + alias; }); } methodNames = methodNames.concat(aliases || []); }); expect(methodNames).to.have.members([ // NOTE(bajtos) These three methods are disabled by default // Because all tests share the same global registry model // and one of the tests was enabling remoting of "destroyAll", // this test was seeing this method (with all aliases) as public // 'destroyAll', 'deleteAll', 'remove', 'create', 'upsert', 'updateOrCreate', 'patchOrCreate', 'upsertWithWhere', 'patchOrCreateWithWhere', 'exists', 'findById', 'replaceById', 'replaceOrCreate', 'find', 'findOne', 'updateAll', 'update', 'deleteById', 'destroyById', 'removeById', 'count', 'prototype.patchAttributes', 'prototype.updateAttributes', 'createChangeStream', ]); }); it('emits a `remoteMethodDisabled` event', function() { var app = loopback(); var model = PersistedModel.extend('TestModelForDisablingRemoteMethod'); app.dataSource('db', {connector: 'memory'}); app.model(model, {dataSource: 'db'}); var callbackSpy = require('sinon').spy(); var TestModel = app.models.TestModelForDisablingRemoteMethod; TestModel.on('remoteMethodDisabled', callbackSpy); TestModel.disableRemoteMethod('findOne', true); expect(callbackSpy).to.have.been.calledWith(TestModel.sharedClass, 'findOne'); }); it('emits a `remoteMethodDisabled` event from disableRemoteMethodByName', function() { var app = loopback(); var model = PersistedModel.extend('TestModelForDisablingRemoteMethod'); app.dataSource('db', {connector: 'memory'}); app.model(model, {dataSource: 'db'}); var callbackSpy = require('sinon').spy(); var TestModel = app.models.TestModelForDisablingRemoteMethod; TestModel.on('remoteMethodDisabled', callbackSpy); TestModel.disableRemoteMethodByName('findOne'); expect(callbackSpy).to.have.been.calledWith(TestModel.sharedClass, 'findOne'); }); it('emits a `remoteMethodAdded` event', function() { var app = loopback(); app.dataSource('db', {connector: 'memory'}); var User = app.registry.getModel('User'); app.model(User, {dataSource: 'db'}); var Token = app.registry.getModel('AccessToken'); app.model(Token, {dataSource: 'db'}); var callbackSpy = require('sinon').spy(); var TestModel = app.models.User; TestModel.on('remoteMethodAdded', callbackSpy); TestModel.nestRemoting('accessTokens'); expect(callbackSpy).to.have.been.calledWith(TestModel.sharedClass); }); }); it('emits a `remoteMethodAdded` event from remoteMethod', function() { var app = loopback(); var model = PersistedModel.extend('TestModelForAddingRemoteMethod'); app.dataSource('db', {connector: 'memory'}); app.model(model, {dataSource: 'db'}); var callbackSpy = require('sinon').spy(); var TestModel = app.models.TestModelForAddingRemoteMethod; TestModel.on('remoteMethodAdded', callbackSpy); TestModel.remoteMethod('getTest', { accepts: {arg: 'options', type: 'object', http: 'optionsFromRequest'}, returns: {arg: 'test', type: 'object'}, http: {verb: 'GET', path: '/test'}, }); expect(callbackSpy).to.have.been.calledWith(TestModel.sharedClass); }); describe('Model.getApp(cb)', function() { var app, TestModel; beforeEach(function setup() { app = loopback(); TestModel = loopback.createModel('TestModelForGetApp'); // unique name app.dataSource('db', {connector: 'memory'}); }); it('calls the callback when already attached', function(done) { app.model(TestModel, {dataSource: 'db'}); TestModel.getApp(function(err, a) { if (err) return done(err); expect(a).to.equal(app); done(); }); // fails on time-out when not implemented correctly }); it('calls the callback after attached', function(done) { TestModel.getApp(function(err, a) { if (err) return done(err); expect(a).to.equal(app); done(); }); app.model(TestModel, {dataSource: 'db'}); // fails on time-out when not implemented correctly }); }); describe('Model.createOptionsFromRemotingContext', function() { var app, TestModel, accessToken, userId, actualOptions; before(setupAppAndRequest); before(createUserAndAccessToken); it('sets empty options.accessToken for anonymous requests', function(done) { request(app).get('/TestModels/saveOptions') .expect(204, function(err) { if (err) return done(err); expect(actualOptions).to.include({accessToken: null}); done(); }); }); it('sets options.accessToken for authorized requests', function(done) { request(app).get('/TestModels/saveOptions') .set('Authorization', accessToken.id) .expect(204, function(err) { if (err) return done(err); expect(actualOptions).to.have.property('accessToken'); expect(actualOptions.accessToken.toObject()) .to.eql(accessToken.toObject()); done(); }); }); it('allows "beforeRemote" hooks to contribute options', function(done) { TestModel.beforeRemote('saveOptions', function(ctx, unused, next) { ctx.args.options.hooked = true; next(); }); request(app).get('/TestModels/saveOptions') .expect(204, function(err) { if (err) return done(err); expect(actualOptions).to.have.property('hooked', true); done(); }); }); it('sets empty options.accessToken for requests coming from websocket/primus adapters', function() { const primusContext = {}; const opts = TestModel.createOptionsFromRemotingContext(primusContext); expect(opts).to.have.property('accessToken', null); }); it('allows apps to add options before remoting hooks', function(done) { TestModel.createOptionsFromRemotingContext = function(ctx) { return {hooks: []}; }; TestModel.beforeRemote('saveOptions', function(ctx, unused, next) { ctx.args.options.hooks.push('beforeRemote'); next(); }); // In real apps, this code can live in a component or in a boot script app.remotes().phases .addBefore('invoke', 'options-from-request') .use(function(ctx, next) { ctx.args.options.hooks.push('custom'); next(); }); request(app).get('/TestModels/saveOptions') .expect(204, function(err) { if (err) return done(err); expect(actualOptions.hooks).to.eql(['custom', 'beforeRemote']); done(); }); }); function setupAppAndRequest() { app = loopback({localRegistry: true, loadBuiltinModels: true}); app.dataSource('db', {connector: 'memory'}); TestModel = app.registry.createModel('TestModel', {base: 'Model'}); TestModel.saveOptions = function(options, cb) { actualOptions = options; cb(); }; TestModel.remoteMethod('saveOptions', { accepts: {arg: 'options', type: 'object', http: 'optionsFromRequest'}, http: {verb: 'GET', path: '/saveOptions'}, }); app.model(TestModel, {dataSource: null}); app.enableAuth({dataSource: 'db'}); app.use(loopback.token()); app.use(loopback.rest()); } function createUserAndAccessToken() { var CREDENTIALS = {email: 'context@example.com', password: 'pass'}; var User = app.registry.getModel('User'); return User.create(CREDENTIALS) .then(function(u) { return User.login(CREDENTIALS); }).then(function(token) { accessToken = token; userId = token.userId; }); } }); });