summaryrefslogtreecommitdiff
path: root/src/mongo/db/pipeline/document_source_sample_test.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/mongo/db/pipeline/document_source_sample_test.cpp')
-rw-r--r--src/mongo/db/pipeline/document_source_sample_test.cpp18
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