Function TestRestrictionBuilding

generator/generator_tests/restriction_test.cpp:137–187

\brief Generates a restriction section, adds it to an empty mwm, loads the restriction section and test loaded restrictions. \param |restrictionPath| comma separated text with restrictions in osm id terms. \param |osmIdsToFeatureIdContent| comma separated text with mapping from osm ids to feature ids.

/// \param |restrictionPath| comma separated text with restrictions in osm id terms.
/// \param |osmIdsToFeatureIdContent| comma separated text with mapping from osm ids to feature ids.
void TestRestrictionBuilding(string const & restrictionPath, string const & osmIdsToFeatureIdContent,
                             unique_ptr<IndexGraph> graph, vector<Restriction> & expectedNotUTurn,
                             vector<RestrictionUTurnForTests> & expectedUTurn)
{
  Platform & platform = GetPlatform();
  string const writableDir = platform.WritableDir();

  string const targetDir = base::JoinPath(writableDir, kTestDir);
  // Building empty mwm.
  LocalCountryFile country(targetDir, CountryFile(kTestMwm), 0 /* version */);
  ScopedDir const scopedDir(kTestDir);
  string const mwmRelativePath = base::JoinPath(kTestDir, kTestMwm + DATA_FILE_EXTENSION);
  ScopedFile const scopedMwm(mwmRelativePath, ScopedFile::Mode::Create);
  BuildEmptyMwm(country);

  // Creating a file with restrictions.
  string const restrictionRelativePath = base::JoinPath(kTestDir, kRestrictionFileName);
  ScopedFile const restrictionScopedFile(restrictionRelativePath, restrictionPath);

  // Creating osm ids to feature ids mapping.
  string const mappingRelativePath = base::JoinPath(kTestDir, kOsmIdsToFeatureIdsName);
  ScopedFile const mappingFile(mappingRelativePath, ScopedFile::Mode::Create);
  string const & osmIdsToFeatureIdFullPath = mappingFile.GetFullPath();
  ReEncodeOsmIdsToFeatureIdsMapping(osmIdsToFeatureIdContent, osmIdsToFeatureIdFullPath);

  string const restrictionFullPath = base::JoinPath(writableDir, restrictionRelativePath);
  string const & mwmFullPath = scopedMwm.GetFullPath();

  // Prepare data to collector.
  auto restrictionCollector =
      std::make_unique<routing_builder::RestrictionCollector>(osmIdsToFeatureIdFullPath, *graph);

  TEST(restrictionCollector->Process(restrictionFullPath), ("Bad restrictions were given."));

  // Adding restriction section to mwm.
  SerializeRestrictions(*restrictionCollector, mwmFullPath);

  // Reading from mwm section and testing restrictions.
  vector<Restriction> restrictionsFromMwm;
  vector<RestrictionUTurnForTests> restrictionsUTurnFromMwm;
  LoadRestrictions(mwmFullPath, restrictionsFromMwm, restrictionsUTurnFromMwm);

  sort(restrictionsFromMwm.begin(), restrictionsFromMwm.end());
  sort(restrictionsUTurnFromMwm.begin(), restrictionsUTurnFromMwm.end());

  sort(expectedNotUTurn.begin(), expectedNotUTurn.end());
  sort(expectedUTurn.begin(), expectedUTurn.end());

  TEST_EQUAL(restrictionsFromMwm, expectedNotUTurn, ());
  TEST_EQUAL(restrictionsUTurnFromMwm, expectedUTurn, ());
}

// 2                 *
//                ↗     ↘