co-maps/libs/search/search_tests_support/test_search_engine.cpp
2025-11-22 13:58:55 +01:00

35 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