[search] Added matching of edited features by type.

1 files changed, 7 insertions, 5 deletions

diff --git a/indexer/osm_editor.hpp b/indexer/osm_editor.hpp
index 282f9b98ef..dcdf39255a 100644
--- a/indexer/osm_editor.hpp
+++ b/indexer/osm_editor.hpp
@@ -10,6 +10,7 @@
 
 #include "editor/editor_config.hpp"
 #include "editor/editor_notes.hpp"
+#include "editor/editor_storage.hpp"
 #include "editor/xml_feature.hpp"
 
 #include "base/timer.hpp"
@@ -23,11 +24,6 @@
 
 namespace editor
 {
-class StorageBase;
-}  // namespace editor
-
-namespace editor
-{
 namespace testing
 {
 class EditorTest;
@@ -74,9 +70,15 @@ public:
   };
 
   static Editor & Instance();
+
   // Reference to the index will be used in editor functors, it should not be temporary object.
   void SetIndex(Index const & index);
 
+  inline void SetStorageForTesting(unique_ptr<editor::StorageBase> storage)
+  {
+    m_storage = move(storage);
+  }
+
   void SetInvalidateFn(TInvalidateFn const & fn) { m_invalidateFn = fn; }
 
   void LoadMapEdits();