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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#include "search/search_tests_support/helpers.hpp"
#include "storage/country_info_getter.hpp"
#include "indexer/scales.hpp"
#include "geometry/mercator.hpp"
#include "geometry/rect2d.hpp"
using namespace std;
namespace search
{
SearchTest::SearchTest()
: m_scopedLog(LDEBUG)
, m_engine(m_dataSource, make_unique<storage::CountryInfoGetterForTesting>(), Engine::Params{})
{
SetViewport(MercatorBounds::FullRect());
}
void SearchTest::RegisterCountry(string const & name, m2::RectD const & rect)
{
auto & infoGetter =
static_cast<storage::CountryInfoGetterForTesting &>(m_engine.GetCountryInfoGetter());
infoGetter.AddCountry(storage::CountryDef(name, rect));
}
bool SearchTest::ResultsMatch(string const & query,
vector<shared_ptr<tests_support::MatchingRule>> const & rules)
{
return ResultsMatch(query, "en" /* locale */, rules);
}
bool SearchTest::ResultsMatch(string const & query, string const & locale,
vector<shared_ptr<tests_support::MatchingRule>> const & rules)
{
tests_support::TestSearchRequest request(m_engine, query, locale, Mode::Everywhere, m_viewport);
request.Run();
return MatchResults(m_dataSource, rules, request.Results());
}
bool SearchTest::ResultsMatch(string const & query, Mode mode,
vector<shared_ptr<tests_support::MatchingRule>> const & rules)
{
tests_support::TestSearchRequest request(m_engine, query, "en", mode, m_viewport);
request.Run();
return MatchResults(m_dataSource, rules, request.Results());
}
bool SearchTest::ResultsMatch(vector<search::Result> const & results, Rules const & rules)
{
return MatchResults(m_dataSource, rules, results);
}
bool SearchTest::ResultsMatch(SearchParams const & params, Rules const & rules)
{
tests_support::TestSearchRequest request(m_engine, params);
request.Run();
return ResultsMatch(request.Results(), rules);
}
bool SearchTest::ResultMatches(search::Result const & result, Rule const & rule)
{
return tests_support::ResultMatches(m_dataSource, rule, result);
}
size_t SearchTest::GetResultsNumber(string const & query, string const & locale)
{
tests_support::TestSearchRequest request(m_engine, query, locale, Mode::Everywhere, m_viewport);
request.Run();
return request.Results().size();
}
unique_ptr<tests_support::TestSearchRequest> SearchTest::MakeRequest(
string const & query, string const & locale /* = "en" */)
{
SearchParams params;
params.m_query = query;
params.m_inputLocale = locale;
params.m_viewport = m_viewport;
params.m_mode = Mode::Everywhere;
params.m_needAddress = true;
params.m_suggestsEnabled = false;
auto request = make_unique<tests_support::TestSearchRequest>(m_engine, params);
request->Run();
return request;
}
size_t SearchTest::CountFeatures(m2::RectD const & rect)
{
size_t count = 0;
auto counter = [&count](const FeatureType & /* ft */) { ++count; };
m_dataSource.ForEachInRect(counter, rect, scales::GetUpperScale());
return count;
}
// static
void SearchTest::OnMwmBuilt(MwmInfo const & info)
{
switch (info.GetType())
{
case MwmInfo::COUNTRY: RegisterCountry(info.GetCountryName(), info.m_bordersRect); break;
case MwmInfo::WORLD: m_engine.LoadCitiesBoundaries(); break;
case MwmInfo::COASTS: break;
}
}
} // namespace search
|
