New upstream version 6.0.20upstream

1 files changed, 18 insertions, 0 deletions

diff --git a/src/mongo/db/pipeline/document_source_sample_test.cpp b/src/mongo/db/pipeline/document_source_sample_test.cpp
index 68cd93c4698..ade80752010 100644
--- a/src/mongo/db/pipeline/document_source_sample_test.cpp
+++ b/src/mongo/db/pipeline/document_source_sample_test.cpp
@@ -188,6 +188,17 @@ TEST_F(SampleBasics, ShouldPropagatePauses) {
     assertEOF();
 }
 
+TEST_F(SampleBasics, RedactsCorrectly) {
+    createSample(10);
+    ASSERT_BSONOBJ_EQ_AUTO(  // NOLINT
+        R"({
+            "$sample": {
+                "size": "?number"
+            }
+        })",
+        redact(*sample()));
+}
+
 /**
  * Fixture to test error cases of the $sample stage.
  */
@@ -402,5 +413,12 @@ DEATH_TEST_REGEX_F(SampleFromRandomCursorBasics,
     sample()->getNext();
 }
 
+TEST_F(SampleFromRandomCursorBasics, RedactsCorrectly) {
+    createSample(2);
+    ASSERT_VALUE_EQ_AUTO(  // NOLINT
+        "{ $sampleFromRandomCursor: { size: \"?number\" } }",
+        redact(*sample()));
+}
+
 }  // namespace
 }  // namespace mongo