[PresentationSevice] Use PresentationConnection to send messages from

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 60 matching lines @@
 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));
 };
 
+class MockPresentationConnectionProxy
+    : public NON_EXPORTED_BASE(blink::mojom::PresentationConnection) {
+ public:
+  // PresentationConnectionMessage is move-only.
+  // TODO(crbug.com/729950): Use MOCK_METHOD directly once GMock gets the
+  // move-only type support.
+  MockPresentationConnectionProxy();
+  ~MockPresentationConnectionProxy() override;
+  void OnMessage(content::PresentationConnectionMessage message,
+                 OnMessageCallback cb) {
+    OnMessageInternal(message, cb);
+  }
+  MOCK_METHOD2(OnMessageInternal,
+               void(const content::PresentationConnectionMessage&,
+                    OnMessageCallback&));
+  MOCK_METHOD1(DidChangeState,
+               void(content::PresentationConnectionState state));
+  MOCK_METHOD0(OnClose, void());
+};
+
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_MEDIA_ROUTER_TEST_HELPER_H_