deno.land / x / mongoose@6.7.5 / test / model.middleware.test.js
نووسراو ببینە
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475'use strict';
/** * Test dependencies. */
const start = require('./common');
const assert = require('assert');
const mongoose = start.mongoose;const Schema = mongoose.Schema;
describe('model middleware', function() { let db;
before(function() { db = start(); });
after(async function() { await db.close(); });
beforeEach(() => db.deleteModel(/.*/)); afterEach(() => require('./util').clearTestData(db)); afterEach(() => require('./util').stopRemainingOps(db));
it('post save', function(done) { const schema = new Schema({ title: String });
let called = 0;
schema.post('save', function(obj) { assert.equal(obj.title, 'Little Green Running Hood'); assert.equal(this.title, 'Little Green Running Hood'); assert.equal(called, 0); called++; });
schema.post('save', function(obj) { assert.equal(obj.title, 'Little Green Running Hood'); assert.equal(this.title, 'Little Green Running Hood'); assert.equal(called, 1); called++; });
schema.post('save', function(obj, next) { assert.equal(obj.title, 'Little Green Running Hood'); assert.equal(called, 2); called++; next(); });
const TestMiddleware = db.model('Test', schema);
const test = new TestMiddleware({ title: 'Little Green Running Hood' });
test.save(function(err) { assert.ifError(err); assert.equal(test.title, 'Little Green Running Hood'); assert.equal(called, 3); done(); }); });
it('sync error in post save (gh-3483)', function(done) { const schema = new Schema({ title: String });
schema.post('save', function() { throw new Error('woops!'); });
const TestMiddleware = db.model('Test', schema);
const test = new TestMiddleware({ title: 'Test' });
test.save(function(err) { assert.ok(err); assert.equal(err.message, 'woops!'); done(); }); });
it('pre hook promises (gh-3779)', function(done) { const schema = new Schema({ title: String });
let calledPre = 0; schema.pre('save', function() { return new Promise(resolve => { setTimeout(() => { ++calledPre; resolve(); }, 100); }); });
const TestMiddleware = db.model('Test', schema);
const test = new TestMiddleware({ title: 'Test' });
test.save(function(err) { assert.ifError(err); assert.equal(calledPre, 1); done(); }); });
it('post hook promises (gh-3779)', async function() { const schema = new Schema({ title: String });
schema.post('save', function(doc) { return new Promise(resolve => { setTimeout(() => { doc.title = 'From Post Save'; resolve(); }, 100); }); });
const TestMiddleware = db.model('Test', schema);
const test = new TestMiddleware({ title: 'Test' });
const doc = await test.save();
assert.equal(doc.title, 'From Post Save'); });
it('validate middleware runs before save middleware (gh-2462)', function(done) { const schema = new Schema({ title: String }); let count = 0;
schema.pre('validate', function(next) { assert.equal(count++, 0); next(); });
schema.pre('save', function(next) { assert.equal(count++, 1); next(); });
const Book = db.model('Test', schema);
Book.create({}, function() { assert.equal(count, 2); done(); }); });
it('works', function(done) { const schema = new Schema({ title: String });
let called = 0;
schema.pre('init', function() { called++; });
schema.pre('save', function(next) { called++; next(new Error('Error 101')); });
schema.pre('remove', function(next) { called++; next(); });
const TestMiddleware = db.model('TestMiddleware', schema);
const test = new TestMiddleware();
test.init({ title: 'Test' }, function(err) { assert.ifError(err); assert.equal(called, 1);
test.save(function(err) { assert.ok(err instanceof Error); assert.equal(err.message, 'Error 101'); assert.equal(called, 2);
test.remove(function(err) { assert.ifError(err); assert.equal(called, 3); done(); }); }); }); });
describe('post init hooks', function() { it('success', function() { const schema = new Schema({ title: String, loadedAt: Date });
schema.pre('init', pojo => { assert.equal(pojo.constructor.name, 'Object'); // Plain object before init });
const now = new Date(); schema.post('init', doc => { assert.ok(doc instanceof mongoose.Document); // Mongoose doc after init doc.loadedAt = now; });
const Test = db.model('Test', schema);
return Test.create({ title: 'Casino Royale' }). then(doc => Test.findById(doc)). then(doc => assert.equal(doc.loadedAt.valueOf(), now.valueOf())); });
it('with errors', function() { const schema = new Schema({ title: String });
const swallowedError = new Error('will not show'); // acquit:ignore:start swallowedError.$expected = true; // acquit:ignore:end // init hooks do **not** handle async errors or any sort of async behavior schema.pre('init', () => Promise.reject(swallowedError)); schema.post('init', () => { throw Error('will show'); });
const Test = db.model('Test', schema);
return Test.create({ title: 'Casino Royale' }). then(doc => Test.findById(doc)). catch(error => assert.equal(error.message, 'will show')); }); });
it('gh-1829', function(done) { const childSchema = new mongoose.Schema({ name: String });
let childPreCalls = 0; const childPreCallsByName = {}; let parentPreCalls = 0;
childSchema.pre('save', function(next) { childPreCallsByName[this.name] = childPreCallsByName[this.name] || 0; ++childPreCallsByName[this.name]; ++childPreCalls; next(); });
const parentSchema = new mongoose.Schema({ name: String, children: [childSchema] });
parentSchema.pre('save', function(next) { ++parentPreCalls; next(); });
const Parent = db.model('Parent', parentSchema);
const parent = new Parent({ name: 'Han', children: [ { name: 'Jaina' }, { name: 'Jacen' } ] });
parent.save(function(error) { assert.ifError(error); assert.equal(childPreCalls, 2); assert.equal(childPreCallsByName.Jaina, 1); assert.equal(childPreCallsByName.Jacen, 1); assert.equal(parentPreCalls, 1); parent.children[0].name = 'Anakin'; parent.save(function(error) { assert.ifError(error); assert.equal(childPreCalls, 4); assert.equal(childPreCallsByName.Anakin, 1); assert.equal(childPreCallsByName.Jaina, 1); assert.equal(childPreCallsByName.Jacen, 2);
assert.equal(parentPreCalls, 2); done(); }); }); });
it('sync error in pre save (gh-3483)', function(done) { const schema = new Schema({ title: String });
schema.post('save', function() { throw new Error('woops!'); });
const TestMiddleware = db.model('Test', schema);
const test = new TestMiddleware({ title: 'Test' });
test.save(function(err) { assert.ok(err); assert.equal(err.message, 'woops!'); done(); }); });
it('sync error in pre save after next() (gh-3483)', function(done) { const schema = new Schema({ title: String });
let called = 0;
schema.pre('save', function(next) { next(); // This error will not get reported, because you already called next() throw new Error('woops!'); });
schema.pre('save', function(next) { ++called; next(); });
const TestMiddleware = db.model('Test', schema);
const test = new TestMiddleware({ title: 'Test' });
test.save(function(error) { assert.ifError(error); assert.equal(called, 1); done(); }); });
it('validate + remove', function(done) { const schema = new Schema({ title: String });
let preValidate = 0, postValidate = 0, preRemove = 0, postRemove = 0;
schema.pre('validate', function(next) { ++preValidate; next(); });
schema.pre('remove', function(next) { ++preRemove; next(); });
schema.post('validate', function(doc) { assert.ok(doc instanceof mongoose.Document); ++postValidate; });
schema.post('remove', function(doc) { assert.ok(doc instanceof mongoose.Document); ++postRemove; });
const Test = db.model('Test', schema);
const test = new Test({ title: 'banana' });
test.save(function(err) { assert.ifError(err); assert.equal(preValidate, 1); assert.equal(postValidate, 1); assert.equal(preRemove, 0); assert.equal(postRemove, 0); test.remove(function(err) { assert.ifError(err); assert.equal(preValidate, 1); assert.equal(postValidate, 1); assert.equal(preRemove, 1); assert.equal(postRemove, 1); done(); }); }); });
it('static hooks (gh-5982)', async function() { const schema = new Schema({ name: String });
schema.statics.findByName = function(name) { return this.find({ name: name }); };
let preCalled = 0; schema.pre('findByName', function() { ++preCalled; });
let postCalled = 0; schema.post('findByName', function(docs) { ++postCalled; assert.equal(docs.length, 1); assert.equal(docs[0].name, 'foo'); });
const Model = db.model('Test', schema);
await Model.create({ name: 'foo' });
const docs = await Model.findByName('foo'); assert.equal(docs.length, 1); assert.equal(docs[0].name, 'foo'); assert.equal(preCalled, 1); assert.equal(postCalled, 1); });
it('deleteOne hooks (gh-7538)', async function() { const schema = new Schema({ name: String });
let queryPreCalled = 0; let preCalled = 0; schema.pre('deleteOne', { document: false, query: true }, function() { ++queryPreCalled; }); schema.pre('deleteOne', { document: true, query: false }, function() { ++preCalled; });
let postCalled = 0; schema.post('deleteOne', { document: true, query: false }, function() { assert.equal(this.name, 'foo'); ++postCalled; });
const Model = db.model('Test', schema);
await Model.create({ name: 'foo' });
const doc = await Model.findOne();
assert.equal(preCalled, 0); assert.equal(postCalled, 0);
await doc.deleteOne();
assert.equal(queryPreCalled, 0); assert.equal(preCalled, 1); assert.equal(postCalled, 1);
await Model.deleteOne();
assert.equal(queryPreCalled, 1); assert.equal(preCalled, 1); assert.equal(postCalled, 1); });});
mongoose
Version Info
a year ago
