fix(query): make findOneAndReplace() work with orFail()

Fix #8030
Automattic · Jul 30, 2019 · 9db4857 · 9db4857
1 parent f10f621
commit 9db4857
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/query.js b/lib/query.js
@@ -3286,15 +3286,15 @@ Query.prototype._findOneAndReplace = wrapThunk(function(callback) {
       castedDoc = castedDoc.toBSON();
     }
 
-    this._collection.collection.findOneAndReplace(filter, castedDoc, options, (err, res) => {
+    this._collection.collection.findOneAndReplace(filter, castedDoc, options, _wrapThunkCallback(this, (err, res) => {
       if (err) {
         return callback(err);
       }
 
       const doc = res.value;
 
       return this._completeOne(doc, res, callback);
-    });
+    }));
   });
 });
 
@@ -4199,6 +4199,7 @@ Query.prototype.orFail = function(err) {
         }
         break;
       case 'findOneAndUpdate':
+      case 'findOneAndReplace':
         if (get(res, 'lastErrorObject.updatedExisting') === false) {
           throw _orFailError(err, this);
         }