diff --git a/partners_api/partners_api_tests/taxi_engine_tests.cpp b/partners_api/partners_api_tests/taxi_engine_tests.cpp index 1a7489e973..630de3601a 100644 --- a/partners_api/partners_api_tests/taxi_engine_tests.cpp +++ b/partners_api/partners_api_tests/taxi_engine_tests.cpp @@ -34,6 +34,14 @@ public: std::string GetCityName(ms::LatLon const & latlon) override { return "Odessa"; } }; +class BulgariaSofiaDelegate : public taxi::Delegate +{ +public: + storage::TCountriesVec GetCountryIds(ms::LatLon const & latlon) override { return {"Bulgaria"}; } + + std::string GetCityName(ms::LatLon const & latlon) override { return "Sofia"; } +}; + class UsaDelegate : public taxi::Delegate { public: @@ -364,6 +372,11 @@ UNIT_TEST(TaxiEngine_GetProvidersAtPos) engine.SetDelegate(my::make_unique()); providers = engine.GetProvidersAtPos(latlon); TEST_EQUAL(providers.size(), 1, ()); + TEST_EQUAL(providers[0], taxi::Provider::Type::Yandex, ()); + + engine.SetDelegate(my::make_unique()); + providers = engine.GetProvidersAtPos(latlon); + TEST_EQUAL(providers.size(), 1, ()); TEST_EQUAL(providers[0], taxi::Provider::Type::Uber, ()); engine.SetDelegate(my::make_unique());