36 lines
980 B
C++
36 lines
980 B
C++
|
#include "search/search_tests_support/test_search_engine.hpp"
|
||
|
|
|
||
|
|
#include "indexer/categories_holder.hpp"
|
||
|
|
|
||
|
|
#include "storage/storage.hpp"
|
||
|
|
|
||
|
|
#include "platform/platform.hpp"
|
||
|
|
|
||
|
|
#include <utility>
|
||
|
|
|
||
|
|
namespace search
|
||
|
|
{
|
||
|
|
namespace tests_support
|
||
|
|
{
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
TestSearchEngine::TestSearchEngine(DataSource & dataSource, Engine::Params const & params, bool mockCountryInfo)
|
||
|
|
: m_infoGetter(mockCountryInfo ? make_unique<storage::CountryInfoGetterForTesting>()
|
||
|
|
: storage::CountryInfoReader::CreateCountryInfoGetter(GetPlatform()))
|
||
|
|
, m_engine(dataSource, GetDefaultCategories(), *m_infoGetter, params)
|
||
|
|
{}
|
||
|
|
|
||
|
|
void TestSearchEngine::InitAffiliations()
|
||
|
|
{
|
||
|
|
storage::Storage storage;
|
||
|
|
m_affiliations = *storage.GetAffiliations();
|
||
|
|
m_infoGetter->SetAffiliations(&m_affiliations);
|
||
|
|
}
|
||
|
|
|
||
|
|
weak_ptr<ProcessorHandle> TestSearchEngine::Search(SearchParams const & params)
|
||
|
|
{
|
||
|
|
return m_engine.Search(params);
|
||
|
|
}
|
||
|
|
} // namespace tests_support
|
||
|
|
} // namespace search
|