mirror of
https://codeberg.org/comaps/comaps
synced 2026-01-09 05:44:34 +00:00
Organic Maps sources as of 02.04.2025 (fad26bbf22ac3da75e01e62aa01e5c8e11861005)
To expand with full Organic Maps and Maps.ME commits history run: git remote add om-historic [om-historic.git repo url] git fetch --tags om-historic git replace squashed-history historic-commits
This commit is contained in:
135
search/search_tests_support/test_search_request.cpp
Normal file
135
search/search_tests_support/test_search_request.cpp
Normal file
@@ -0,0 +1,135 @@
|
||||
#include "search/search_tests_support/test_search_request.hpp"
|
||||
|
||||
#include "search/search_tests_support/test_search_engine.hpp"
|
||||
|
||||
#include "base/assert.hpp"
|
||||
|
||||
#include <functional>
|
||||
|
||||
namespace search
|
||||
{
|
||||
namespace tests_support
|
||||
{
|
||||
using namespace std::chrono;
|
||||
using namespace std;
|
||||
|
||||
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;
|
||||
m_params.m_useDebugInfo = true;
|
||||
|
||||
SetUpCallbacks();
|
||||
SetUpResultParams();
|
||||
}
|
||||
|
||||
TestSearchRequest::TestSearchRequest(TestSearchEngine & engine, SearchParams const & params)
|
||||
: m_engine(engine), m_params(params)
|
||||
{
|
||||
m_params.m_useDebugInfo = true;
|
||||
|
||||
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;
|
||||
m_params.m_useDebugInfo = true;
|
||||
|
||||
SetUpResultParams();
|
||||
}
|
||||
|
||||
void TestSearchRequest::Run()
|
||||
{
|
||||
Start();
|
||||
Wait();
|
||||
}
|
||||
|
||||
TestSearchRequest::TimeDurationT 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);
|
||||
m_results.assign(results.begin(), results.end());
|
||||
if (results.IsEndMarker())
|
||||
{
|
||||
m_done = true;
|
||||
m_endTime = m_timer.TimeElapsed();
|
||||
m_cv.notify_one();
|
||||
}
|
||||
}
|
||||
|
||||
void TestSearchRequest::SetCustomOnResults(SearchParams::OnResults const & onResults)
|
||||
{
|
||||
m_params.m_onResults = onResults;
|
||||
}
|
||||
} // namespace tests_support
|
||||
} // namespace search
|
||||
Reference in New Issue
Block a user