diff options
author | Tor Egge <Tor.Egge@online.no> | 2022-09-15 16:04:04 +0200 |
---|---|---|
committer | Tor Egge <Tor.Egge@online.no> | 2022-09-15 16:04:04 +0200 |
commit | a5fc7e1e48f27b6385b1ee6849eb664f2cfa267c (patch) | |
tree | 58ccb86797a4cf6dd54ce503b41940ae734ae7da /searchsummary/src/tests/docsummary | |
parent | 88ea0524a2af51f997c87d616315fa6e0562b72f (diff) |
Adjust api for DocsumWriter and DocsumFieldWriter:
Pass reference to state instead of pointer.
Drop unused ResType argument.
Diffstat (limited to 'searchsummary/src/tests/docsummary')
6 files changed, 20 insertions, 7 deletions
diff --git a/searchsummary/src/tests/docsummary/attribute_combiner/attribute_combiner_test.cpp b/searchsummary/src/tests/docsummary/attribute_combiner/attribute_combiner_test.cpp index 8bf3db4d112..005fed41838 100644 --- a/searchsummary/src/tests/docsummary/attribute_combiner/attribute_combiner_test.cpp +++ b/searchsummary/src/tests/docsummary/attribute_combiner/attribute_combiner_test.cpp @@ -91,7 +91,7 @@ AttributeCombinerTest::assertWritten(const vespalib::string &exp_slime_as_json, { vespalib::Slime act; vespalib::slime::SlimeInserter inserter(act); - writer->insertField(docId, nullptr, &state, search::docsummary::RES_JSONSTRING, inserter); + writer->insertField(docId, nullptr, state, inserter); SlimeValue exp(exp_slime_as_json); EXPECT_EQ(exp.slime, act); diff --git a/searchsummary/src/tests/docsummary/attributedfw/attributedfw_test.cpp b/searchsummary/src/tests/docsummary/attributedfw/attributedfw_test.cpp index e9d00629d6f..bba3a5ab506 100644 --- a/searchsummary/src/tests/docsummary/attributedfw/attributedfw_test.cpp +++ b/searchsummary/src/tests/docsummary/attributedfw/attributedfw_test.cpp @@ -69,7 +69,7 @@ public: void expect_field(const vespalib::string& exp_slime_as_json, uint32_t docid) { vespalib::Slime act; vespalib::slime::SlimeInserter inserter(act); - _writer->insertField(docid, nullptr, &_state, search::docsummary::RES_JSONSTRING, inserter); + _writer->insertField(docid, nullptr, _state, inserter); SlimeValue exp(exp_slime_as_json); EXPECT_EQ(exp.slime, act); diff --git a/searchsummary/src/tests/docsummary/document_id_dfw/document_id_dfw_test.cpp b/searchsummary/src/tests/docsummary/document_id_dfw/document_id_dfw_test.cpp index 1c4e201e745..4819c34272c 100644 --- a/searchsummary/src/tests/docsummary/document_id_dfw/document_id_dfw_test.cpp +++ b/searchsummary/src/tests/docsummary/document_id_dfw/document_id_dfw_test.cpp @@ -5,6 +5,8 @@ #include <vespa/document/fieldvalue/document.h> #include <vespa/document/repo/configbuilder.h> #include <vespa/document/repo/documenttyperepo.h> +#include <vespa/searchlib/common/matching_elements.h> +#include <vespa/searchsummary/docsummary/docsumstate.h> #include <vespa/searchsummary/docsummary/docsum_store_document.h> #include <vespa/searchsummary/docsummary/document_id_dfw.h> #include <vespa/searchsummary/docsummary/resultclass.h> @@ -20,8 +22,12 @@ using document::DocumentType; using document::DocumentTypeRepo; using document::config_builder::DocumenttypesConfigBuilderHelper; using document::config_builder::Struct; +using search::MatchingElements; +using search::MatchingElementsFields; using search::docsummary::DocsumStoreDocument; using search::docsummary::DocumentIdDFW; +using search::docsummary::GetDocsumsState; +using search::docsummary::GetDocsumsStateCallback; using search::docsummary::IDocsumStoreDocument; using search::docsummary::ResultClass; using search::docsummary::ResultConfig; @@ -47,6 +53,12 @@ make_doc_type_repo() return std::make_unique<const DocumentTypeRepo>(builder.config()); } +struct MyGetDocsumsStateCallback : GetDocsumsStateCallback { + virtual void FillSummaryFeatures(GetDocsumsState&) override {} + virtual void FillRankFeatures(GetDocsumsState&) override {} + std::unique_ptr<MatchingElements> fill_matching_elements(const MatchingElementsFields &) override { abort(); } +}; + class DocumentIdDFWTest : public ::testing::Test { vespalib::string _field_name; @@ -96,7 +108,9 @@ DocumentIdDFWTest::write(const IDocsumStoreDocument* doc) Cursor & docsum = top_inserter.insertObject(); ObjectInserter field_inserter(docsum, _field_name_view); DocumentIdDFW writer; - writer.insertField(0, doc, nullptr, search::docsummary::RES_LONG_STRING, field_inserter); + MyGetDocsumsStateCallback callback; + GetDocsumsState state(callback); + writer.insertField(0, doc, state, field_inserter); return slime; } diff --git a/searchsummary/src/tests/docsummary/matched_elements_filter/matched_elements_filter_test.cpp b/searchsummary/src/tests/docsummary/matched_elements_filter/matched_elements_filter_test.cpp index 519961dedb6..8ac37ae76eb 100644 --- a/searchsummary/src/tests/docsummary/matched_elements_filter/matched_elements_filter_test.cpp +++ b/searchsummary/src/tests/docsummary/matched_elements_filter/matched_elements_filter_test.cpp @@ -218,7 +218,7 @@ private: Slime slime; SlimeInserter inserter(slime); - writer->insertField(doc_id, doc.get(), &state, ResType::RES_JSONSTRING, inserter); + writer->insertField(doc_id, doc.get(), state, inserter); return slime; } diff --git a/searchsummary/src/tests/docsummary/positionsdfw_test.cpp b/searchsummary/src/tests/docsummary/positionsdfw_test.cpp index f2e949cbddf..b4e82ed2aa8 100644 --- a/searchsummary/src/tests/docsummary/positionsdfw_test.cpp +++ b/searchsummary/src/tests/docsummary/positionsdfw_test.cpp @@ -132,14 +132,13 @@ void checkWritePositionField(AttrType &attr, MyAttributeManager attribute_man(attr); PositionsDFW::UP writer = PositionsDFW::create(attr.getName().c_str(), &attribute_man, false); ASSERT_TRUE(writer.get()); - ResType res_type = RES_JSONSTRING; MyGetDocsumsStateCallback callback; GetDocsumsState state(callback); state._attributes.push_back(&attr); vespalib::Slime target; vespalib::slime::SlimeInserter inserter(target); - writer->insertField(doc_id, &state, res_type, inserter); + writer->insertField(doc_id, state, inserter); test::SlimeValue expected(expect_json); EXPECT_EQUAL(expected.slime, target); diff --git a/searchsummary/src/tests/docsummary/slime_summary/slime_summary_test.cpp b/searchsummary/src/tests/docsummary/slime_summary/slime_summary_test.cpp index cbde3d77b4a..5f2f1578f66 100644 --- a/searchsummary/src/tests/docsummary/slime_summary/slime_summary_test.cpp +++ b/searchsummary/src/tests/docsummary/slime_summary/slime_summary_test.cpp @@ -55,7 +55,7 @@ struct DocsumFixture : IDocsumStore, GetDocsumsStateCallback { Slime slimeOut; SlimeInserter inserter(slimeOut); auto rci = writer->resolveClassInfo(state._args.getResultClassName()); - writer->insertDocsum(rci, 1u, &state, this, inserter); + writer->insertDocsum(rci, 1u, state, this, inserter); vespalib::SmartBuffer buf(4_Ki); BinaryFormat::encode(slimeOut, buf); EXPECT_GREATER(BinaryFormat::decode(buf.obtain(), slime), 0u); |