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
|
#pragma once
#include "search/search_tests_support/test_mwm_builder.hpp"
#include "search/search_tests_support/test_results_matching.hpp"
#include "search/search_tests_support/test_search_engine.hpp"
#include "storage/country_decl.hpp"
#include "storage/country_info_getter.hpp"
#include "indexer/classificator_loader.hpp"
#include "indexer/mwm_set.hpp"
#include "geometry/rect2d.hpp"
#include "platform/country_file.hpp"
#include "platform/local_country_file.hpp"
#include "platform/local_country_file_utils.hpp"
#include "base/logging.hpp"
#include "std/unique_ptr.hpp"
#include "std/vector.hpp"
class Platform;
namespace search
{
class TestWithClassificator
{
public:
TestWithClassificator() { classificator::Load(); }
};
class SearchTest : public TestWithClassificator
{
public:
SearchTest();
~SearchTest();
// Registers country in internal records. Note that physical country
// file may be absent.
void RegisterCountry(string const & name, m2::RectD const & rect);
// Creates a physical country file on a disk, which will be removed
// at the end of the test. |fn| is a delegate that accepts a single
// argument - TestMwmBuilder and adds all necessary features to the
// country file.
//
// *NOTE* when |type| is feature::DataHeader::country, the country
// with |name| will be automatically registered.
template <typename TBuildFn>
MwmSet::MwmId BuildMwm(string const & name, feature::DataHeader::MapType type, TBuildFn && fn)
{
m_files.emplace_back(m_platform.WritableDir(), platform::CountryFile(name), 0 /* version */);
auto & file = m_files.back();
Cleanup(file);
{
tests_support::TestMwmBuilder builder(file, type);
fn(builder);
}
auto result = m_engine.RegisterMap(file);
CHECK_EQUAL(result.second, MwmSet::RegResult::Success, ());
auto const & id = result.first;
if (type == feature::DataHeader::country)
{
if (auto const & info = id.GetInfo())
RegisterCountry(name, info->m_limitRect);
}
return id;
}
inline void SetViewport(m2::RectD const & viewport) { m_viewport = viewport; }
bool ResultsMatch(string const & query,
vector<shared_ptr<tests_support::MatchingRule>> const & rules);
bool ResultsMatch(string const & query, Mode mode,
vector<shared_ptr<tests_support::MatchingRule>> const & rules);
size_t CountFeatures(m2::RectD const & rect);
protected:
static void Cleanup(platform::LocalCountryFile const & map);
Platform & m_platform;
my::ScopedLogLevelChanger m_scopedLog;
vector<platform::LocalCountryFile> m_files;
tests_support::TestSearchEngine m_engine;
m2::RectD m_viewport;
};
} // namespace search
|
