[Media Router] Connection state change listening redesign part 2.

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 <string>
 #include <vector>
 
 #include "chrome/browser/media/router/issues_observer.h"
 #include "chrome/browser/media/router/media_router.mojom.h"
 #include "chrome/browser/media/router/media_router_mojo_impl.h"
 #include "chrome/browser/media/router/media_routes_observer.h"
 #include "chrome/browser/media/router/media_sinks_observer.h"
-#include "chrome/browser/media/router/presentation_connection_state_observer.h"
 #include "extensions/browser/event_page_tracker.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace media_router {
 
 // Matcher for objects that uses Equals() member function for equality check.
 MATCHER_P(Equals, other, "") {
   return arg.Equals(other);
 }
 
@@ skipping 111 matching lines @@
 };
 
 class MockMediaRoutesObserver : public MediaRoutesObserver {
  public:
   explicit MockMediaRoutesObserver(MediaRouter* router);
   ~MockMediaRoutesObserver() override;
 
   MOCK_METHOD1(OnRoutesUpdated, void(const std::vector<MediaRoute>& sinks));
 };
 
-class MockPresentationConnectionStateObserver
-    : public PresentationConnectionStateObserver {
- public:
-  MockPresentationConnectionStateObserver(MediaRouter* router,
-                                          const MediaRoute::Id& route_id);
-  ~MockPresentationConnectionStateObserver() override;
-
-  MOCK_METHOD1(OnStateChanged,
-               void(content::PresentationConnectionState state));
-};
-
 class MockEventPageTracker : public extensions::EventPageTracker {
  public:
   MockEventPageTracker();
   ~MockEventPageTracker();
 
   MOCK_METHOD1(IsEventPageSuspended, bool(const std::string& extension_id));
   MOCK_METHOD2(WakeEventPage,
                bool(const std::string& extension_id,
                     const base::Callback<void(bool)>& callback));
 };
 
+class MockPresentationConnectionStateChangedCallback {
+ public:
+  MockPresentationConnectionStateChangedCallback();
+  ~MockPresentationConnectionStateChangedCallback();
+  MOCK_METHOD1(Run, void(content::PresentationConnectionState));
+};
+
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_MEDIA_ROUTER_TEST_HELPER_H_