// This test written in mocha+should.js var should = require('./init.js'); var db, User, Post, Passport, City, Street, Building; var nbSchemaRequests = 0; describe('include', function() { before(setup); it('should fetch belongsTo relation', function(done) { Passport.all({include: 'owner'}, function (err, passports) { passports.length.should.be.ok; passports.forEach(function(p) { p.__cachedRelations.should.have.property('owner'); var owner = p.__cachedRelations.owner; if (!p.ownerId) { should.not.exist(owner); } else { should.exist(owner); owner.id.should.equal(p.ownerId); } }); done(); }); }); it('should fetch hasMany relation', function(done) { User.all({include: 'posts'}, function (err, users) { should.not.exist(err); should.exist(users); users.length.should.be.ok; users.forEach(function(u) { u.__cachedRelations.should.have.property('posts'); u.__cachedRelations.posts.forEach(function(p) { p.userId.should.equal(u.id); }); }); done(); }); }); it('should fetch Passport - Owner - Posts', function(done) { Passport.all({include: {owner: 'posts'}}, function(err, passports) { should.not.exist(err); should.exist(passports); passports.length.should.be.ok; passports.forEach(function(p) { p.__cachedRelations.should.have.property('owner'); var user = p.__cachedRelations.owner; if (!p.ownerId) { should.not.exist(user); } else { should.exist(user); user.id.should.equal(p.ownerId); user.__cachedRelations.should.have.property('posts'); user.__cachedRelations.posts.forEach(function(pp) { pp.userId.should.equal(user.id); }); } }); done(); }); }); it('should fetch Passports - User - Posts - User', function(done) { Passport.all({ include: {owner: {posts: 'author'}} }, function(err, passports) { should.not.exist(err); should.exist(passports); passports.length.should.be.ok; passports.forEach(function(p) { p.__cachedRelations.should.have.property('owner'); var user = p.__cachedRelations.owner; if (!p.ownerId) { should.not.exist(user); } else { should.exist(user); user.id.should.equal(p.ownerId); user.__cachedRelations.should.have.property('posts'); user.__cachedRelations.posts.forEach(function(pp) { pp.userId.should.equal(user.id); pp.__cachedRelations.should.have.property('author'); var author = pp.__cachedRelations.author; author.id.should.equal(user.id); }); } }); done(); }); }); it('should fetch User - Posts AND Passports', function(done) { User.all({include: ['posts', 'passports']}, function(err, users) { should.not.exist(err); should.exist(users); users.length.should.be.ok; users.forEach(function(user) { user.__cachedRelations.should.have.property('posts'); user.__cachedRelations.should.have.property('passports'); user.__cachedRelations.posts.forEach(function(p) { p.userId.should.equal(user.id); }); user.__cachedRelations.passports.forEach(function(pp) { pp.ownerId.should.equal(user.id); }); }); done(); }); }); }); function setup(done) { db = getSchema(); City = db.define('City'); Street = db.define('Street'); Building = db.define('Building'); User = db.define('User', { name: String, age: Number }); Passport = db.define('Passport', { number: String }); Post = db.define('Post', { title: String }); Passport.belongsTo('owner', {model: User}); User.hasMany('passports', {foreignKey: 'ownerId'}); User.hasMany('posts', {foreignKey: 'userId'}); Post.belongsTo('author', {model: User, foreignKey: 'userId'}); db.automigrate(function() { var createdUsers = []; var createdPassports = []; var createdPosts = []; createUsers(); function createUsers() { clearAndCreate( User, [ {name: 'User A', age: 21}, {name: 'User B', age: 22}, {name: 'User C', age: 23}, {name: 'User D', age: 24}, {name: 'User E', age: 25} ], function(items) { createdUsers = items; createPassports(); } ); } function createPassports() { clearAndCreate( Passport, [ {number: '1', ownerId: createdUsers[0].id}, {number: '2', ownerId: createdUsers[1].id}, {number: '3'} ], function(items) { createdPassports = items; createPosts(); } ); } function createPosts() { clearAndCreate( Post, [ {title: 'Post A', userId: createdUsers[0].id}, {title: 'Post B', userId: createdUsers[0].id}, {title: 'Post C', userId: createdUsers[0].id}, {title: 'Post D', userId: createdUsers[1].id}, {title: 'Post E'} ], function(items) { createdPosts = items; done(); } ); } }); } function clearAndCreate(model, data, callback) { var createdItems = []; model.destroyAll(function () { nextItem(null, null); }); var itemIndex = 0; function nextItem(err, lastItem) { if (lastItem !== null) { createdItems.push(lastItem); } if (itemIndex >= data.length) { callback(createdItems); return; } model.create(data[itemIndex], nextItem); itemIndex++; } }