Repo created

libs/search/search_tests_support/test_search_engine.cpp

@@ -0,0 +1,35 @@
+#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