test: fix tests re: #5881

Automattic · Dec 8, 2017 · 1c3e210 · 1c3e210
1 parent 4f1e1a0
commit 1c3e210
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/lib/document.js b/lib/document.js
@@ -190,7 +190,7 @@ Document.prototype.$__buildDoc = function(obj, fields, skipId) {
       while (ki--) {
         // Does this projection explicitly define inclusion/exclusion?
         // Explicitly avoid `$meta` and `$slice`
-        if (keys[ki] !== '_id' && isDefiningProjection(keys[ki])) {
+        if (keys[ki] !== '_id' && isDefiningProjection(fields[keys[ki]])) {
           exclude = !fields[keys[ki]];
           break;
         }

diff --git a/test/query.test.js b/test/query.test.js
@@ -2412,8 +2412,7 @@ describe('Query', function() {
     });
 
     it('slice projection', function(done) {
-      MyModel.findOne({name: 'John'}, {dependents: {$slice: 1}}).
-      exec(function(error, person) {
+      MyModel.findOne({name: 'John'}, {dependents: {$slice: 1}}).exec(function(error, person) {
         assert.ifError(error);
         assert.equal(person.salary, 25000);
         done();