Non-Local Join for Media Router and Presentation API

chrome/browser/ui/webui/media_router/media_router_webui_message_handler_unittest.cc

Index: chrome/browser/ui/webui/media_router/media_router_webui_message_handler_unittest.cc
diff --git a/chrome/browser/ui/webui/media_router/media_router_webui_message_handler_unittest.cc b/chrome/browser/ui/webui/media_router/media_router_webui_message_handler_unittest.cc
index d3873ed369ee482d03b9f12e6844ccb443449849..b3be64292ade0a5bd3baaa53667527b77de09e34 100644
--- a/chrome/browser/ui/webui/media_router/media_router_webui_message_handler_unittest.cc
+++ b/chrome/browser/ui/webui/media_router/media_router_webui_message_handler_unittest.cc
@@ -119,16 +119,19 @@ TEST_F(MediaRouterWebUIMessageHandlerTest, UpdateRoutes) {
   MediaRoute::Id route_id("routeId123");
   MediaSink::Id sink_id("sinkId123");
   MediaSink sink(sink_id, "The sink", MediaSink::IconType::CAST);
+  MediaSource media_source("mediaSource");
   std::string description("This is a route");
   bool is_local = true;
   std::vector<MediaRoute> routes;
-  routes.push_back(MediaRoute(route_id, MediaSource("mediaSource"), sink_id,
+  routes.push_back(MediaRoute(route_id, media_source, sink_id,
                               description, is_local, kControllerPathForTesting,
                               true));
+  std::vector<MediaRoute::Id> joinable_route_ids;
+  joinable_route_ids.push_back(route_id);
 
-  EXPECT_CALL(*mock_media_router_ui_, GetRouteProviderExtensionId()).WillOnce(
-      ReturnRef(provider_extension_id()));
-  handler_->UpdateRoutes(routes);
+  EXPECT_CALL(*mock_media_router_ui_, GetRouteProviderExtensionId()).Times(1)
+    .WillRepeatedly(ReturnRef(provider_extension_id()));
+  handler_->UpdateRoutes(routes, joinable_route_ids);
   EXPECT_EQ(1u, web_ui_->call_data().size());
   const content::TestWebUI::CallData& call_data = *web_ui_->call_data()[0];
   EXPECT_EQ("media_router.ui.setRouteList", call_data.function_name());
@@ -137,6 +140,13 @@ TEST_F(MediaRouterWebUIMessageHandlerTest, UpdateRoutes) {
   EXPECT_TRUE(arg1->GetAsList(&routes_list_value));
   const base::DictionaryValue* route_value = nullptr;
   EXPECT_TRUE(routes_list_value->GetDictionary(0, &route_value));
+  const base::Value* arg2 = call_data.arg2();
+  const base::ListValue* joinable_routes_list_value = nullptr;
+  EXPECT_TRUE(arg2->GetAsList(&joinable_routes_list_value));
+  std::string joinable_route_value;
+  EXPECT_TRUE(joinable_routes_list_value->GetString(0,
+                                                    &joinable_route_value));
+  EXPECT_EQ(joinable_route_value, route_id);
 
   std::string value;
   EXPECT_TRUE(route_value->GetString("id", &value));
@@ -145,6 +155,14 @@ TEST_F(MediaRouterWebUIMessageHandlerTest, UpdateRoutes) {
   EXPECT_EQ(sink_id, value);
   EXPECT_TRUE(route_value->GetString("description", &value));
   EXPECT_EQ(description, value);
+  /*
+   *EXPECT_TRUE(joinable_route_value->GetString("id", &value));
+   *EXPECT_EQ(joinable_route_id, value);
+   *EXPECT_TRUE(joinable_route_value->GetString("sinkId", &value));
+   *EXPECT_EQ(joinable_sink_id, value);
+   *EXPECT_TRUE(joinable_route_value->GetString("description", &value));
+   *EXPECT_EQ(description, value);
+   */
 
   bool actual_is_local = false;
   EXPECT_TRUE(route_value->GetBoolean("isLocal", &actual_is_local));