diff options
Diffstat (limited to 'src/mongo/db/pipeline/document_source_mock.cpp')
-rw-r--r-- | src/mongo/db/pipeline/document_source_mock.cpp | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/src/mongo/db/pipeline/document_source_mock.cpp b/src/mongo/db/pipeline/document_source_mock.cpp index 86aa733d53d..0f15b96a038 100644 --- a/src/mongo/db/pipeline/document_source_mock.cpp +++ b/src/mongo/db/pipeline/document_source_mock.cpp @@ -63,28 +63,28 @@ void DocumentSourceMock::doDispose() { isDisposed = true; } -intrusive_ptr<DocumentSourceMock> DocumentSourceMock::create(Document doc) { +intrusive_ptr<DocumentSourceMock> DocumentSourceMock::createForTest(Document doc) { return new DocumentSourceMock({std::move(doc)}); } -intrusive_ptr<DocumentSourceMock> DocumentSourceMock::create(deque<GetNextResult> results) { +intrusive_ptr<DocumentSourceMock> DocumentSourceMock::createForTest(deque<GetNextResult> results) { return new DocumentSourceMock(std::move(results)); } -intrusive_ptr<DocumentSourceMock> DocumentSourceMock::create() { +intrusive_ptr<DocumentSourceMock> DocumentSourceMock::createForTest() { return new DocumentSourceMock(deque<GetNextResult>()); } -intrusive_ptr<DocumentSourceMock> DocumentSourceMock::create(const GetNextResult& result) { +intrusive_ptr<DocumentSourceMock> DocumentSourceMock::createForTest(const GetNextResult& result) { deque<GetNextResult> results = {result}; return new DocumentSourceMock(std::move(results)); } -intrusive_ptr<DocumentSourceMock> DocumentSourceMock::create(const char* json) { - return create(Document(fromjson(json))); +intrusive_ptr<DocumentSourceMock> DocumentSourceMock::createForTest(const char* json) { + return createForTest(Document(fromjson(json))); } -intrusive_ptr<DocumentSourceMock> DocumentSourceMock::create( +intrusive_ptr<DocumentSourceMock> DocumentSourceMock::createForTest( const std::initializer_list<const char*>& jsons) { deque<GetNextResult> results; for (auto&& json : jsons) { @@ -93,6 +93,11 @@ intrusive_ptr<DocumentSourceMock> DocumentSourceMock::create( return new DocumentSourceMock(std::move(results)); } +boost::intrusive_ptr<DocumentSourceMock> DocumentSourceMock::create( + const boost::intrusive_ptr<ExpressionContext>& expCtx) { + return new DocumentSourceMock({}, expCtx); +} + DocumentSource::GetNextResult DocumentSourceMock::getNext() { invariant(!isDisposed); invariant(!isDetachedFromOpCtx); |