[Media Router] Add features to control browser side discovery

chrome/browser/media/router/mojo/media_router_mojo_test.cc

 // Copyright 2015 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 "chrome/browser/media/router/mojo/media_router_mojo_test.h"
 
 #include <utility>
 
 #include "base/run_loop.h"
 #include "extensions/common/test_util.h"
 
 namespace media_router {
 namespace {
-
 const char kInstanceId[] = "instance123";
-
-void ExpectEqualStrings(const std::string& expected,
-                        const std::string& actual) {
-  EXPECT_EQ(expected, actual);
-}
-
 }  // namespace
 
 MockMediaRouteProvider::MockMediaRouteProvider() {}
 
 MockMediaRouteProvider::~MockMediaRouteProvider() {}
 
 MockEventPageTracker::MockEventPageTracker() {}
 
 MockEventPageTracker::~MockEventPageTracker() {}
 
 MockMediaController::MockMediaController() : binding_(this) {}
 
 MockMediaController::~MockMediaController() {}
 
+RegisterMediaRouteProviderHandler::RegisterMediaRouteProviderHandler() {}
+
+RegisterMediaRouteProviderHandler::~RegisterMediaRouteProviderHandler() {}
+
 void MockMediaController::Bind(mojom::MediaControllerRequest request) {
   binding_.Bind(std::move(request));
 }
 
 mojom::MediaControllerPtr MockMediaController::BindInterfacePtr() {
   return binding_.CreateInterfacePtrAndBind();
 }
 
 void MockMediaController::CloseBinding() {
   binding_.Close();
@@ skipping 27 matching lines @@
 void MediaRouterMojoTest::ConnectProviderManagerService() {
   // Bind the |media_route_provider| interface to |media_route_provider_|.
   auto request = mojo::MakeRequest(&media_router_proxy_);
   mock_media_router_->BindToMojoRequest(std::move(request), *extension_);
 
   // Bind the Mojo MediaRouter interface used by |mock_media_router_| to
   // |mock_media_route_provider_service_|.
   mojom::MediaRouteProviderPtr mojo_media_router;
   binding_.reset(new mojo::Binding<mojom::MediaRouteProvider>(
       &mock_media_route_provider_, mojo::MakeRequest(&mojo_media_router)));
+  EXPECT_CALL(provide_handler_, InvokeInternal(kInstanceId, testing::_));
   media_router_proxy_->RegisterMediaRouteProvider(
       std::move(mojo_media_router),
-      base::Bind(&ExpectEqualStrings, kInstanceId));
+      base::Bind(&RegisterMediaRouteProviderHandler::Invoke,
+                 base::Unretained(&provide_handler_)));
 }
 
 void MediaRouterMojoTest::SetUp() {
   ON_CALL(mock_event_page_tracker_, IsEventPageSuspended(extension_id()))
       .WillByDefault(testing::Return(false));
   ConnectProviderManagerService();
   base::RunLoop().RunUntilIdle();
 }
 
 void MediaRouterMojoTest::TearDown() {
   mock_media_router_->Shutdown();
 }
 
 void MediaRouterMojoTest::ProcessEventLoop() {
   base::RunLoop().RunUntilIdle();
 }
 
 }  // namespace media_router