diff options
| author | Lucas de Castro Borges <lucas@gnuabordo.com.br> | 2025-02-14 14:26:38 -0300 |
|---|---|---|
| committer | Lucas de Castro Borges <lucas@gnuabordo.com.br> | 2025-02-14 14:26:38 -0300 |
| commit | 294bc6ecabf14c09c9bc8644704921dcf97cb44e (patch) | |
| tree | 279b1e0bab53901a1647ac63c1c724f0f789a663 /src/mongo/db/pipeline/document_source_redact_test.cpp | |
| parent | 70be7c27a251621187a1de533462ae2bb1e3bd39 (diff) | |
| parent | 1e917fd798aa25b7066d4b414b51184f13d5a092 (diff) | |
Update upstream source from tag 'upstream/6.0.10'debian/6.0.10-1
Update to upstream version '6.0.10'
with Debian dir 2d176fa254eee97b139f712fec5709641335a8c3
Diffstat (limited to 'src/mongo/db/pipeline/document_source_redact_test.cpp')
| -rw-r--r-- | src/mongo/db/pipeline/document_source_redact_test.cpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/mongo/db/pipeline/document_source_redact_test.cpp b/src/mongo/db/pipeline/document_source_redact_test.cpp index 95d001e4dac..6605796138b 100644 --- a/src/mongo/db/pipeline/document_source_redact_test.cpp +++ b/src/mongo/db/pipeline/document_source_redact_test.cpp @@ -81,5 +81,26 @@ TEST_F(DocumentSourceRedactTest, ShouldPropagatePauses) { ASSERT_TRUE(redact->getNext().isEOF()); ASSERT_TRUE(redact->getNext().isEOF()); } + +TEST_F(DocumentSourceRedactTest, ReportsVariableDependencies) { + auto varId = getExpCtx()->variablesParseState.defineVariable("var"); + auto redactSpec = fromjson(R"({ + "$redact" : { + "$cond" : { + "if" : "$$var", + "then" : "$$PRUNE", + "else" : "$$DESCEND" + } + } + })"); + auto redact = DocumentSourceRedact::createFromBson(redactSpec.firstElement(), getExpCtx()); + + DepsTracker deps; + ASSERT_EQ(redact->getDependencies(&deps), DepsTracker::State::SEE_NEXT); + ASSERT_EQ(deps.needWholeDocument, true); + ASSERT_TRUE(deps.fields.empty()); + ASSERT_EQ(deps.vars.count(varId), 1); +} + } // namespace } // namespace mongo |