Index: content/renderer/presentation/presentation_dispatcher_unittest.cc |
diff --git a/content/renderer/presentation/presentation_dispatcher_unittest.cc b/content/renderer/presentation/presentation_dispatcher_unittest.cc |
index 9397f852feda403fd7705c88b7a7412e3ec26875..3ec39bad8cd51ee6b66bc56ec12a3a9884f99231 100644 |
--- a/content/renderer/presentation/presentation_dispatcher_unittest.cc |
+++ b/content/renderer/presentation/presentation_dispatcher_unittest.cc |
@@ -25,6 +25,7 @@ using blink::WebPresentationSessionInfo; |
using blink::WebString; |
using blink::WebURL; |
using blink::WebVector; |
+using blink::mojom::PresentationConnection; |
using blink::mojom::PresentationError; |
using blink::mojom::PresentationErrorPtr; |
using blink::mojom::PresentationErrorType; |
@@ -102,6 +103,17 @@ class MockPresentationService : public PresentationService { |
} |
MOCK_METHOD1(ListenForConnectionMessagesInternal, |
void(PresentationSessionInfo* session_info)); |
+ |
+ void SetPresentationConnection( |
+ blink::mojom::PresentationSessionInfoPtr session, |
+ blink::mojom::PresentationConnectionPtr controller_conn_ptr, |
+ blink::mojom::PresentationConnectionRequest receiver_conn_request) |
+ override { |
+ SetPresentationConnection(session.get(), controller_conn_ptr.get()); |
+ } |
+ MOCK_METHOD2(SetPresentationConnection, |
+ void(PresentationSessionInfo* session_info, |
+ PresentationConnection* connection)); |
}; |
class TestWebPresentationConnectionCallback |