diff options
Diffstat (limited to 'searchlib/src/tests/queryeval')
6 files changed, 19 insertions, 0 deletions
diff --git a/searchlib/src/tests/queryeval/blueprint/blueprint_test.cpp b/searchlib/src/tests/queryeval/blueprint/blueprint_test.cpp index 51164427690..d739151a2c4 100644 --- a/searchlib/src/tests/queryeval/blueprint/blueprint_test.cpp +++ b/searchlib/src/tests/queryeval/blueprint/blueprint_test.cpp @@ -145,6 +145,9 @@ struct MyTerm : SimpleLeafBlueprint { MyTerm(FieldSpecBase field, uint32_t hitEstimate) : SimpleLeafBlueprint(field) { setEstimate(HitEstimate(hitEstimate, false)); } + FlowStats calculate_flow_stats(uint32_t docid_limit) const override { + return default_flow_stats(docid_limit, getState().estimate().estHits, 0); + } SearchIterator::UP createLeafSearch(const search::fef::TermFieldMatchDataArray &, bool) const override { return {}; } diff --git a/searchlib/src/tests/queryeval/filter_search/filter_search_test.cpp b/searchlib/src/tests/queryeval/filter_search/filter_search_test.cpp index ca450c6d712..3e8bc06bfd8 100644 --- a/searchlib/src/tests/queryeval/filter_search/filter_search_test.cpp +++ b/searchlib/src/tests/queryeval/filter_search/filter_search_test.cpp @@ -71,6 +71,7 @@ struct LeafProxy : SimpleLeafBlueprint { : SimpleLeafBlueprint(), child(std::move(child_in)) { init(); } LeafProxy(FieldSpecBase field, std::unique_ptr<Blueprint> child_in) : SimpleLeafBlueprint(field), child(std::move(child_in)) { init(); } + FlowStats calculate_flow_stats(uint32_t) const override { abort(); } SearchIteratorUP createLeafSearch(const TermFieldMatchDataArray &, bool) const override { abort(); } SearchIteratorUP createFilterSearch(bool strict, Constraint constraint) const override { return child->createFilterSearch(strict, constraint); diff --git a/searchlib/src/tests/queryeval/queryeval_test.cpp b/searchlib/src/tests/queryeval/queryeval_test.cpp index c3a9a0d326e..e8f30ba84ff 100644 --- a/searchlib/src/tests/queryeval/queryeval_test.cpp +++ b/searchlib/src/tests/queryeval/queryeval_test.cpp @@ -358,6 +358,12 @@ public: _sc = _a.getSearch(std::make_unique<search::QueryTermSimple>("1", search::QueryTermSimple::Type::WORD), SearchContextParams().useBitVector(true)); } + FlowStats calculate_flow_stats(uint32_t docid_limit) const override { + auto est = _sc->calc_hit_estimate(); + return est.is_unknown() + ? default_flow_stats(0) + : default_flow_stats(docid_limit, est.est_hits(), 0); + } SearchIterator::UP createLeafSearch(const TermFieldMatchDataArray &tfmda, bool strict) const override { diff --git a/searchlib/src/tests/queryeval/simple_phrase/simple_phrase_test.cpp b/searchlib/src/tests/queryeval/simple_phrase/simple_phrase_test.cpp index cbe497d6363..99812900d40 100644 --- a/searchlib/src/tests/queryeval/simple_phrase/simple_phrase_test.cpp +++ b/searchlib/src/tests/queryeval/simple_phrase/simple_phrase_test.cpp @@ -35,6 +35,9 @@ struct MyTerm : public search::queryeval::SimpleLeafBlueprint { { setEstimate(HitEstimate(hits, (hits == 0))); } + FlowStats calculate_flow_stats(uint32_t docid_limit) const override { + return default_flow_stats(docid_limit, getState().estimate().estHits, 0); + } SearchIterator::UP createLeafSearch(const search::fef::TermFieldMatchDataArray &, bool) const override { return {}; } diff --git a/searchlib/src/tests/queryeval/termwise_eval/termwise_eval_test.cpp b/searchlib/src/tests/queryeval/termwise_eval/termwise_eval_test.cpp index 3a10ed6df53..2a3ce66dba6 100644 --- a/searchlib/src/tests/queryeval/termwise_eval/termwise_eval_test.cpp +++ b/searchlib/src/tests/queryeval/termwise_eval/termwise_eval_test.cpp @@ -83,6 +83,9 @@ struct MyBlueprint : SimpleLeafBlueprint { set_allow_termwise_eval(allow_termwise_eval); } ~MyBlueprint() override; + FlowStats calculate_flow_stats(uint32_t docid_limit) const override { + return default_flow_stats(docid_limit, getState().estimate().estHits, 0); + } SearchIterator::UP createLeafSearch(const fef::TermFieldMatchDataArray &, bool strict) const override { diff --git a/searchlib/src/tests/queryeval/weighted_set_term/weighted_set_term_test.cpp b/searchlib/src/tests/queryeval/weighted_set_term/weighted_set_term_test.cpp index 24d62f66714..54663e48060 100644 --- a/searchlib/src/tests/queryeval/weighted_set_term/weighted_set_term_test.cpp +++ b/searchlib/src/tests/queryeval/weighted_set_term/weighted_set_term_test.cpp @@ -344,6 +344,9 @@ struct VerifyMatchData { ++vmd.child_cnt; return std::make_unique<EmptySearch>(); } + FlowStats calculate_flow_stats(uint32_t docid_limit) const override { + return default_flow_stats(docid_limit, 0, 0); + } [[nodiscard]] SearchIteratorUP createFilterSearch(bool strict, FilterConstraint constraint) const override { return create_default_filter(strict, constraint); } |