SERVER-40555 add searchSnippet metadata

2 files changed, 7 insertions, 1 deletions

diff --git a/jstests/core/find_projection_meta_errors.js b/jstests/core/find_projection_meta_errors.js
index bbf91801ee0..6fd69cb9d04 100644
--- a/jstests/core/find_projection_meta_errors.js
+++ b/jstests/core/find_projection_meta_errors.js
@@ -14,6 +14,10 @@
         ErrorCodes.BadValue);
 
     assert.commandFailedWithCode(
+        db.runCommand({find: coll.getName(), projection: {score: {$meta: "searchHighlights"}}}),
+        ErrorCodes.BadValue);
+
+    assert.commandFailedWithCode(
         db.runCommand({find: coll.getName(), projection: {score: {$meta: "some garbage"}}}),
         ErrorCodes.BadValue);
 }());
diff --git a/jstests/sharding/aggregates_during_balancing.js b/jstests/sharding/aggregates_during_balancing.js
index 149f6031583..003674e1530 100644
--- a/jstests/sharding/aggregates_during_balancing.js
+++ b/jstests/sharding/aggregates_during_balancing.js
@@ -179,7 +179,9 @@
     function testSample() {
         jsTestLog('testing $sample');
         [0, 1, 10, nItems, nItems + 1].forEach(function(size) {
-            var res = db.ts1.aggregate([{$sample: {size: size}}]).toArray();
+            // Run with 'allowDiskUse' set to true because this may exceed the in-memory sort
+            // limit.
+            var res = db.ts1.aggregate([{$sample: {size: size}}], {allowDiskUse: true}).toArray();
             assert.eq(res.length, Math.min(nItems, size));
         });
     }