Repo created

libs/indexer/indexer_tests/data_source_test.cpp

@@ -0,0 +1,157 @@
+#include "testing/testing.hpp"
+
+#include "indexer/data_header.hpp"
+#include "indexer/data_source.hpp"
+#include "indexer/feature_source.hpp"
+#include "indexer/mwm_set.hpp"
+
+#include "coding/internal/file_data.hpp"
+
+#include "platform/country_file.hpp"
+#include "platform/local_country_file.hpp"
+#include "platform/platform.hpp"
+
+#include "base/file_name_utils.hpp"
+#include "base/logging.hpp"
+#include "base/macros.hpp"
+#include "base/scope_guard.hpp"
+#include "base/stl_helpers.hpp"
+
+#include <algorithm>
+#include <string>
+
+namespace data_source_test
+{
+using platform::CountryFile;
+using platform::LocalCountryFile;
+
+class DataSourceTest : public MwmSet::Observer
+{
+public:
+  DataSourceTest() { TEST(m_dataSource.AddObserver(*this), ()); }
+
+  ~DataSourceTest() override { TEST(m_dataSource.RemoveObserver(*this), ()); }
+
+  void ExpectRegistered(platform::LocalCountryFile const & localFile)
+  {
+    AddEvent(m_expected, MwmSet::Event::TYPE_REGISTERED, localFile);
+  }
+
+  void ExpectDeregistered(platform::LocalCountryFile const & localFile)
+  {
+    AddEvent(m_expected, MwmSet::Event::TYPE_DEREGISTERED, localFile);
+  }
+
+  // Checks expectations and clears them.
+  bool CheckExpectations()
+  {
+    bool ok = true;
+    if (m_actual != m_expected)
+    {
+      LOG(LINFO, ("Check failed. Expected:", m_expected, "actual:", m_actual));
+      ok = false;
+    }
+    m_actual.clear();
+    m_expected.clear();
+    return ok;
+  }
+
+  // MwmSet::Observer overrides:
+  void OnMapRegistered(platform::LocalCountryFile const & localFile) override
+  {
+    AddEvent(m_actual, MwmSet::Event::TYPE_REGISTERED, localFile);
+  }
+
+  void OnMapDeregistered(platform::LocalCountryFile const & localFile) override
+  {
+    AddEvent(m_actual, MwmSet::Event::TYPE_DEREGISTERED, localFile);
+  }
+
+protected:
+  template <typename... TArgs>
+  void AddEvent(std::vector<MwmSet::Event> & events, TArgs... args)
+  {
+    events.emplace_back(std::forward<TArgs>(args)...);
+  }
+
+  FrozenDataSource m_dataSource;
+  std::vector<MwmSet::Event> m_expected;
+  std::vector<MwmSet::Event> m_actual;
+};
+
+UNIT_CLASS_TEST(DataSourceTest, Parse)
+{
+  UNUSED_VALUE(m_dataSource.RegisterMap(platform::LocalCountryFile::MakeForTesting("minsk-pass")));
+
+  // Make sure that index is actually parsed.
+  m_dataSource.ForEachInScale([](FeatureType &) { return; }, 15);
+}
+
+UNIT_CLASS_TEST(DataSourceTest, StatusNotifications)
+{
+  std::string const mapsDir = GetPlatform().WritableDir();
+  CountryFile const country("minsk-pass");
+
+  // These two classes point to the same file, but will be considered
+  // by DataSource as distinct files because versions are artificially set
+  // to different numbers.
+  LocalCountryFile const file1(mapsDir, country, 1 /* version */);
+  LocalCountryFile const file2(mapsDir, country, 2 /* version */);
+
+  MwmSet::MwmId id1;
+
+  // Checks that observers are triggered after map registration.
+  {
+    auto result = m_dataSource.RegisterMap(file1);
+    TEST_EQUAL(MwmSet::RegResult::Success, result.second, ());
+
+    id1 = result.first;
+    TEST(id1.IsAlive(), ());
+
+    ExpectRegistered(file1);
+    TEST(CheckExpectations(), ());
+  }
+
+  // Checks that map can't registered twice.
+  {
+    auto result = m_dataSource.RegisterMap(file1);
+    TEST_EQUAL(MwmSet::RegResult::VersionAlreadyExists, result.second, ());
+
+    TEST(result.first.IsAlive(), ());
+    TEST_EQUAL(id1, result.first, ());
+
+    TEST(CheckExpectations(), ());
+  }
+
+  // Checks that observers are notified when map is updated.
+  MwmSet::MwmId id2;
+  {
+    auto result = m_dataSource.RegisterMap(file2);
+    TEST_EQUAL(MwmSet::RegResult::Success, result.second, ());
+
+    id2 = result.first;
+    TEST(id2.IsAlive(), ());
+    TEST_NOT_EQUAL(id1, id2, ());
+
+    ExpectDeregistered(file1);
+    ExpectRegistered(file2);
+    TEST(CheckExpectations(), ());
+  }
+
+  // Tries to deregister a map in presence of an active handle. Map
+  // should be marked "to be removed" but can't be deregistered. After
+  // leaving the inner block the map should be deregistered.
+  {
+    {
+      MwmSet::MwmHandle const handle = m_dataSource.GetMwmHandleByCountryFile(country);
+      TEST(handle.IsAlive(), ());
+
+      TEST(!m_dataSource.DeregisterMap(country), ());
+      TEST(CheckExpectations(), ());
+    }
+
+    ExpectDeregistered(file2);
+    TEST(CheckExpectations(), ());
+  }
+}
+}  // namespace data_source_test