1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/searchlib/query/streaming/hit_iterator_pack.h>
#include <vespa/vespalib/gtest/gtest.h>
using search::streaming::HitIterator;
using search::streaming::HitIteratorPack;
using search::streaming::QueryNodeList;
using search::streaming::QueryTerm;
using search::streaming::QueryNodeResultBase;
using FieldElement = HitIterator::FieldElement;
TEST(HitIteratorPackTest, seek_to_matching_field_element)
{
QueryNodeList qnl;
auto qt = std::make_unique<QueryTerm>(std::unique_ptr<QueryNodeResultBase>(), "7", "", QueryTerm::Type::WORD);
qt->add(11, 0, 10, 0);
qt->add(11, 0, 10, 5);
qt->add(11, 1, 12, 0);
qt->add(11, 1, 12, 0);
qt->add(12, 1, 13, 0);
qt->add(12, 1, 13, 0);
qnl.emplace_back(std::move(qt));
qt = std::make_unique<QueryTerm>(std::unique_ptr<QueryNodeResultBase>(), "8", "", QueryTerm::Type::WORD);
qt->add(2, 0, 4, 0);
qt->add(11, 0, 10, 0);
qt->add(12, 1, 13, 0);
qt->add(12, 2, 14, 0);
qnl.emplace_back(std::move(qt));
HitIteratorPack itr_pack(qnl);
EXPECT_TRUE(itr_pack.all_valid());
EXPECT_TRUE(itr_pack.seek_to_matching_field_element());
EXPECT_EQ(FieldElement(11, 0), itr_pack.get_field_element_ref());
EXPECT_TRUE(itr_pack.seek_to_matching_field_element());
EXPECT_EQ(FieldElement(11, 0), itr_pack.get_field_element_ref());
++itr_pack.get_field_element_ref().second;
EXPECT_TRUE(itr_pack.seek_to_matching_field_element());
EXPECT_EQ(FieldElement(12, 1), itr_pack.get_field_element_ref());
++itr_pack.get_field_element_ref().second;
EXPECT_FALSE(itr_pack.seek_to_matching_field_element());
}
GTEST_MAIN_RUN_ALL_TESTS()
|