Merge pull request #34 from Ebulus7899/master

Add support for findOneAndReplace
mongoosejs · Apr 27, 2024 · a1dcb4a · a1dcb4a
2 parents 87eb2ca + bc5b488
commit a1dcb4a
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/index.js b/index.js
@@ -28,6 +28,7 @@ module.exports = function mongooseLeanGetters(schema) {
   schema.post('findOne', fn);
   schema.post('findOneAndUpdate', fn);
   schema.post('findOneAndDelete', fn);
+  schema.post('findOneAndReplace', fn);
 };
 
 function applyGettersMiddleware(schema) {

diff --git a/test/index.test.js b/test/index.test.js
@@ -330,4 +330,24 @@ describe('mongoose-lean-getters', function() {
     assert.strictEqual(user.name, 'ONE');
     assert.deepStrictEqual(foundUser.emails, ['TWO', 'THREE']);
   });
+
+  it('should work with findOneAndReplace (gh-31)', async function() {
+    const testSchema = new mongoose.Schema({
+      field: Number,
+    });
+    testSchema.plugin(mongooseLeanGetters);
+
+    testSchema.path('field').get(function(field) {
+      return field.toString();
+    });
+    const Test = mongoose.model('gh-31', testSchema);
+
+    await Test.deleteMany({});
+    const entry = await Test.create({
+      field: 1337
+    });
+    const doc = await Test.findOneAndReplace({ _id: entry._id }, entry).lean({ getters: true });
+    assert.equal(typeof doc.field, 'string');
+    assert.strictEqual(doc.field, '1337');
+  });
 });