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

1734 lines
58 KiB
JavaScript

// Copyright IBM Corp. 2013,2015. All Rights Reserved.
// Node module: loopback-datasource-juggler
// This file is licensed under the MIT License.
// License text available at https://opensource.org/licenses/MIT
'use strict';
/* global getSchema:false, connectorCapabilities:false */
const assert = require('assert');
const async = require('async');
const bdd = require('./helpers/bdd-if');
const should = require('./init.js');
const DataSource = require('../').DataSource;
let db, User, Profile, AccessToken, Post, Passport, City, Street, Building, Assembly, Part;
const knownUsers = ['User A', 'User B', 'User C', 'User D', 'User E'];
const knownPassports = ['1', '2', '3', '4'];
const knownPosts = ['Post A', 'Post B', 'Post C', 'Post D', 'Post E'];
const knownProfiles = ['Profile A', 'Profile B', 'Profile Z'];
describe('include', function() {
before(setup);
it('should fetch belongsTo relation', function(done) {
Passport.find({include: 'owner'}, function(err, passports) {
passports.length.should.be.ok;
passports.forEach(function(p) {
p.__cachedRelations.should.have.property('owner');
// The relation should be promoted as the 'owner' property
p.should.have.property('owner');
// The __cachedRelations should be removed from json output
p.toJSON().should.not.have.property('__cachedRelations');
const owner = p.__cachedRelations.owner;
if (!p.ownerId) {
should.not.exist(owner);
} else {
should.exist(owner);
owner.id.should.eql(p.ownerId);
}
});
done();
});
});
it('does not return included item if FK is excluded', function(done) {
Passport.find({include: 'owner', fields: 'number'}, function(err, passports) {
if (err) return done(err);
const owner = passports[0].toJSON().owner;
should.not.exist(owner);
done();
});
});
it('should fetch hasMany relation', function(done) {
User.find({include: 'posts'}, function(err, users) {
should.not.exist(err);
should.exist(users);
users.length.should.be.ok;
users.forEach(function(u) {
// The relation should be promoted as the 'owner' property
u.should.have.property('posts');
// The __cachedRelations should be removed from json output
u.toJSON().should.not.have.property('__cachedRelations');
u.__cachedRelations.should.have.property('posts');
u.__cachedRelations.posts.forEach(function(p) {
// FIXME There are cases that p.userId is string
p.userId.toString().should.eql(u.id.toString());
});
});
done();
});
});
it('should report errors if the PK is excluded', function(done) {
User.find({include: 'posts', fields: 'posts'}, function(err) {
should.exist(err);
err.message.should.match(/ID property "id" is missing/);
done();
});
});
it('should not have changed the __strict flag of the model', function(done) {
const originalStrict = User.definition.settings.strict;
User.definition.settings.strict = true; // Change to test regression for issue #1252
const finish = (err) => {
// Restore original user strict property
User.definition.settings.strict = originalStrict;
done(err);
};
User.find({include: 'posts'}, function(err, users) {
if (err) return finish(err);
users.forEach(user => {
user.should.have.property('__strict', true); // we changed it
});
finish();
});
});
bdd.itIf(connectorCapabilities.cloudantCompatible !== false,
'should not save in db included models, in query returned models',
function(done) {
const originalStrict = User.definition.settings.strict;
User.definition.settings.strict = true; // Change to test regression for issue #1252
const finish = (err) => {
// Restore original user strict property
User.definition.settings.strict = originalStrict;
done(err);
};
User.findOne({where: {name: 'User A'}, include: 'posts'}, function(err, user) {
if (err) return finish(err);
if (!user) return finish(new Error('User Not found to check relation not saved'));
user.save(function(err) { // save the returned user
if (err) return finish(err);
// should not store in db the posts
const dsName = User.dataSource.name;
if (dsName === 'memory') {
JSON.parse(User.dataSource.adapter.cache.User[1]).should.not.have.property('posts');
finish();
} else if (dsName === 'mongodb') { // Check native mongodb connector
// get hold of native mongodb collection
const dbCollection = User.dataSource.connector.collection(User.modelName);
dbCollection.findOne({_id: user.id})
.then(function(foundUser) {
if (!foundUser) {
finish(new Error('User not found to check posts not saved'));
}
foundUser.should.not.have.property('posts');
finish();
})
.catch(finish);
} else { // TODO make native checks for other connectors as well
finish();
}
});
});
});
it('should fetch Passport - Owner - Posts', function(done) {
Passport.find({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');
// The relation should be promoted as the 'owner' property
p.should.have.property('owner');
// The __cachedRelations should be removed from json output
p.toJSON().should.not.have.property('__cachedRelations');
const user = p.__cachedRelations.owner;
if (!p.ownerId) {
should.not.exist(user);
} else {
should.exist(user);
user.id.should.eql(p.ownerId);
user.__cachedRelations.should.have.property('posts');
user.should.have.property('posts');
user.toJSON().should.have.property('posts').and.be.an.Array;
user.__cachedRelations.posts.forEach(function(pp) {
// FIXME There are cases that pp.userId is string
pp.userId.toString().should.eql(user.id.toString());
});
}
});
done();
});
});
it('should fetch Passport - Owner - empty Posts', function(done) {
Passport.findOne({where: {number: '4'}, include: {owner: 'posts'}}, function(err, passport) {
should.not.exist(err);
should.exist(passport);
passport.__cachedRelations.should.have.property('owner');
// The relation should be promoted as the 'owner' property
passport.should.have.property('owner');
// The __cachedRelations should be removed from json output
passport.toJSON().should.not.have.property('__cachedRelations');
const user = passport.__cachedRelations.owner;
should.exist(user);
user.id.should.eql(passport.ownerId);
user.__cachedRelations.should.have.property('posts');
user.should.have.property('posts');
user.toJSON().should.have.property('posts').and.be.an.Array().with
.length(0);
done();
});
});
it('should fetch Passport - Owner - Posts - alternate syntax', function(done) {
Passport.find({include: {owner: {relation: 'posts'}}}, function(err, passports) {
should.not.exist(err);
should.exist(passports);
passports.length.should.be.ok;
let posts;
if (connectorCapabilities.adhocSort !== false) {
posts = passports[0].owner().posts();
posts.should.have.length(3);
} else {
if (passports[0].owner()) {
posts = passports[0].owner().posts();
posts.length.should.be.belowOrEqual(3);
}
}
done();
});
});
it('should fetch Passports - User - Posts - User', function(done) {
Passport.find({
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');
const user = p.__cachedRelations.owner;
if (!p.ownerId) {
should.not.exist(user);
} else {
should.exist(user);
user.id.should.eql(p.ownerId);
user.__cachedRelations.should.have.property('posts');
user.__cachedRelations.posts.forEach(function(pp) {
pp.should.have.property('id');
// FIXME There are cases that pp.userId is string
pp.userId.toString().should.eql(user.id.toString());
pp.should.have.property('author');
pp.__cachedRelations.should.have.property('author');
const author = pp.__cachedRelations.author;
author.id.should.eql(user.id);
});
}
});
done();
});
});
it('should fetch Passports with include scope on Posts', function(done) {
Passport.find({
include: {owner: {relation: 'posts', scope: {
fields: ['title'], include: ['author'],
order: 'title DESC',
}}},
}, function(err, passports) {
should.not.exist(err);
should.exist(passports);
let passport, owner, posts;
if (connectorCapabilities.adhocSort !== false) {
passports.length.should.equal(4);
passport = passports[0];
passport.number.should.equal('1');
passport.owner().name.should.equal('User A');
owner = passport.owner().toObject();
posts = passport.owner().posts();
posts.should.be.an.array;
posts.should.have.length(3);
posts[0].title.should.equal('Post C');
posts[0].should.have.property('id', undefined); // omitted
posts[0].author().should.be.instanceOf(User);
posts[0].author().name.should.equal('User A');
posts[1].title.should.equal('Post B');
posts[1].author().name.should.equal('User A');
posts[2].title.should.equal('Post A');
posts[2].author().name.should.equal('User A');
} else {
passports.length.should.be.belowOrEqual(4);
passport = passports[0];
passport.number.should.be.oneOf(knownPassports);
if (passport.owner()) {
passport.owner().name.should.be.oneOf(knownUsers);
owner = passport.owner().toObject();
posts = passport.owner().posts();
posts.should.be.an.array;
posts.length.should.be.belowOrEqual(3);
if (posts[0]) {
posts[0].title.should.be.oneOf(knownPosts);
posts[0].author().should.be.instanceOf(User);
posts[0].author().name.should.be.oneOf(knownUsers);
}
}
}
done();
});
});
bdd.itIf(connectorCapabilities.adhocSort !== false,
'should support limit', function(done) {
Passport.find({
include: {
owner: {
relation: 'posts', scope: {
fields: ['title'], include: ['author'],
order: 'title DESC',
limit: 1,
},
},
},
limit: 2,
}, function(err, passports) {
if (err) return done(err);
passports.length.should.equal(2);
const posts1 = passports[0].toJSON().owner.posts;
posts1.length.should.equal(1);
posts1[0].title.should.equal('Post C');
const posts2 = passports[1].toJSON().owner.posts;
posts2.length.should.equal(1);
posts2[0].title.should.equal('Post D');
done();
});
});
bdd.itIf(connectorCapabilities.cloudantCompatible !== false,
'should support limit - no sort', function(done) {
Passport.find({
include: {
owner: {
relation: 'posts', scope: {
fields: ['title'], include: ['author'],
order: 'title DESC',
limit: 1,
},
},
},
limit: 2,
}, function(err, passports) {
if (err) return done(err);
passports.length.should.equal(2);
let owner = passports[0].toJSON().owner;
if (owner) {
const posts1 = owner.posts;
posts1.length.should.belowOrEqual(1);
if (posts1.length === 1) {
posts1[0].title.should.be.oneOf(knownPosts);
}
}
owner = passports[1].toJSON().owner;
if (owner) {
const posts2 = owner.posts;
posts2.length.should.belowOrEqual(1);
if (posts2.length === 1) {
posts2[0].title.should.be.oneOf(knownPosts);
}
}
done();
});
});
bdd.describeIf(connectorCapabilities.adhocSort !== false,
'inq limit', function() {
before(function() {
Passport.dataSource.settings.inqLimit = 2;
});
after(function() {
delete Passport.dataSource.settings.inqLimit;
});
it('should support include by pagination', function(done) {
// `pagination` in this case is inside the implementation and set by
// `inqLimit = 2` in the before block. This will need to be reworked once
// we decouple `findWithForeignKeysByPage`.
//
// --superkhau
Passport.find({
include: {
owner: {
relation: 'posts',
scope: {
fields: ['title'], include: ['author'],
order: 'title ASC',
},
},
},
}, function(err, passports) {
if (err) return done(err);
passports.length.should.equal(4);
const posts1 = passports[0].toJSON().owner.posts;
posts1.length.should.equal(3);
posts1[0].title.should.equal('Post A');
const posts2 = passports[1].toJSON().owner.posts;
posts2.length.should.equal(1);
posts2[0].title.should.equal('Post D');
done();
});
});
});
bdd.describeIf(connectorCapabilities.adhocSort !== false,
'findWithForeignKeysByPage', function() {
context('filter', function() {
it('works when using a `where` with a foreign key', function(done) {
User.findOne({
include: {
relation: 'passports',
},
}, function(err, user) {
if (err) return done(err);
const passport = user.passports()[0];
// eql instead of equal because mongo uses object id type
passport.id.should.eql(createdPassports[0].id);
passport.ownerId.should.eql(createdPassports[0].ownerId);
passport.number.should.eql(createdPassports[0].number);
done();
});
});
it('works when using a `where` with `and`', function(done) {
User.findOne({
include: {
relation: 'posts',
scope: {
where: {
and: [
{id: createdPosts[0].id},
// Remove the duplicate userId to avoid Cassandra failure
// {userId: createdPosts[0].userId},
{title: 'Post A'},
],
},
},
},
}, function(err, user) {
if (err) return done(err);
user.name.should.equal('User A');
user.age.should.equal(21);
user.id.should.eql(createdUsers[0].id);
const posts = user.posts();
posts.length.should.equal(1);
const post = posts[0];
post.title.should.equal('Post A');
// eql instead of equal because mongo uses object id type
post.userId.should.eql(createdPosts[0].userId);
post.id.should.eql(createdPosts[0].id);
done();
});
});
it('works when using `where` with `limit`', function(done) {
User.findOne({
include: {
relation: 'posts',
scope: {
limit: 1,
},
},
}, function(err, user) {
if (err) return done(err);
user.posts().length.should.equal(1);
done();
});
});
it('works when using `where` with `skip`', function(done) {
User.findOne({
include: {
relation: 'posts',
scope: {
skip: 1,
},
},
}, function(err, user) {
if (err) return done(err);
const ids = user.posts().map(function(p) { return p.id; });
ids.should.eql([createdPosts[1].id, createdPosts[2].id]);
done();
});
});
it('works when using `where` with `offset`', function(done) {
User.findOne({
include: {
relation: 'posts',
scope: {
offset: 1,
},
},
}, function(err, user) {
if (err) return done(err);
const ids = user.posts().map(function(p) { return p.id; });
ids.should.eql([createdPosts[1].id, createdPosts[2].id]);
done();
});
});
it('works when using `where` without `limit`, `skip` or `offset`',
function(done) {
User.findOne({include: {relation: 'posts'}}, function(err, user) {
if (err) return done(err);
const posts = user.posts();
const ids = posts.map(function(p) { return p.id; });
ids.should.eql([
createdPosts[0].id,
createdPosts[1].id,
createdPosts[2].id,
]);
done();
});
});
});
context('pagination', function() {
it('works with the default page size (0) and `inqlimit` is exceeded',
function(done) {
// inqLimit modifies page size in the impl (there is no way to modify
// page size directly as it is hardcoded (once we decouple the func,
// we can use ctor injection to pass in whatever page size we want).
//
// --superkhau
Post.dataSource.settings.inqLimit = 2;
User.find({include: {relation: 'posts'}}, function(err, users) {
if (err) return done(err);
users.length.should.equal(5);
delete Post.dataSource.settings.inqLimit;
done();
});
});
it('works when page size is set to 0', function(done) {
Post.dataSource.settings.inqLimit = 0;
User.find({include: {relation: 'posts'}}, function(err, users) {
if (err) return done(err);
users.length.should.equal(5);
delete Post.dataSource.settings.inqLimit;
done();
});
});
});
context('relations', function() {
// WARNING
// The code paths for in this suite of tests were verified manually due to
// the tight coupling of the `findWithForeignKeys` in `include.js`.
//
// TODO
// Decouple the utility functions into their own modules and export each
// function individually to allow for unit testing via DI.
//
// --superkhau
it('works when hasOne is called', function(done) {
User.findOne({include: {relation: 'profile'}}, function(err, user) {
if (err) return done(err);
user.name.should.equal('User A');
user.age.should.equal(21);
// eql instead of equal because mongo uses object id type
user.id.should.eql(createdUsers[0].id);
const profile = user.profile();
profile.profileName.should.equal('Profile A');
// eql instead of equal because mongo uses object id type
profile.userId.should.eql(createdProfiles[0].userId);
profile.id.should.eql(createdProfiles[0].id);
done();
});
});
it('does not return included item if hasOne is missing the id property', function(done) {
User.findOne({include: {relation: 'profile'}, fields: 'name'}, function(err, user) {
if (err) return done(err);
should.exist(user);
// Convert to JSON as the user instance has `profile` as a relational method
should.not.exist(user.toJSON().profile);
done();
});
});
it('works when hasMany is called', function(done) {
User.findOne({include: {relation: 'posts'}}, function(err, user) {
if (err) return done();
user.name.should.equal('User A');
user.age.should.equal(21);
// eql instead of equal because mongo uses object id type
user.id.should.eql(createdUsers[0].id);
user.posts().length.should.equal(3);
done();
});
});
it('works when hasManyThrough is called', function(done) {
const Physician = db.define('Physician', {name: String});
const Patient = db.define('Patient', {name: String});
const Appointment = db.define('Appointment', {
date: {
type: Date,
default: function() {
return new Date();
},
},
});
const Address = db.define('Address', {name: String});
Physician.hasMany(Patient, {through: Appointment});
Patient.hasMany(Physician, {through: Appointment});
Patient.belongsTo(Address);
Appointment.belongsTo(Patient);
Appointment.belongsTo(Physician);
db.automigrate(['Physician', 'Patient', 'Appointment', 'Address'],
function() {
Physician.create(function(err, physician) {
physician.patients.create({name: 'a'}, function(err, patient) {
Address.create({name: 'z'}, function(err, address) {
patient.address(address);
patient.save(function() {
physician.patients({include: 'address'},
function(err, patients) {
if (err) return done(err);
patients.should.have.length(1);
const p = patients[0];
p.name.should.equal('a');
p.addressId.should.eql(patient.addressId);
p.address().id.should.eql(address.id);
p.address().name.should.equal('z');
done();
});
});
});
});
});
});
});
it('works when belongsTo is called', function(done) {
Profile.findOne({include: 'user'}, function(err, profile) {
if (err) return done(err);
profile.profileName.should.equal('Profile A');
profile.userId.should.eql(createdProfiles[0].userId);
profile.id.should.eql(createdProfiles[0].id);
const user = profile.user();
user.name.should.equal('User A');
user.age.should.equal(21);
user.id.should.eql(createdUsers[0].id);
done();
});
});
});
});
bdd.describeIf(connectorCapabilities.adhocSort === false,
'findWithForeignKeysByPage', function() {
// eslint-disable-next-line mocha/no-identical-title
context('filter', function() {
it('works when using a `where` with a foreign key', function(done) {
User.findOne({
include: {
relation: 'passports',
},
}, function(err, user) {
if (err) return done(err);
const passport = user.passports()[0];
if (passport) {
const knownPassportIds = [];
const knownOwnerIds = [];
createdPassports.forEach(function(p) {
if (p.id) knownPassportIds.push(p.id);
if (p.ownerId) knownOwnerIds.push(p.ownerId.toString());
});
passport.id.should.be.oneOf(knownPassportIds);
// FIXME passport.ownerId may be string
passport.ownerId.toString().should.be.oneOf(knownOwnerIds);
passport.number.should.be.oneOf(knownPassports);
}
done();
});
});
it('works when using a `where` with `and`', function(done) {
User.findOne({
include: {
relation: 'posts',
scope: {
where: {
and: [
{id: createdPosts[0].id},
// Remove the duplicate userId to avoid Cassandra failure
// {userId: createdPosts[0].userId},
{title: createdPosts[0].title},
],
},
},
},
}, function(err, user) {
if (err) return done(err);
let posts, post;
if (connectorCapabilities.adhocSort !== false) {
user.name.should.equal('User A');
user.age.should.equal(21);
user.id.should.eql(createdUsers[0].id);
posts = user.posts();
posts.length.should.equal(1);
post = posts[0];
post.title.should.equal('Post A');
// eql instead of equal because mongo uses object id type
post.userId.should.eql(createdPosts[0].userId);
post.id.should.eql(createdPosts[0].id);
} else {
user.name.should.be.oneOf(knownUsers);
const knownUserIds = [];
createdUsers.forEach(function(u) {
knownUserIds.push(u.id.toString());
});
user.id.toString().should.be.oneOf(knownUserIds);
posts = user.posts();
if (posts && posts.length > 0) {
post = posts[0];
post.title.should.be.oneOf(knownPosts);
post.userId.toString().should.be.oneOf(knownUserIds);
const knownPostIds = [];
createdPosts.forEach(function(p) {
knownPostIds.push(p.id);
});
post.id.should.be.oneOf(knownPostIds);
}
}
done();
});
});
it('works when using `where` with `limit`', function(done) {
User.findOne({
include: {
relation: 'posts',
scope: {
limit: 1,
},
},
}, function(err, user) {
if (err) return done(err);
user.posts().length.should.belowOrEqual(1);
done();
});
});
it('works when using `where` with `skip`', function(done) {
User.findOne({
include: {
relation: 'posts',
scope: {
skip: 1, // will be ignored
},
},
}, function(err, user) {
if (err) return done(err);
const ids = user.posts().map(function(p) { return p.id; });
if (ids.length > 0) {
const knownPosts = [];
createdPosts.forEach(function(p) {
if (p.id) knownPosts.push(p.id);
});
ids.forEach(function(id) {
if (id) id.should.be.oneOf(knownPosts);
});
}
done();
});
});
it('works when using `where` with `offset`', function(done) {
User.findOne({
include: {
relation: 'posts',
scope: {
offset: 1, // will be ignored
},
},
}, function(err, user) {
if (err) return done(err);
const ids = user.posts().map(function(p) { return p.id; });
if (ids.length > 0) {
const knownPosts = [];
createdPosts.forEach(function(p) {
if (p.id) knownPosts.push(p.id);
});
ids.forEach(function(id) {
if (id) id.should.be.oneOf(knownPosts);
});
}
done();
});
});
it('works when using `where` without `limit`, `skip` or `offset`',
function(done) {
User.findOne({include: {relation: 'posts'}}, function(err, user) {
if (err) return done(err);
const posts = user.posts();
const ids = posts.map(function(p) { return p.id; });
if (ids.length > 0) {
const knownPosts = [];
createdPosts.forEach(function(p) {
if (p.id) knownPosts.push(p.id);
});
ids.forEach(function(id) {
if (id) id.should.be.oneOf(knownPosts);
});
}
done();
});
});
});
// eslint-disable-next-line mocha/no-identical-title
context('pagination', function() {
it('works with the default page size (0) and `inqlimit` is exceeded',
function(done) {
// inqLimit modifies page size in the impl (there is no way to modify
// page size directly as it is hardcoded (once we decouple the func,
// we can use ctor injection to pass in whatever page size we want).
//
// --superkhau
Post.dataSource.settings.inqLimit = 2;
User.find({include: {relation: 'posts'}}, function(err, users) {
if (err) return done(err);
users.length.should.equal(5);
delete Post.dataSource.settings.inqLimit;
done();
});
});
it('works when page size is set to 0', function(done) {
Post.dataSource.settings.inqLimit = 0;
User.find({include: {relation: 'posts'}}, function(err, users) {
if (err) return done(err);
users.length.should.equal(5);
delete Post.dataSource.settings.inqLimit;
done();
});
});
});
// eslint-disable-next-line mocha/no-identical-title
context('relations', function() {
// WARNING
// The code paths for in this suite of tests were verified manually due to
// the tight coupling of the `findWithForeignKeys` in `include.js`.
//
// TODO
// Decouple the utility functions into their own modules and export each
// function individually to allow for unit testing via DI.
//
// --superkhau
it('works when hasOne is called', function(done) {
User.findOne({include: {relation: 'profile'}}, function(err, user) {
if (err) return done(err);
const knownUserIds = [];
const knownProfileIds = [];
createdUsers.forEach(function(u) {
// FIXME user.id below might be string, so knownUserIds should match
knownUserIds.push(u.id.toString());
});
createdProfiles.forEach(function(p) {
// knownProfileIds.push(p.id ? p.id.toString() : '');
knownProfileIds.push(p.id);
});
if (user) {
user.name.should.be.oneOf(knownUsers);
// eql instead of equal because mongo uses object id type
user.id.toString().should.be.oneOf(knownUserIds);
const profile = user.profile();
if (profile) {
profile.profileName.should.be.oneOf(knownProfiles);
// eql instead of equal because mongo uses object id type
if (profile.userId) profile.userId.toString().should.be.oneOf(knownUserIds);
profile.id.should.be.oneOf(knownProfileIds);
}
}
done();
});
});
it('works when hasMany is called', function(done) {
User.findOne({include: {relation: 'posts'}}, function(err, user) {
if (err) return done();
const knownUserIds = [];
createdUsers.forEach(function(u) {
knownUserIds.push(u.id);
});
user.name.should.be.oneOf(knownUsers);
// eql instead of equal because mongo uses object id type
user.id.should.be.oneOf(knownUserIds);
user.posts().length.should.be.belowOrEqual(3);
done();
});
});
it('works when hasManyThrough is called', function(done) {
const Physician = db.define('Physician', {name: String});
const Patient = db.define('Patient', {name: String});
const Appointment = db.define('Appointment', {
date: {
type: Date,
default: function() {
return new Date();
},
},
});
const Address = db.define('Address', {name: String});
Physician.hasMany(Patient, {through: Appointment});
Patient.hasMany(Physician, {through: Appointment});
Patient.belongsTo(Address);
Appointment.belongsTo(Patient);
Appointment.belongsTo(Physician);
db.automigrate(['Physician', 'Patient', 'Appointment', 'Address'],
function() {
Physician.create(function(err, physician) {
physician.patients.create({name: 'a'}, function(err, patient) {
Address.create({name: 'z'}, function(err, address) {
patient.address(address);
patient.save(function() {
physician.patients({include: 'address'},
function(err, patients) {
if (err) return done(err);
patients.should.have.length(1);
const p = patients[0];
p.name.should.equal('a');
p.addressId.should.eql(patient.addressId);
p.address().id.should.eql(address.id);
p.address().name.should.equal('z');
done();
});
});
});
});
});
});
});
it('works when belongsTo is called', function(done) {
Profile.findOne({include: 'user'}, function(err, profile) {
if (err) return done(err);
if (!profile) return done(); // not every user has progile
const knownUserIds = [];
const knownProfileIds = [];
createdUsers.forEach(function(u) {
knownUserIds.push(u.id.toString());
});
createdProfiles.forEach(function(p) {
if (p.id) knownProfileIds.push(p.id.toString());
});
if (profile) {
profile.profileName.should.be.oneOf(knownProfiles);
if (profile.userId) profile.userId.toString().should.be.oneOf(knownUserIds);
if (profile.id) profile.id.toString().should.be.oneOf(knownProfileIds);
const user = profile.user();
if (user) {
user.name.should.be.oneOf(knownUsers);
user.id.toString().should.be.oneOf(knownUserIds);
}
}
done();
});
});
});
});
bdd.itIf(connectorCapabilities.adhocSort !== false,
'should fetch Users with include scope on Posts - belongsTo',
function(done) {
Post.find({include: {relation: 'author', scope: {fields: ['name']}}},
function(err, posts) {
should.not.exist(err);
should.exist(posts);
posts.length.should.equal(5);
const author = posts[0].author();
author.name.should.equal('User A');
author.should.have.property('id');
author.should.have.property('age', undefined);
done();
});
});
bdd.itIf(connectorCapabilities.adhocSort === false,
'should fetch Users with include scope on Posts - belongsTo - no sort',
function(done) {
Post.find({include: {relation: 'author', scope: {fields: ['name']}}},
function(err, posts) {
should.not.exist(err);
should.exist(posts);
posts.length.should.be.belowOrEqual(5);
const author = posts[0].author();
if (author) {
author.name.should.be.oneOf('User A', 'User B', 'User C', 'User D', 'User E');
author.should.have.property('id');
author.should.have.property('age', undefined);
}
done();
});
});
it('should fetch Users with include scope on Posts - hasMany', function(done) {
User.find({
include: {relation: 'posts', scope: {
order: 'title DESC',
}},
}, function(err, users) {
should.not.exist(err);
should.exist(users);
users.length.should.equal(5);
if (connectorCapabilities.adhocSort !== false) {
users[0].name.should.equal('User A');
users[1].name.should.equal('User B');
let posts = users[0].posts();
posts.should.be.an.array;
posts.should.have.length(3);
posts[0].title.should.equal('Post C');
posts[1].title.should.equal('Post B');
posts[2].title.should.equal('Post A');
posts = users[1].posts();
posts.should.be.an.array;
posts.should.have.length(1);
posts[0].title.should.equal('Post D');
} else {
users.forEach(function(u) {
u.name.should.be.oneOf(knownUsers);
const posts = u.posts();
if (posts) {
posts.should.be.an.array;
posts.length.should.be.belowOrEqual(3);
posts.forEach(function(p) {
p.title.should.be.oneOf(knownPosts);
});
}
});
}
done();
});
});
it('should fetch User - Posts AND Passports', function(done) {
User.find({include: ['posts', 'passports']}, function(err, users) {
should.not.exist(err);
should.exist(users);
users.length.should.be.ok;
users.forEach(function(user) {
// The relation should be promoted as the 'owner' property
user.should.have.property('posts');
user.should.have.property('passports');
const userObj = user.toJSON();
userObj.should.have.property('posts');
userObj.should.have.property('passports');
userObj.posts.should.be.an.instanceOf(Array);
userObj.passports.should.be.an.instanceOf(Array);
// The __cachedRelations should be removed from json output
userObj.should.not.have.property('__cachedRelations');
user.__cachedRelations.should.have.property('posts');
user.__cachedRelations.should.have.property('passports');
user.__cachedRelations.posts.forEach(function(p) {
// FIXME there are cases that p.userId is string
p.userId.toString().should.eql(user.id.toString());
});
user.__cachedRelations.passports.forEach(function(pp) {
// FIXME there are cases that p.ownerId is string
pp.ownerId.toString().should.eql(user.id.toString());
});
});
done();
});
});
it('should fetch User - Posts AND Passports in relation syntax',
function(done) {
User.find({include: [
{relation: 'posts', scope: {
where: {title: 'Post A'},
}},
'passports',
]}, function(err, users) {
should.not.exist(err);
should.exist(users);
users.length.should.be.ok;
users.forEach(function(user) {
// The relation should be promoted as the 'owner' property
user.should.have.property('posts');
user.should.have.property('passports');
const userObj = user.toJSON();
userObj.should.have.property('posts');
userObj.should.have.property('passports');
userObj.posts.should.be.an.instanceOf(Array);
userObj.passports.should.be.an.instanceOf(Array);
// The __cachedRelations should be removed from json output
userObj.should.not.have.property('__cachedRelations');
user.__cachedRelations.should.have.property('posts');
user.__cachedRelations.should.have.property('passports');
user.__cachedRelations.posts.forEach(function(p) {
// FIXME there are cases that p.userId is string
p.userId.toString().should.eql(user.id.toString());
p.title.should.be.equal('Post A');
});
user.__cachedRelations.passports.forEach(function(pp) {
// FIXME there are cases that p.ownerId is string
pp.ownerId.toString().should.eql(user.id.toString());
});
});
done();
});
});
it('should not fetch User - AccessTokens', function(done) {
User.find({include: ['accesstokens']}, function(err, users) {
should.not.exist(err);
should.exist(users);
users.length.should.be.ok;
users.forEach(function(user) {
const userObj = user.toJSON();
userObj.should.not.have.property('accesstokens');
});
done();
});
});
it('should support hasAndBelongsToMany', function(done) {
Assembly.create({name: 'car'}, function(err, assembly) {
Part.create({partNumber: 'engine'}, function(err, part) {
assembly.parts.add(part, function(err, data) {
assembly.parts(function(err, parts) {
should.not.exist(err);
should.exists(parts);
parts.length.should.equal(1);
parts[0].partNumber.should.equal('engine');
// Create a part
assembly.parts.create({partNumber: 'door'}, function(err, part4) {
Assembly.find({include: 'parts'}, function(err, assemblies) {
assemblies.length.should.equal(1);
assemblies[0].parts().length.should.equal(2);
done();
});
});
});
});
});
});
});
it('should fetch User - Profile (HasOne)', function(done) {
User.find({include: ['profile']}, function(err, users) {
should.not.exist(err);
should.exist(users);
users.length.should.be.ok;
let usersWithProfile = 0;
users.forEach(function(user) {
// The relation should be promoted as the 'owner' property
user.should.have.property('profile');
const userObj = user.toJSON();
const profile = user.profile();
if (profile) {
profile.should.be.an.instanceOf(Profile);
usersWithProfile++;
} else {
(profile === null).should.be.true;
}
// The __cachedRelations should be removed from json output
userObj.should.not.have.property('__cachedRelations');
user.__cachedRelations.should.have.property('profile');
if (user.__cachedRelations.profile) {
// FIXME there are cases that profile.userId is string
user.__cachedRelations.profile.userId.toString().should.eql(user.id.toString());
usersWithProfile++;
}
});
usersWithProfile.should.equal(2 * 2);
done();
});
});
// Not implemented correctly, see: loopback-datasource-juggler/issues/166
// fixed by DB optimization
it('should support include scope on hasAndBelongsToMany', function(done) {
Assembly.find({include: {relation: 'parts', scope: {
where: {partNumber: 'engine'},
}}}, function(err, assemblies) {
assemblies.length.should.equal(1);
const parts = assemblies[0].parts();
parts.should.have.length(1);
parts[0].partNumber.should.equal('engine');
done();
});
});
it('should save related items separately', function(done) {
User.find({
include: 'posts',
})
.then(function(users) {
const posts = users[0].posts();
if (connectorCapabilities.adhocSort !== false) {
posts.should.have.length(3);
} else {
if (posts) posts.length.should.be.belowOrEqual(3);
}
return users[0].save();
})
.then(function(updatedUser) {
return User.findById(updatedUser.id, {
include: 'posts',
});
})
.then(function(user) {
const posts = user.posts();
if (connectorCapabilities.adhocSort !== false) {
posts.should.have.length(3);
} else {
if (posts) posts.length.should.be.belowOrEqual(3);
}
})
.then(done)
.catch(done);
});
describe('performance', function() {
let all;
beforeEach(function() {
this.called = 0;
const self = this;
all = db.connector.all;
db.connector.all = function(model, filter, options, cb) {
self.called++;
return all.apply(db.connector, arguments);
};
});
afterEach(function() {
db.connector.all = all;
});
const nDBCalls = connectorCapabilities.supportTwoOrMoreInq !== false ? 2 : 4;
it('including belongsTo should make only ' + nDBCalls + ' db calls', function(done) {
const self = this;
Passport.find({include: 'owner'}, function(err, passports) {
passports.length.should.be.ok;
passports.forEach(function(p) {
p.__cachedRelations.should.have.property('owner');
// The relation should be promoted as the 'owner' property
p.should.have.property('owner');
// The __cachedRelations should be removed from json output
p.toJSON().should.not.have.property('__cachedRelations');
const owner = p.__cachedRelations.owner;
if (!p.ownerId) {
should.not.exist(owner);
} else {
should.exist(owner);
owner.id.should.eql(p.ownerId);
}
});
self.called.should.eql(nDBCalls);
done();
});
});
it('including hasManyThrough should make only 3 db calls', function(done) {
const self = this;
Assembly.create([{name: 'sedan'}, {name: 'hatchback'},
{name: 'SUV'}],
function(err, assemblies) {
Part.create([{partNumber: 'engine'}, {partNumber: 'bootspace'},
{partNumber: 'silencer'}],
function(err, parts) {
async.each(parts, function(part, next) {
async.each(assemblies, function(assembly, next) {
if (assembly.name === 'SUV') {
return next();
}
if (assembly.name === 'hatchback' &&
part.partNumber === 'bootspace') {
return next();
}
assembly.parts.add(part, function(err, data) {
next();
});
}, next);
}, function(err) {
const autos = connectorCapabilities.supportTwoOrMoreInq !== false ?
['sedan', 'hatchback', 'SUV'] : ['sedan'];
const resultLength = connectorCapabilities.supportTwoOrMoreInq !== false ? 3 : 1;
const dbCalls = connectorCapabilities.supportTwoOrMoreInq !== false ? 3 : 5;
self.called = 0;
Assembly.find({
where: {
name: {
inq: autos,
},
},
include: 'parts',
}, function(err, result) {
should.not.exist(err);
should.exists(result);
result.length.should.equal(resultLength);
// Please note the order of assemblies is random
const assemblies = {};
result.forEach(function(r) {
assemblies[r.name] = r;
});
if (autos.indexOf('sedan') >= 0) assemblies.sedan.parts().should.have.length(3);
if (autos.indexOf('hatchback') >= 0) assemblies.hatchback.parts().should.have.length(2);
if (autos.indexOf('SUV') >= 0) assemblies.SUV.parts().should.have.length(0);
self.called.should.eql(dbCalls);
done();
});
});
});
});
});
const dbCalls = connectorCapabilities.supportTwoOrMoreInq !== false ? 3 : 11;
it('including hasMany should make only ' + dbCalls + ' db calls', function(done) {
const self = this;
User.find({include: ['posts', 'passports']}, function(err, users) {
should.not.exist(err);
should.exist(users);
users.length.should.be.ok;
users.forEach(function(user) {
// The relation should be promoted as the 'owner' property
user.should.have.property('posts');
user.should.have.property('passports');
const userObj = user.toJSON();
userObj.should.have.property('posts');
userObj.should.have.property('passports');
userObj.posts.should.be.an.instanceOf(Array);
userObj.passports.should.be.an.instanceOf(Array);
// The __cachedRelations should be removed from json output
userObj.should.not.have.property('__cachedRelations');
user.__cachedRelations.should.have.property('posts');
user.__cachedRelations.should.have.property('passports');
user.__cachedRelations.posts.forEach(function(p) {
// FIXME p.userId is string in some cases.
if (p.userId) p.userId.toString().should.eql(user.id.toString());
});
user.__cachedRelations.passports.forEach(function(pp) {
// FIXME pp.owerId is string in some cases.
if (pp.owerId) pp.ownerId.toString().should.eql(user.id.toString());
});
});
self.called.should.eql(dbCalls);
done();
});
});
it('should not make n+1 db calls in relation syntax',
function(done) {
const self = this;
User.find({include: [{relation: 'posts', scope: {
where: {title: 'Post A'},
}}, 'passports']}, function(err, users) {
should.not.exist(err);
should.exist(users);
users.length.should.be.ok;
users.forEach(function(user) {
// The relation should be promoted as the 'owner' property
user.should.have.property('posts');
user.should.have.property('passports');
const userObj = user.toJSON();
userObj.should.have.property('posts');
userObj.should.have.property('passports');
userObj.posts.should.be.an.instanceOf(Array);
userObj.passports.should.be.an.instanceOf(Array);
// The __cachedRelations should be removed from json output
userObj.should.not.have.property('__cachedRelations');
user.__cachedRelations.should.have.property('posts');
user.__cachedRelations.should.have.property('passports');
user.__cachedRelations.posts.forEach(function(p) {
// FIXME p.userId is string in some cases.
p.userId.toString().should.eql(user.id.toString());
p.title.should.be.equal('Post A');
});
user.__cachedRelations.passports.forEach(function(pp) {
// FIXME p.userId is string in some cases.
pp.ownerId.toString().should.eql(user.id.toString());
});
});
self.called.should.eql(dbCalls);
done();
});
});
});
it('should support disableInclude for hasAndBelongsToMany', function() {
const Patient = db.define('Patient', {name: String});
const Doctor = db.define('Doctor', {name: String});
const DoctorPatient = db.define('DoctorPatient');
Doctor.hasAndBelongsToMany('patients', {
model: 'Patient',
options: {disableInclude: true},
});
let doctor;
return db.automigrate(['Patient', 'Doctor', 'DoctorPatient']).then(function() {
return Doctor.create({name: 'Who'});
}).then(function(inst) {
doctor = inst;
return doctor.patients.create({name: 'Lazarus'});
}).then(function() {
return Doctor.find({include: ['patients']});
}).then(function(list) {
list.should.have.length(1);
list[0].toJSON().should.not.have.property('patients');
});
});
});
let createdUsers = [];
let createdPassports = [];
let createdProfiles = [];
let createdPosts = [];
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,
});
Profile = db.define('Profile', {
profileName: String,
});
AccessToken = db.define('AccessToken', {
token: String,
});
Passport = db.define('Passport', {
number: String,
expirationDate: Date,
});
Post = db.define('Post', {
title: String,
});
Passport.belongsTo('owner', {model: User});
User.hasMany('passports', {foreignKey: 'ownerId'});
User.hasMany('posts', {foreignKey: 'userId'});
User.hasMany('accesstokens', {
foreignKey: 'userId',
options: {disableInclude: true},
});
Profile.belongsTo('user', {model: User});
User.hasOne('profile', {foreignKey: 'userId'});
Post.belongsTo('author', {model: User, foreignKey: 'userId'});
Assembly = db.define('Assembly', {
name: String,
});
Part = db.define('Part', {
partNumber: String,
});
Assembly.hasAndBelongsToMany(Part);
Part.hasAndBelongsToMany(Assembly);
db.automigrate(function() {
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();
createAccessTokens();
}
);
}
function createAccessTokens() {
clearAndCreate(
AccessToken,
[
{token: '1', userId: createdUsers[0].id},
{token: '2', userId: createdUsers[1].id},
],
function(items) {}
);
}
function createPassports() {
clearAndCreate(
Passport,
[
{number: '1', ownerId: createdUsers[0].id},
{number: '2', ownerId: createdUsers[1].id},
{number: '3'},
{number: '4', ownerId: createdUsers[2].id},
],
function(items) {
createdPassports = items;
createPosts();
}
);
}
function createProfiles() {
clearAndCreate(
Profile,
[
{profileName: 'Profile A', userId: createdUsers[0].id},
{profileName: 'Profile B', userId: createdUsers[1].id},
{profileName: 'Profile Z'},
],
function(items) {
createdProfiles = items;
done();
}
);
}
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;
createProfiles();
}
);
}
});
}
function clearAndCreate(model, data, callback) {
const createdItems = [];
model.destroyAll(function() {
nextItem(null, null);
});
let 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++;
}
}
describe('Model instance with included relation .toJSON()', function() {
let db, ChallengerModel, GameParticipationModel, ResultModel;
before(function(done) {
db = new DataSource({connector: 'memory'});
ChallengerModel = db.createModel('Challenger',
{
name: String,
},
{
relations: {
gameParticipations: {
type: 'hasMany',
model: 'GameParticipation',
foreignKey: '',
},
},
});
GameParticipationModel = db.createModel('GameParticipation',
{
date: Date,
},
{
relations: {
challenger: {
type: 'belongsTo',
model: 'Challenger',
foreignKey: '',
},
results: {
type: 'hasMany',
model: 'Result',
foreignKey: '',
},
},
});
ResultModel = db.createModel('Result', {
points: Number,
}, {
relations: {
gameParticipation: {
type: 'belongsTo',
model: 'GameParticipation',
foreignKey: '',
},
},
});
async.waterfall([
createChallengers,
createGameParticipations,
createResults],
function(err) {
done(err);
});
});
function createChallengers(callback) {
ChallengerModel.create([{name: 'challenger1'}, {name: 'challenger2'}], callback);
}
function createGameParticipations(challengers, callback) {
GameParticipationModel.create([
{challengerId: challengers[0].id, date: Date.now()},
{challengerId: challengers[0].id, date: Date.now()},
], callback);
}
function createResults(gameParticipations, callback) {
ResultModel.create([
{gameParticipationId: gameParticipations[0].id, points: 10},
{gameParticipationId: gameParticipations[0].id, points: 20},
], callback);
}
it('should recursively serialize objects', function(done) {
const filter = {include: {gameParticipations: 'results'}};
ChallengerModel.find(filter, function(err, challengers) {
const levelOneInclusion = challengers[0].toJSON().gameParticipations[0];
assert(levelOneInclusion.__data === undefined, '.__data of a level 1 inclusion is undefined.');
const levelTwoInclusion = challengers[0].toJSON().gameParticipations[0].results[0];
assert(levelTwoInclusion.__data === undefined, '__data of a level 2 inclusion is undefined.');
done();
});
});
});