Refactor test in query.test.js file.

Automattic · Apr 15, 2022 · d3e3b2f · d3e3b2f
1 parent db8469c
commit d3e3b2f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/query.test.js b/test/query.test.js
@@ -3947,7 +3947,6 @@ describe('Query', function() {
   });
 
   it('should return query helper supplied in schema options query property instead of undefined', function(done) {
-
     const Model = db.model('Test', new Schema({
       userName: {
         type: String,
@@ -3962,7 +3961,9 @@ describe('Query', function() {
     }));
 
     Model.create({ userName: 'test' }, function(error) {
-      assert.ifError(error);
+      if (error instanceof Error) {
+        return done(error);
+      }
       Model.find().byUserName('test').exec(function(error, docs) {
         if (error instanceof Error) {
           return done(error);