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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
#include "search/search_tests_support/test_search_request.hpp"
#include "search/search_tests_support/test_search_engine.hpp"
#include "geometry/latlon.hpp"
#include "geometry/mercator.hpp"
#include "base/assert.hpp"
#include "base/logging.hpp"
#include <functional>
using namespace std::chrono;
using namespace std;
namespace search
{
namespace tests_support
{
TestSearchRequest::TestSearchRequest(TestSearchEngine & engine, string const & query,
string const & locale, Mode mode, m2::RectD const & viewport)
: m_engine(engine)
{
m_params.m_query = query;
m_params.m_inputLocale = locale;
m_params.m_viewport = viewport;
m_params.m_mode = mode;
SetUpCallbacks();
SetUpResultParams();
}
TestSearchRequest::TestSearchRequest(TestSearchEngine & engine, SearchParams const & params)
: m_engine(engine), m_params(params)
{
SetUpCallbacks();
}
TestSearchRequest::TestSearchRequest(TestSearchEngine & engine, string const & query,
string const & locale, Mode mode, m2::RectD const & viewport,
SearchParams::OnStarted const & onStarted,
SearchParams::OnResults const & onResults)
: m_engine(engine)
{
m_params.m_query = query;
m_params.m_inputLocale = locale;
m_params.m_viewport = viewport;
m_params.m_mode = mode;
m_params.m_onStarted = onStarted;
m_params.m_onResults = onResults;
SetUpResultParams();
}
void TestSearchRequest::Run()
{
Start();
Wait();
}
steady_clock::duration TestSearchRequest::ResponseTime() const
{
lock_guard<mutex> lock(m_mu);
CHECK(m_done, ("This function may be called only when request is processed."));
return m_endTime - m_startTime;
}
vector<search::Result> const & TestSearchRequest::Results() const
{
lock_guard<mutex> lock(m_mu);
CHECK(m_done, ("This function may be called only when request is processed."));
return m_results;
}
void TestSearchRequest::Start()
{
m_engine.Search(m_params);
}
void TestSearchRequest::Wait()
{
unique_lock<mutex> lock(m_mu);
m_cv.wait(lock, [this]() { return m_done; });
}
void TestSearchRequest::SetUpCallbacks()
{
m_params.m_onStarted = bind(&TestSearchRequest::OnStarted, this);
m_params.m_onResults = bind(&TestSearchRequest::OnResults, this, placeholders::_1);
}
void TestSearchRequest::SetUpResultParams()
{
switch (m_params.m_mode)
{
case Mode::Everywhere:
m_params.m_needAddress = true;
m_params.m_suggestsEnabled = false;
m_params.m_needHighlighting = true;
break;
case Mode::Viewport: // fallthrough
case Mode::Downloader: // fallthrough
case Mode::Bookmarks:
m_params.m_needAddress = false;
m_params.m_suggestsEnabled = false;
m_params.m_needHighlighting = false;
break;
case Mode::Count: CHECK(false, ()); break;
}
}
void TestSearchRequest::OnStarted()
{
lock_guard<mutex> lock(m_mu);
m_startTime = m_timer.TimeElapsed();
}
void TestSearchRequest::OnResults(search::Results const & results)
{
lock_guard<mutex> lock(m_mu);
if (results.IsEndMarker())
{
m_done = true;
m_endTime = m_timer.TimeElapsed();
m_cv.notify_one();
}
else
{
m_results.assign(results.begin(), results.end());
}
}
void TestSearchRequest::SetCustomOnResults(SearchParams::OnResults const & onResults)
{
m_params.m_onResults = onResults;
}
} // namespace tests_support
} // namespace search
|
