diff options
Diffstat (limited to 'src/mongo/db/pipeline')
4 files changed, 15 insertions, 7 deletions
diff --git a/src/mongo/db/pipeline/document_source_bucket_auto_test.cpp b/src/mongo/db/pipeline/document_source_bucket_auto_test.cpp index b7b50b6527d..c6dcf05e1ea 100644 --- a/src/mongo/db/pipeline/document_source_bucket_auto_test.cpp +++ b/src/mongo/db/pipeline/document_source_bucket_auto_test.cpp @@ -644,7 +644,9 @@ void assertCannotSpillToDisk(const boost::intrusive_ptr<ExpressionContext>& expC {Document{{"a", 0}, {"largeStr", largeStr}}, Document{{"a", 1}, {"largeStr", largeStr}}}); bucketAutoStage->setSource(mock.get()); - ASSERT_THROWS_CODE(bucketAutoStage->getNext(), AssertionException, 16819); + ASSERT_THROWS_CODE(bucketAutoStage->getNext(), + AssertionException, + ErrorCodes::QueryExceededMemoryLimitNoDiskUseAllowed); } TEST_F(BucketAutoTests, ShouldFailIfBufferingTooManyDocuments) { @@ -687,7 +689,9 @@ TEST_F(BucketAutoTests, ShouldCorrectlyTrackMemoryUsageBetweenPauses) { ASSERT_TRUE(bucketAutoStage->getNext().isPaused()); // The next should realize it's used too much memory. - ASSERT_THROWS_CODE(bucketAutoStage->getNext(), AssertionException, 16819); + ASSERT_THROWS_CODE(bucketAutoStage->getNext(), + AssertionException, + ErrorCodes::QueryExceededMemoryLimitNoDiskUseAllowed); } TEST_F(BucketAutoTests, ShouldRoundUpMaximumBoundariesWithGranularitySpecified) { diff --git a/src/mongo/db/pipeline/document_source_group.cpp b/src/mongo/db/pipeline/document_source_group.cpp index 8268f0a1dd1..230162d2f51 100644 --- a/src/mongo/db/pipeline/document_source_group.cpp +++ b/src/mongo/db/pipeline/document_source_group.cpp @@ -483,7 +483,7 @@ DocumentSource::GetNextResult DocumentSourceGroup::initialize() { GetNextResult input = pSource->getNext(); for (; input.isAdvanced(); input = pSource->getNext()) { if (_memoryUsageBytes > _maxMemoryUsageBytes) { - uassert(16945, + uassert(ErrorCodes::QueryExceededMemoryLimitNoDiskUseAllowed, "Exceeded memory limit for $group, but didn't allow external sort." " Pass allowDiskUse:true to opt in.", _allowDiskUse); diff --git a/src/mongo/db/pipeline/document_source_group_test.cpp b/src/mongo/db/pipeline/document_source_group_test.cpp index c3c764c4f6a..0289195c878 100644 --- a/src/mongo/db/pipeline/document_source_group_test.cpp +++ b/src/mongo/db/pipeline/document_source_group_test.cpp @@ -166,7 +166,8 @@ TEST_F(DocumentSourceGroupTest, ShouldErrorIfNotAllowedToSpillToDiskAndResultSet Document{{"_id", 1}, {"largeStr", largeStr}}}); group->setSource(mock.get()); - ASSERT_THROWS_CODE(group->getNext(), AssertionException, 16945); + ASSERT_THROWS_CODE( + group->getNext(), AssertionException, ErrorCodes::QueryExceededMemoryLimitNoDiskUseAllowed); } TEST_F(DocumentSourceGroupTest, ShouldCorrectlyTrackMemoryUsageBetweenPauses) { @@ -197,7 +198,8 @@ TEST_F(DocumentSourceGroupTest, ShouldCorrectlyTrackMemoryUsageBetweenPauses) { ASSERT_TRUE(group->getNext().isPaused()); // The next should realize it's used too much memory. - ASSERT_THROWS_CODE(group->getNext(), AssertionException, 16945); + ASSERT_THROWS_CODE( + group->getNext(), AssertionException, ErrorCodes::QueryExceededMemoryLimitNoDiskUseAllowed); } TEST_F(DocumentSourceGroupTest, ShouldReportSingleFieldGroupKeyAsARename) { diff --git a/src/mongo/db/pipeline/document_source_sort_test.cpp b/src/mongo/db/pipeline/document_source_sort_test.cpp index fc7011c03a3..4e1ee295924 100644 --- a/src/mongo/db/pipeline/document_source_sort_test.cpp +++ b/src/mongo/db/pipeline/document_source_sort_test.cpp @@ -443,7 +443,8 @@ TEST_F(DocumentSourceSortExecutionTest, Document{{"_id", 1}, {"largeStr", largeStr}}}); sort->setSource(mock.get()); - ASSERT_THROWS_CODE(sort->getNext(), AssertionException, 16819); + ASSERT_THROWS_CODE( + sort->getNext(), AssertionException, ErrorCodes::QueryExceededMemoryLimitNoDiskUseAllowed); } TEST_F(DocumentSourceSortExecutionTest, ShouldCorrectlyTrackMemoryUsageBetweenPauses) { @@ -465,7 +466,8 @@ TEST_F(DocumentSourceSortExecutionTest, ShouldCorrectlyTrackMemoryUsageBetweenPa ASSERT_TRUE(sort->getNext().isPaused()); // The next should realize it's used too much memory. - ASSERT_THROWS_CODE(sort->getNext(), AssertionException, 16819); + ASSERT_THROWS_CODE( + sort->getNext(), AssertionException, ErrorCodes::QueryExceededMemoryLimitNoDiskUseAllowed); } } // namespace |