[Media Router] Add a presentation id to MediaRoute mapping in the MR

chrome/browser/media/router/media_router_base_unittest.cc

Index: chrome/browser/media/router/media_router_base_unittest.cc
diff --git a/chrome/browser/media/router/media_router_base_unittest.cc b/chrome/browser/media/router/media_router_base_unittest.cc
index 4b942cfcd1f2134bbe52d1fcac6e60c930fb5b54..ac3a3b6f8bede8f285962ba285aa602617c4a018 100644
--- a/chrome/browser/media/router/media_router_base_unittest.cc
+++ b/chrome/browser/media/router/media_router_base_unittest.cc
@@ -39,6 +39,11 @@ class MockMediaRouterBase : public MockMediaRouter {
   std::vector<MediaRoute> GetCurrentRoutes() const override {
     return MediaRouterBase::GetCurrentRoutes();
   }
+
+  const MediaRoute* GetPresentationRoute(
+      const std::string& presentation_id) const override {
+    return MediaRouterBase::GetPresentationRoute(presentation_id);
+  }
 };
 
 class MediaRouterBaseTest : public testing::Test {
@@ -136,4 +141,33 @@ TEST_F(MediaRouterBaseTest, GetCurrentRoutes) {
   EXPECT_TRUE(router_.GetCurrentRoutes().empty());
 }
 
+TEST_F(MediaRouterBaseTest, GetPresentationRoute) {
+  MediaSource source1("source_1");
+  MediaSource source2("source_1");
+  MediaRoute route1("route_1", source1, "sink_1", "", false, "", false);
+  MediaRoute route2("route_2", source2, "sink_2", "", true, "", false);
+  std::string presentation_id1("presentation_id1");
+  std::string presentation_id2("presentation_id2");
+  std::string presentation_id3("presentation_id3");
+  route1.set_presentation_id(presentation_id1);
+  route2.set_presentation_id(presentation_id2);
+  std::vector<MediaRoute> routes = {route1, route2};
+  std::vector<MediaRoute::Id> joinable_route_ids = {"route_1"};
+
+  EXPECT_FALSE(router_.GetPresentationRoute(presentation_id1));
+
+  routes_observer_->OnRoutesUpdated(routes, joinable_route_ids);
+  auto* actual_route1 = router_.GetPresentationRoute(presentation_id1);
+  auto* actual_route2 = router_.GetPresentationRoute(presentation_id2);
+
+  EXPECT_TRUE(actual_route1 && actual_route1->Equals(route1));
+  EXPECT_TRUE(actual_route2 && actual_route2->Equals(route2));
+  EXPECT_FALSE(router_.GetPresentationRoute(presentation_id3));
+
+  routes.erase(routes.begin());
+  routes_observer_->OnRoutesUpdated(routes, joinable_route_ids);
+
+  EXPECT_FALSE(router_.GetPresentationRoute(presentation_id1));
+}
+
 }  // namespace media_router