SERVER-24153 Add unit tests for DocumentSource pausing behavior.

1 files changed, 11 insertions, 1 deletions

diff --git a/src/mongo/db/pipeline/document_source_mock_test.cpp b/src/mongo/db/pipeline/document_source_mock_test.cpp
index acf4f21f3fe..e4826ff5b24 100644
--- a/src/mongo/db/pipeline/document_source_mock_test.cpp
+++ b/src/mongo/db/pipeline/document_source_mock_test.cpp
@@ -43,13 +43,23 @@ TEST(DocumentSourceMockTest, OneDoc) {
     ASSERT(source->getNext().isEOF());
 }
 
-TEST(DocumentSourceMockTest, DequeDocuments) {
+TEST(DocumentSourceMockTest, ShouldBeConstructableFromInitializerListOfDocuments) {
     auto source = DocumentSourceMock::create({Document{{"a", 1}}, Document{{"a", 2}}});
     ASSERT_DOCUMENT_EQ(source->getNext().getDocument(), (Document{{"a", 1}}));
     ASSERT_DOCUMENT_EQ(source->getNext().getDocument(), (Document{{"a", 2}}));
     ASSERT(source->getNext().isEOF());
 }
 
+TEST(DocumentSourceMockTest, ShouldBeConstructableFromDequeOfResults) {
+    auto source = DocumentSourceMock::create({Document{{"a", 1}},
+                                              DocumentSource::GetNextResult::makePauseExecution(),
+                                              Document{{"a", 2}}});
+    ASSERT_DOCUMENT_EQ(source->getNext().getDocument(), (Document{{"a", 1}}));
+    ASSERT_TRUE(source->getNext().isPaused());
+    ASSERT_DOCUMENT_EQ(source->getNext().getDocument(), (Document{{"a", 2}}));
+    ASSERT(source->getNext().isEOF());
+}
+
 TEST(DocumentSourceMockTest, StringJSON) {
     auto source = DocumentSourceMock::create("{a : 1}");
     ASSERT_DOCUMENT_EQ(source->getNext().getDocument(), (Document{{"a", 1}}));