[indexer] Test features retrieval from expected zoom level.

2 files changed, 10 insertions, 0 deletions

diff --git a/generator/generator_tests_support/test_feature.cpp b/generator/generator_tests_support/test_feature.cpp
index 5b6f4831a7..1ab2e297d8 100644
--- a/generator/generator_tests_support/test_feature.cpp
+++ b/generator/generator_tests_support/test_feature.cpp
@@ -313,6 +313,14 @@ string TestPOI::ToDebugString() const
   return os.str();
 }
 
+feature::TypesHolder TestPOI::GetTypes() const
+{
+  feature::TypesHolder types;
+  for (auto const path : m_types)
+    types.Add(classif().GetTypeByPath(path));
+  return types;
+}
+
 // TestMultilingualPOI -----------------------------------------------------------------------------
 TestMultilingualPOI::TestMultilingualPOI(m2::PointD const & center, string const & defaultName,
                                          map<string, string> const & multilingualNames)
diff --git a/generator/generator_tests_support/test_feature.hpp b/generator/generator_tests_support/test_feature.hpp
index 7591cc44a5..e9fbc7b01c 100644
--- a/generator/generator_tests_support/test_feature.hpp
+++ b/generator/generator_tests_support/test_feature.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include "indexer/feature_data.hpp"
 #include "indexer/feature_decl.hpp"
 #include "indexer/feature_meta.hpp"
 #include "indexer/mwm_set.hpp"
@@ -156,6 +157,7 @@ public:
   void Serialize(FeatureBuilder1 & fb) const override;
   std::string ToDebugString() const override;
 
+  feature::TypesHolder GetTypes() const;
   void SetHouseNumber(std::string const & houseNumber) { m_houseNumber = houseNumber; }
   void SetStreetName(std::string const & name) { m_streetName = name; }
   void SetTypes(std::vector<std::vector<std::string>> const & types) { m_types = types; }