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
|
#pragma once
#include "generator/generator_tests_support/test_mwm_builder.hpp"
#include "generator/generator_tests_support/test_with_classificator.hpp"
#include "editor/editable_data_source.hpp"
#include "indexer/data_header.hpp"
#include "indexer/feature.hpp"
#include "indexer/mwm_set.hpp"
#include "platform/country_file.hpp"
#include "platform/local_country_file.hpp"
#include "platform/platform.hpp"
#include "base/assert.hpp"
#include <string>
#include <utility>
#include <vector>
namespace generator
{
namespace tests_support
{
class TestWithCustomMwms : public TestWithClassificator
{
public:
~TestWithCustomMwms() override;
// 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 BuildFn>
MwmSet::MwmId BuildMwm(std::string const & name, feature::DataHeader::MapType type, BuildFn && fn)
{
m_files.emplace_back(GetPlatform().WritableDir(), platform::CountryFile(name), 0 /* version */);
auto & file = m_files.back();
Cleanup(file);
{
generator::tests_support::TestMwmBuilder builder(file, type);
fn(builder);
}
auto result = m_dataSource.RegisterMap(file);
CHECK_EQUAL(result.second, MwmSet::RegResult::Success, ());
auto const id = result.first;
auto const info = id.GetInfo();
CHECK(info.get(), ());
OnMwmBuilt(*info);
return id;
}
void DeregisterMap(std::string const & name)
{
auto const file = platform::CountryFile(name);
auto it = find_if(
m_files.begin(), m_files.end(),
[&file](platform::LocalCountryFile const & f) { return f.GetCountryFile() == file; });
if (it == m_files.end())
return;
m_dataSource.DeregisterMap(file);
Cleanup(*it);
m_files.erase(it);
}
template <typename BuildFn>
MwmSet::MwmId BuildWorld(BuildFn && fn)
{
return BuildMwm("testWorld", feature::DataHeader::world, std::forward<BuildFn>(fn));
}
template <typename BuildFn>
MwmSet::MwmId BuildCountry(std::string const & name, BuildFn && fn)
{
return BuildMwm(name, feature::DataHeader::country, std::forward<BuildFn>(fn));
}
protected:
static void Cleanup(platform::LocalCountryFile const & file);
virtual void OnMwmBuilt(MwmInfo const & /* info */) {}
EditableDataSource m_dataSource;
std::vector<platform::LocalCountryFile> m_files;
};
} // namespace tests_support
} // namespace generator
|
