Repo created

libs/search/search_tests_support/test_results_matching.hpp

@@ -0,0 +1,85 @@
+#pragma once
+
+#include "search/result.hpp"
+
+#include "indexer/mwm_set.hpp"
+
+#include <initializer_list>
+#include <memory>
+#include <string>
+#include <utility>
+#include <vector>
+
+class FeatureType;
+class DataSource;
+
+namespace generator
+{
+namespace tests_support
+{
+class TestFeature;
+}
+}  // namespace generator
+
+namespace search
+{
+namespace tests_support
+{
+class MatchingRule
+{
+public:
+  virtual ~MatchingRule() = default;
+
+  virtual bool Matches(FeatureType & feature) const = 0;
+  virtual std::string ToString() const = 0;
+};
+
+class ExactMatchingRule : public MatchingRule
+{
+public:
+  ExactMatchingRule(MwmSet::MwmId const & mwmId, generator::tests_support::TestFeature const & feature);
+
+  // MatchingRule overrides:
+  bool Matches(FeatureType & feature) const override;
+  std::string ToString() const override;
+
+private:
+  MwmSet::MwmId m_mwmId;
+  generator::tests_support::TestFeature const & m_feature;
+};
+
+class AlternativesMatchingRule : public MatchingRule
+{
+public:
+  AlternativesMatchingRule(std::vector<std::shared_ptr<MatchingRule>> && rules);
+
+  // MatchingRule overrides:
+  bool Matches(FeatureType & feature) const override;
+  std::string ToString() const override;
+
+private:
+  std::vector<std::shared_ptr<MatchingRule>> m_rules;
+};
+
+template <typename... Args>
+std::shared_ptr<MatchingRule> ExactMatch(Args &&... args)
+{
+  return std::make_shared<ExactMatchingRule>(std::forward<Args>(args)...);
+}
+
+inline std::shared_ptr<MatchingRule> AlternativesMatch(std::vector<std::shared_ptr<MatchingRule>> && rules)
+{
+  return std::make_shared<AlternativesMatchingRule>(std::move(rules));
+}
+
+bool MatchResults(DataSource const & dataSource, std::vector<std::shared_ptr<MatchingRule>> rules,
+                  std::vector<search::Result> const & actual);
+bool MatchResults(DataSource const & dataSource, std::vector<std::shared_ptr<MatchingRule>> rules,
+                  search::Results const & actual);
+bool ResultMatches(DataSource const & dataSource, std::shared_ptr<MatchingRule> rule, search::Result const & result);
+bool AlternativeMatch(DataSource const & dataSource, std::vector<std::vector<std::shared_ptr<MatchingRule>>> rules,
+                      std::vector<search::Result> const & results);
+
+std::string DebugPrint(MatchingRule const & rule);
+}  // namespace tests_support
+}  // namespace search