328 lines
10 KiB
JavaScript
328 lines
10 KiB
JavaScript
// This test written in mocha+should.js
|
|
var should = require('./init.js');
|
|
var db, User;
|
|
|
|
describe('basic-querying', function() {
|
|
|
|
before(function(done) {
|
|
db = getSchema();
|
|
|
|
User = db.define('User', {
|
|
name: {type: String, sort: true},
|
|
email: {type: String, index: true},
|
|
role: {type: String, index: true},
|
|
order: {type: Number, index: true, sort: true}
|
|
});
|
|
|
|
db.automigrate(done);
|
|
|
|
});
|
|
|
|
|
|
describe('findById', function() {
|
|
|
|
before(function(done) {
|
|
User.destroyAll(done);
|
|
});
|
|
|
|
it('should query by id: not found', function(done) {
|
|
User.findById(1, function(err, u) {
|
|
should.not.exist(u);
|
|
should.not.exist(err);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query by id: found', function(done) {
|
|
User.create(function(err, u) {
|
|
should.not.exist(err);
|
|
should.exist(u.id);
|
|
User.findById(u.id, function(err, u) {
|
|
should.exist(u);
|
|
should.not.exist(err);
|
|
u.should.be.an.instanceOf(User);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
});
|
|
|
|
describe('find', function() {
|
|
|
|
before(seed);
|
|
|
|
it('should query collection', function(done) {
|
|
User.find(function(err, users) {
|
|
should.exists(users);
|
|
should.not.exists(err);
|
|
users.should.have.lengthOf(6);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query limited collection', function(done) {
|
|
User.find({limit: 3}, function(err, users) {
|
|
should.exists(users);
|
|
should.not.exists(err);
|
|
users.should.have.lengthOf(3);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query offset collection with limit', function(done) {
|
|
User.find({skip: 1, limit: 4}, function(err, users) {
|
|
should.exists(users);
|
|
should.not.exists(err);
|
|
users.should.have.lengthOf(4);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query filtered collection', function(done) {
|
|
User.find({where: {role: 'lead'}}, function(err, users) {
|
|
should.exists(users);
|
|
should.not.exists(err);
|
|
users.should.have.lengthOf(2);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query collection sorted by numeric field', function(done) {
|
|
User.find({order: 'order'}, function(err, users) {
|
|
should.exists(users);
|
|
should.not.exists(err);
|
|
users.forEach(function(u, i) {
|
|
u.order.should.eql(i + 1);
|
|
});
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query collection desc sorted by numeric field', function(done) {
|
|
User.find({order: 'order DESC'}, function(err, users) {
|
|
should.exists(users);
|
|
should.not.exists(err);
|
|
users.forEach(function(u, i) {
|
|
u.order.should.eql(users.length - i);
|
|
});
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query collection sorted by string field', function(done) {
|
|
User.find({order: 'name'}, function(err, users) {
|
|
should.exists(users);
|
|
should.not.exists(err);
|
|
users.shift().name.should.equal('George Harrison');
|
|
users.shift().name.should.equal('John Lennon');
|
|
users.pop().name.should.equal('Stuart Sutcliffe');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query collection desc sorted by string field', function(done) {
|
|
User.find({order: 'name DESC'}, function(err, users) {
|
|
should.exists(users);
|
|
should.not.exists(err);
|
|
users.pop().name.should.equal('George Harrison');
|
|
users.pop().name.should.equal('John Lennon');
|
|
users.shift().name.should.equal('Stuart Sutcliffe');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should only include fields as specified', function(done) {
|
|
var remaining = 0;
|
|
|
|
function sample(fields) {
|
|
|
|
return {
|
|
expect: function (arr) {
|
|
remaining++;
|
|
User.find({fields: fields}, function(err, users) {
|
|
|
|
remaining--;
|
|
if(err) return done(err);
|
|
|
|
should.exists(users);
|
|
|
|
if(remaining === 0) {
|
|
done();
|
|
}
|
|
|
|
users.forEach(function (user) {
|
|
var obj = user.toObject();
|
|
|
|
Object.keys(obj)
|
|
.forEach(function (key) {
|
|
// if the obj has an unexpected value
|
|
if(obj[key] !== undefined && arr.indexOf(key) === -1) {
|
|
console.log('Given fields:', fields);
|
|
console.log('Got:', key, obj[key]);
|
|
console.log('Expected:', arr);
|
|
throw new Error('should not include data for key: '+ key);
|
|
}
|
|
});
|
|
});
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
sample({name: true}).expect(['name']);
|
|
sample({name: false}).expect(['id', 'email', 'role', 'order']);
|
|
sample({name: false, id: true}).expect(['id']);
|
|
sample({id: true}).expect(['id']);
|
|
sample('id').expect(['id']);
|
|
sample(['id']).expect(['id']);
|
|
sample(['email']).expect(['email']);
|
|
});
|
|
|
|
});
|
|
|
|
describe('count', function() {
|
|
|
|
before(seed);
|
|
|
|
it('should query total count', function(done) {
|
|
User.count(function(err, n) {
|
|
should.not.exist(err);
|
|
should.exist(n);
|
|
n.should.equal(6);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should query filtered count', function(done) {
|
|
User.count({role: 'lead'}, function(err, n) {
|
|
should.not.exist(err);
|
|
should.exist(n);
|
|
n.should.equal(2);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('findOne', function() {
|
|
|
|
before(seed);
|
|
|
|
it('should find first record (default sort by id)', function(done) {
|
|
User.all({order: 'id'}, function(err, users) {
|
|
User.findOne(function(e, u) {
|
|
should.not.exist(e);
|
|
should.exist(u);
|
|
u.id.toString().should.equal(users[0].id.toString());
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should find first record', function(done) {
|
|
User.findOne({order: 'order'}, function(e, u) {
|
|
should.not.exist(e);
|
|
should.exist(u);
|
|
u.order.should.equal(1);
|
|
u.name.should.equal('Paul McCartney');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should find last record', function(done) {
|
|
User.findOne({order: 'order DESC'}, function(e, u) {
|
|
should.not.exist(e);
|
|
should.exist(u);
|
|
u.order.should.equal(6);
|
|
u.name.should.equal('Ringo Starr');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should find last record in filtered set', function(done) {
|
|
User.findOne({
|
|
where: {role: 'lead'},
|
|
order: 'order DESC'
|
|
}, function(e, u) {
|
|
should.not.exist(e);
|
|
should.exist(u);
|
|
u.order.should.equal(2);
|
|
u.name.should.equal('John Lennon');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should work even when find by id', function(done) {
|
|
User.findOne(function(e, u) {
|
|
User.findOne({where: {id: u.id}}, function(err, user) {
|
|
should.not.exist(err);
|
|
should.exist(user);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
});
|
|
|
|
describe('exists', function() {
|
|
|
|
before(seed);
|
|
|
|
it('should check whether record exist', function(done) {
|
|
User.findOne(function(e, u) {
|
|
User.exists(u.id, function(err, exists) {
|
|
should.not.exist(err);
|
|
should.exist(exists);
|
|
exists.should.be.ok;
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should check whether record not exist', function(done) {
|
|
User.destroyAll(function() {
|
|
User.exists(42, function(err, exists) {
|
|
should.not.exist(err);
|
|
exists.should.not.be.ok;
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
});
|
|
|
|
|
|
});
|
|
|
|
function seed(done) {
|
|
var count = 0;
|
|
var beatles = [
|
|
{
|
|
name: 'John Lennon',
|
|
mail: 'john@b3atl3s.co.uk',
|
|
role: 'lead',
|
|
order: 2
|
|
}, {
|
|
name: 'Paul McCartney',
|
|
mail: 'paul@b3atl3s.co.uk',
|
|
role: 'lead',
|
|
order: 1
|
|
},
|
|
{name: 'George Harrison', order: 5},
|
|
{name: 'Ringo Starr', order: 6},
|
|
{name: 'Pete Best', order: 4},
|
|
{name: 'Stuart Sutcliffe', order: 3}
|
|
];
|
|
User.destroyAll(function() {
|
|
beatles.forEach(function(beatle) {
|
|
User.create(beatle, ok);
|
|
});
|
|
});
|
|
|
|
function ok() {
|
|
if (++count === beatles.length) {
|
|
done();
|
|
}
|
|
}
|
|
}
|