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 ab2bf4d6dd1592e59ba3f0e8c012da6db26bf8b9..99c6954b41f557d5de66420a9f275cb6d1984e3e 100644 |
--- a/content/renderer/presentation/presentation_dispatcher_unittest.cc |
+++ b/content/renderer/presentation/presentation_dispatcher_unittest.cc |
@@ -124,6 +124,7 @@ class TestPresentationConnectionProxy : public PresentationConnectionProxy { |
MOCK_CONST_METHOD2(SendConnectionMessageInternal, |
void(blink::mojom::ConnectionMessage*, |
const OnMessageCallback&)); |
+ MOCK_CONST_METHOD0(close, void()); |
}; |
class TestPresentationReceiver : public blink::WebPresentationReceiver { |
@@ -506,9 +507,12 @@ TEST_F(PresentationDispatcherTest, TestOnReceiverConnectionAvailable) { |
TEST_F(PresentationDispatcherTest, TestCloseSession) { |
base::RunLoop run_loop; |
+ TestPresentationConnection connection; |
+ TestPresentationConnectionProxy test_proxy(&connection); |
+ EXPECT_CALL(test_proxy, close()); |
EXPECT_CALL(presentation_service_, |
CloseConnection(gurl1_, presentation_id_.utf8())); |
- dispatcher_.closeSession(url1_, presentation_id_); |
+ dispatcher_.closeSession(url1_, presentation_id_, &test_proxy); |
run_loop.RunUntilIdle(); |
} |