[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 118 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_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");
+
+  EXPECT_FALSE(router_.GetPresentationRoute(presentation_id1));
+
+  router_.RegisterPresentationRoute(presentation_id1, route1);
+  router_.RegisterPresentationRoute(presentation_id2, route2);
+
+  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));
+
+  router_.UnregisterPresentationRoute(presentation_id1);
+
+  EXPECT_FALSE(router_.GetPresentationRoute(presentation_id1));
+}
+
 }  // namespace media_router