SERVER-10026 update dbtests for keeping invalidated docs in query

1 files changed, 3 insertions, 3 deletions

diff --git a/src/mongo/dbtests/query_stage_sort.cpp b/src/mongo/dbtests/query_stage_sort.cpp
index 23842445255..2d432215612 100644
--- a/src/mongo/dbtests/query_stage_sort.cpp
+++ b/src/mongo/dbtests/query_stage_sort.cpp
@@ -294,7 +294,7 @@ namespace QueryStageSortTests {
             }
             ss->recoverFromYield();
 
-            // After invalidating all our data, we have nothing left to sort.
+            // Invalidation of data in the sort stage fetches it but passes it through.
             int count = 0;
             while (!ss->isEOF()) {
                 WorkingSetID id;
@@ -306,8 +306,8 @@ namespace QueryStageSortTests {
                 ++count;
             }
 
-            // Therefore, we expect an empty result set from running the sort stage to completion.
-            ASSERT_EQUALS(0, count);
+            // Returns all docs.
+            ASSERT_EQUALS(limit() ? limit() : numObj(), count);
         }
     };