[Media Router] Custom Controls 2 - add MediaRouter::GetRouteController()

chrome/browser/media/router/mock_media_router.h

Index: chrome/browser/media/router/mock_media_router.h
diff --git a/chrome/browser/media/router/mock_media_router.h b/chrome/browser/media/router/mock_media_router.h
index 84d1764183bf1d8d95f6766f140a280ce7f67095..2a91575aac972060ea9de3eab9dfc8bcc74a259f 100644
--- a/chrome/browser/media/router/mock_media_router.h
+++ b/chrome/browser/media/router/mock_media_router.h
@@ -15,6 +15,7 @@
 #include "chrome/browser/media/router/media_router_base.h"
 #include "chrome/browser/media/router/media_sink.h"
 #include "chrome/browser/media/router/media_source.h"
+#include "chrome/browser/media/router/mojo/media_route_controller.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "url/origin.h"
 
@@ -91,6 +92,9 @@ class MockMediaRouter : public MediaRouterBase {
   MOCK_CONST_METHOD0(GetCurrentRoutes, std::vector<MediaRoute>());
 
   MOCK_METHOD0(OnIncognitoProfileShutdown, void());
+  MOCK_METHOD1(
+      GetRouteController,
+      scoped_refptr<MediaRouteController>(const MediaRoute::Id& route_id));
   MOCK_METHOD1(OnAddPresentationConnectionStateChangedCallbackInvoked,
                void(const content::PresentationConnectionStateChangedCallback&
                         callback));
@@ -107,6 +111,9 @@ class MockMediaRouter : public MediaRouterBase {
                void(RouteMessageObserver* observer));
   MOCK_METHOD1(UnregisterRouteMessageObserver,
                void(RouteMessageObserver* observer));
+  MOCK_METHOD2(DetachRouteController,
+               void(const MediaRoute::Id& route_id,
+                    MediaRouteController* controller));
 
  private:
   base::CallbackList<void(