diff options
Diffstat (limited to 'src/mongo/s/query/results_merger_test_fixture.h')
-rw-r--r-- | src/mongo/s/query/results_merger_test_fixture.h | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/src/mongo/s/query/results_merger_test_fixture.h b/src/mongo/s/query/results_merger_test_fixture.h index d653bfe0bfa..10978c11cb7 100644 --- a/src/mongo/s/query/results_merger_test_fixture.h +++ b/src/mongo/s/query/results_merger_test_fixture.h @@ -72,22 +72,22 @@ protected: if (findCmd) { // If there is no '$db', append it. auto cmd = OpMsgRequest::fromDBAndBody(kTestNss.db(), *findCmd).body; - const auto qr = - QueryRequest::makeFromFindCommandForTests(cmd, false /* isExplain */, kTestNss); - if (!qr->getSort().isEmpty()) { - params.setSort(qr->getSort().getOwned()); + const auto findCommand = + query_request_helper::makeFromFindCommandForTests(cmd, kTestNss); + if (!findCommand->getSort().isEmpty()) { + params.setSort(findCommand->getSort().getOwned()); } if (getMoreBatchSize) { params.setBatchSize(getMoreBatchSize); } else { - params.setBatchSize(qr->getBatchSize() - ? boost::optional<std::int64_t>( - static_cast<std::int64_t>(*qr->getBatchSize())) + params.setBatchSize(findCommand->getBatchSize() + ? boost::optional<std::int64_t>(static_cast<std::int64_t>( + *findCommand->getBatchSize())) : boost::none); } - params.setTailableMode(qr->getTailableMode()); - params.setAllowPartialResults(qr->isAllowPartialResults()); + params.setTailableMode(query_request_helper::getTailableMode(*findCommand)); + params.setAllowPartialResults(findCommand->getAllowPartialResults()); } OperationSessionInfoFromClient sessionInfo; |