Repo created

generator/generator_tests/gen_mwm_info_tests.cpp

@@ -0,0 +1,135 @@
+#include "testing/testing.hpp"
+
+#include "generator/composite_id.hpp"
+#include "generator/gen_mwm_info.hpp"
+
+#include "coding/file_writer.hpp"
+
+#include "platform/platform_tests_support/scoped_file.hpp"
+
+namespace
+{
+generator::CompositeId const kCid1(base::MakeOsmNode(0), base::MakeOsmNode(0));
+generator::CompositeId const kCid2(base::MakeOsmNode(1), base::MakeOsmNode(1));
+generator::CompositeId const kCid3(base::MakeOsmNode(1), base::MakeOsmNode(2));
+
+uint32_t const kId1 = 0;
+uint32_t const kId2 = 1;
+uint32_t const kId3 = 2;
+
+std::pair<generator::CompositeId, uint32_t> const kPId1(kCid1, kId1);
+std::pair<generator::CompositeId, uint32_t> const kPId2(kCid2, kId2);
+std::pair<generator::CompositeId, uint32_t> const kPId3(kCid3, kId3);
+
+UNIT_TEST(OsmID2FeatureID_AddIds)
+{
+  generator::OsmID2FeatureID mapping;
+  mapping.AddIds(kCid1, kId1);
+  mapping.AddIds(kCid2, kId2);
+  mapping.AddIds(kCid3, kId3);
+
+  std::vector<std::pair<generator::CompositeId, uint32_t>> const answer{kPId1, kPId2, kPId3};
+  size_t index = 0;
+  mapping.ForEach([&](auto const & pair)
+  {
+    TEST_EQUAL(pair, answer[index], ());
+    ++index;
+  });
+  TEST_EQUAL(index, answer.size(), ());
+}
+
+UNIT_TEST(OsmID2FeatureID_GetFeatureId)
+{
+  generator::OsmID2FeatureID mapping;
+  mapping.AddIds(kCid1, kId1);
+  mapping.AddIds(kCid2, kId2);
+  mapping.AddIds(kCid3, kId3);
+  {
+    std::vector<uint32_t> const answer{kId2, kId3};
+    TEST_EQUAL(mapping.GetFeatureIds(kCid2.m_additionalId), answer, ());
+  }
+  {
+    std::vector<uint32_t> const answer;
+    TEST_EQUAL(mapping.GetFeatureIds(base::GeoObjectId()), answer, ());
+  }
+  {
+    std::vector<uint32_t> const answer{
+        kId1,
+    };
+    TEST_EQUAL(mapping.GetFeatureIds(kCid1.m_additionalId), answer, ());
+  }
+  {
+    std::vector<uint32_t> const answer{
+        kId1,
+    };
+    TEST_EQUAL(mapping.GetFeatureIds(kCid1), answer, ());
+  }
+  {
+    std::vector<uint32_t> const answer{kId2};
+    TEST_EQUAL(mapping.GetFeatureIds(kCid2), answer, ());
+  }
+  {
+    std::vector<uint32_t> const answer{
+        kId3,
+    };
+    TEST_EQUAL(mapping.GetFeatureIds(kCid3), answer, ());
+  }
+  {
+    std::vector<uint32_t> const answer;
+    TEST_EQUAL(mapping.GetFeatureIds(generator::CompositeId(base::GeoObjectId())), answer, ());
+  }
+}
+
+UNIT_TEST(OsmID2FeatureID_ReadWrite)
+{
+  using namespace platform::tests_support;
+
+  auto const filename = "test.osm2ft";
+  ScopedFile sf(filename, ScopedFile::Mode::DoNotCreate);
+  {
+    generator::OsmID2FeatureID mapping;
+    mapping.AddIds(kCid1, kId1);
+    mapping.AddIds(kCid2, kId2);
+    mapping.AddIds(kCid3, kId3);
+    FileWriter writer(sf.GetFullPath());
+    mapping.Write(writer);
+  }
+  {
+    generator::OsmID2FeatureID mapping;
+    mapping.ReadFromFile(sf.GetFullPath());
+    std::vector<std::pair<generator::CompositeId, uint32_t>> const answer{kPId1, kPId2, kPId3};
+    size_t index = 0;
+    mapping.ForEach([&](auto const & pair)
+    {
+      TEST_EQUAL(pair, answer[index], ());
+      ++index;
+    });
+    TEST_EQUAL(index, answer.size(), ());
+  }
+}
+
+UNIT_TEST(OsmID2FeatureID_WorkingWithOldFormat)
+{
+  using namespace platform::tests_support;
+
+  auto const filename = "test.osm2ft";
+  std::vector<std::pair<base::GeoObjectId, uint32_t>> answer{{kCid1.m_mainId, kId1}, {kCid2.m_mainId, kId2}};
+  ScopedFile sf(filename, ScopedFile::Mode::DoNotCreate);
+  {
+    FileWriter writer(sf.GetFullPath());
+    rw::WriteVectorOfPOD(writer, answer);
+  }
+  {
+    generator::OsmID2FeatureID mapping;
+    mapping.ReadFromFile(sf.GetFullPath());
+    size_t index = 0;
+    mapping.ForEach([&](auto const & pair)
+    {
+      TEST_EQUAL(pair.first.m_mainId, answer[index].first, ());
+      TEST_EQUAL(pair.second, answer[index].second, ());
+      ++index;
+    });
+    TEST_EQUAL(index, answer.size(), ());
+  }
+}
+}  // namespace