SERVER-24153 Make DocumentSourceMock hold GetNextResults

This is prep work for adding tests that all stages can correctly handle
a kPauseExecution result.

1 files changed, 2 insertions, 1 deletions

diff --git a/src/mongo/db/pipeline/document_source_sort_test.cpp b/src/mongo/db/pipeline/document_source_sort_test.cpp
index ae409910fd0..1d54a9a5618 100644
--- a/src/mongo/db/pipeline/document_source_sort_test.cpp
+++ b/src/mongo/db/pipeline/document_source_sort_test.cpp
@@ -55,6 +55,7 @@ bool isMongos() {
 namespace {
 
 using boost::intrusive_ptr;
+using std::deque;
 using std::string;
 using std::vector;
 
@@ -184,7 +185,7 @@ TEST_F(DocumentSourceSortTest, OutputSort) {
 
 class DocumentSourceSortExecutionTest : public DocumentSourceSortTest {
 public:
-    void checkResults(std::deque<Document> inputDocs,
+    void checkResults(deque<DocumentSource::GetNextResult> inputDocs,
                       BSONObj sortSpec,
                       string expectedResultSetString) {
         createSort(sortSpec);