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

chrome/browser/media/router/media_router_base_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/bind.h"
 #include "base/test/mock_callback.h"
 #include "chrome/browser/media/router/mock_media_router.h"
 #include "chrome/browser/media/router/test_helper.h"
 #include "content/public/common/presentation_session.h"
 #include "content/public/test/test_browser_thread_bundle.h"
@@ skipping 21 matching lines @@
         route_id, callback);
   }
 
   void OnIncognitoProfileShutdown() override {
     MediaRouterBase::OnIncognitoProfileShutdown();
   }
 
   std::vector<MediaRoute> GetCurrentRoutes() const override {
     return MediaRouterBase::GetCurrentRoutes();
   }
+
+  const MediaRoute* GetPresentationRoute(
+      const std::string& presentation_id) const override {
+    return MediaRouterBase::GetPresentationRoute(presentation_id);
+  }
+
+  void SetPresentationId(const MediaRoute::Id& route_id,
+                         const std::string& presentation_id) override {
+    MediaRouterBase::SetPresentationId(route_id, presentation_id);
+  }
 };
 
 class MediaRouterBaseTest : public testing::Test {
  public:
   void SetUp() override {
     EXPECT_CALL(router_, RegisterMediaRoutesObserver(_))
         .WillOnce(SaveArg<0>(&routes_observer_));
     router_.Initialize();
   }
 
@@ skipping 77 matching lines @@
   std::vector<MediaRoute> current_routes = router_.GetCurrentRoutes();
   ASSERT_EQ(current_routes.size(), 2u);
   EXPECT_TRUE(current_routes[0].Equals(route1));
   EXPECT_TRUE(current_routes[1].Equals(route2));
 
   routes_observer_->OnRoutesUpdated(std::vector<MediaRoute>(),
                                     std::vector<MediaRoute::Id>());
   EXPECT_TRUE(router_.GetCurrentRoutes().empty());
 }
 
+TEST_F(MediaRouterBaseTest, GetPresentationRoute) {
+  MediaSource source1("source_1");
+  MediaSource source2("source_2");
+  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));
+}
+
+TEST_F(MediaRouterBaseTest, OnRoutes) {
+  MediaSource source1("source_1");
+  MediaSource source2("source_2");
+  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");
+  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);
+  router_.SetPresentationId(route1.media_route_id(), presentation_id1);
+  router_.SetPresentationId(route2.media_route_id(), presentation_id2);
+  router_.SetPresentationId("route3", presentation_id3);
+
+  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));
+}
+
+TEST_F(MediaRouterBaseTest, OnRoutesUpdated) {
+  MediaSource source1("source_1");
+  MediaSource source2("source_2");
+  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");
+  std::string presentation_id4("presentation_id4");
+
+  std::vector<MediaRoute> routes1 = {route1, route2};
+  std::vector<MediaRoute::Id> joinable_route_ids = {"route_1"};
+  routes_observer_->OnRoutesUpdated(routes1, joinable_route_ids);
+  router_.SetPresentationId(route2.media_route_id(), presentation_id2);
+
+  std::vector<MediaRoute> routes2 = {route1, route2};
+  routes_observer_->OnRoutesUpdated(routes2, joinable_route_ids);
+
+  // Preserve old presentation id.
+  auto* actual_route2 = router_.GetPresentationRoute(presentation_id2);
+  EXPECT_TRUE(actual_route2 && actual_route2->Equals(route2));
+
+  // Use new presentation id.
+  route1.set_presentation_id(presentation_id3);
+  route2.set_presentation_id(presentation_id4);
+
+  std::vector<MediaRoute> routes3 = {route1, route2};
+  routes_observer_->OnRoutesUpdated(routes3, joinable_route_ids);
+
+  EXPECT_FALSE(router_.GetPresentationRoute(presentation_id2));
+  auto* actual_route4 = router_.GetPresentationRoute(presentation_id4);
+  EXPECT_TRUE(actual_route4 && actual_route4->Equals(route2));
+}
+
 }  // namespace media_router