New upstream version 6.0.20upstream

1 files changed, 26 insertions, 0 deletions

diff --git a/src/mongo/db/pipeline/document_source_exchange_test.cpp b/src/mongo/db/pipeline/document_source_exchange_test.cpp
index ac234357b17..4ed642c49c7 100644
--- a/src/mongo/db/pipeline/document_source_exchange_test.cpp
+++ b/src/mongo/db/pipeline/document_source_exchange_test.cpp
@@ -747,4 +747,30 @@ TEST_F(DocumentSourceExchangeTest, RejectInvalidMissingKeys) {
         Exchange(parseSpec(spec), Pipeline::create({}, getExpCtx())), AssertionException, 50967);
 }
 
+TEST_F(DocumentSourceExchangeTest, QueryShape) {
+    const size_t nDocs = 500;
+
+    auto source = getMockSource(nDocs);
+
+    ExchangeSpec spec;
+    spec.setPolicy(ExchangePolicyEnum::kRoundRobin);
+    spec.setConsumers(1);
+    spec.setBufferSize(1024);
+    boost::intrusive_ptr<Exchange> ex = new Exchange(spec, Pipeline::create({source}, getExpCtx()));
+    boost::intrusive_ptr<DocumentSourceExchange> stage =
+        new DocumentSourceExchange(getExpCtx(), ex, 0, nullptr);
+
+    ASSERT_BSONOBJ_EQ_AUTO(  //
+        R"({
+            "$_internalExchange": {
+                "policy": "roundrobin",
+                "consumers": "?number",
+                "orderPreserving": false,
+                "bufferSize": "?number",
+                "key": "?object"
+            }
+        })",
+        redact(*stage));
+}
+
 }  // namespace mongo