Pass origin to StartObservingMediaSinks.

chrome/browser/media/router/test_helper.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_TEST_HELPER_H_
 #define CHROME_BROWSER_MEDIA_ROUTER_TEST_HELPER_H_
 
 #include <stddef.h>
 #include <stdint.h>
 
@@ skipping 102 matching lines @@
   MOCK_METHOD7(ConnectRouteByRouteId,
                void(const mojo::String& source_urn,
                     const mojo::String& route_id,
                     const mojo::String& presentation_id,
                     const mojo::String& origin,
                     int tab_id,
                     int64_t timeout_secs,
                     const JoinRouteCallback& callback));
   MOCK_METHOD1(DetachRoute, void(const mojo::String& route_id));
   MOCK_METHOD1(TerminateRoute, void(const mojo::String& route_id));
-  MOCK_METHOD1(StartObservingMediaSinks, void(const mojo::String& source));
-  MOCK_METHOD1(StopObservingMediaSinks, void(const mojo::String& source));
+  MOCK_METHOD2(StartObservingMediaSinks,
+               void(const mojo::String& source, const mojo::String& origin));
+  MOCK_METHOD2(StopObservingMediaSinks,
+               void(const mojo::String& source, const mojo::String& origin));
   MOCK_METHOD3(SendRouteMessage,
                void(const mojo::String& media_route_id,
                     const mojo::String& message,
                     const SendRouteMessageCallback& callback));
   void SendRouteBinaryMessage(
       const mojo::String& media_route_id,
       mojo::Array<uint8_t> data,
       const SendRouteMessageCallback& callback) override {
     SendRouteBinaryMessageInternal(media_route_id, data.storage(), callback);
   }
@@ skipping 10 matching lines @@
                void(const mojo::String& route_id));
   MOCK_METHOD1(StartObservingMediaRoutes, void(const mojo::String& source));
   MOCK_METHOD1(StopObservingMediaRoutes, void(const mojo::String& source));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockMediaRouteProvider);
 };
 
 class MockMediaSinksObserver : public MediaSinksObserver {
  public:
-  MockMediaSinksObserver(MediaRouter* router, const MediaSource& source);
+  MockMediaSinksObserver(MediaRouter* router,
+                         const MediaSource& source);
   ~MockMediaSinksObserver() override;
 
-  MOCK_METHOD1(OnSinksReceived, void(const std::vector<MediaSink>& sinks));
+  MOCK_METHOD2(OnSinksReceived, void(const MediaSource& source,
+                                     const std::vector<MediaSink>& sinks));
 };
 
 class MockMediaRoutesObserver : public MediaRoutesObserver {
  public:
   explicit MockMediaRoutesObserver(MediaRouter* router,
       const MediaSource::Id source_id = std::string());
   ~MockMediaRoutesObserver() override;
 
   MOCK_METHOD2(OnRoutesUpdated, void(const std::vector<MediaRoute>& routes,
       const std::vector<MediaRoute::Id>& joinable_route_ids));
@@ skipping 14 matching lines @@
  public:
   MockPresentationConnectionStateChangedCallback();
   ~MockPresentationConnectionStateChangedCallback();
   MOCK_METHOD1(Run,
                void(const content::PresentationConnectionStateChangeInfo&));
 };
 
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_MEDIA_ROUTER_TEST_HELPER_H_