[Media Router] Add features to control browser side discovery

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

 // 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.
 
 #ifndef CHROME_BROWSER_MEDIA_ROUTER_MOJO_MEDIA_ROUTER_MOJO_TEST_H_
 #define CHROME_BROWSER_MEDIA_ROUTER_MOJO_MEDIA_ROUTER_MOJO_TEST_H_
 
 #include <string>
 
 #include "base/macros.h"
@@ skipping 156 matching lines @@
 class MockMediaRouteControllerObserver : public MediaRouteController::Observer {
  public:
   MockMediaRouteControllerObserver(
       scoped_refptr<MediaRouteController> controller);
   ~MockMediaRouteControllerObserver() override;
 
   MOCK_METHOD1(OnMediaStatusUpdated, void(const MediaStatus& status));
   MOCK_METHOD0(OnControllerInvalidated, void());
 };
 
+// Mockable class for awaiting RegisterMediaRouteProvider callbacks.
+class RegisterMediaRouteProviderHandler {
+ public:
+  RegisterMediaRouteProviderHandler();
+  ~RegisterMediaRouteProviderHandler();
+
+  // A wrapper function to deal with move only parameter |config|.
+  void Invoke(const std::string& instance_id,
+              mojom::MediaRouteProviderConfigPtr config) {
+    InvokeRaw(instance_id, config.get());
+  }
+
+  MOCK_METHOD2(InvokeRaw,

[mark a. foltz, 2017/05/11 23:17:59]

In general we have been using InvokeInternal for these adapters to mock methods
with move-only types.

[zhaobin, 2017/05/12 00:16:51]

Done.

+               void(const std::string& instance_id,
+                    mojom::MediaRouteProviderConfig* config));
+};
+
 // Tests the API call flow between the MediaRouterMojoImpl and the Media Router
 // Mojo service in both directions.
 class MediaRouterMojoTest : public ::testing::Test {
  public:
   MediaRouterMojoTest();
   ~MediaRouterMojoTest() override;
 
  protected:
   void SetUp() override;
   void TearDown() override;
 
   void ProcessEventLoop();
 
   void ConnectProviderManagerService();
 
   const std::string& extension_id() const { return extension_->id(); }
 
   MediaRouterMojoImpl* router() const { return mock_media_router_.get(); }
 
   // Mock objects.
   MockMediaRouteProvider mock_media_route_provider_;
   testing::NiceMock<MockEventPageTracker> mock_event_page_tracker_;
 
   // Mojo proxy object for |mock_media_router_|
   media_router::mojom::MediaRouterPtr media_router_proxy_;
 
+  RegisterMediaRouteProviderHandler provide_handler_;
+
  private:
   content::TestBrowserThreadBundle test_thread_bundle_;
   scoped_refptr<extensions::Extension> extension_;
   std::unique_ptr<MediaRouterMojoImpl> mock_media_router_;
   std::unique_ptr<mojo::Binding<mojom::MediaRouteProvider>> binding_;
 
   DISALLOW_COPY_AND_ASSIGN(MediaRouterMojoTest);
 };
 
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_MEDIA_ROUTER_MOJO_MEDIA_ROUTER_MOJO_TEST_H_