Index: content/renderer/presentation/presentation_connection_proxy_unittest.cc |
diff --git a/content/renderer/presentation/presentation_connection_proxy_unittest.cc b/content/renderer/presentation/presentation_connection_proxy_unittest.cc |
index 5e43a2246373923072c645e5d0edc003a21bb00e..159becffc3604a9f34c798b79e0ddbf132caca14 100644 |
--- a/content/renderer/presentation/presentation_connection_proxy_unittest.cc |
+++ b/content/renderer/presentation/presentation_connection_proxy_unittest.cc |
@@ -112,4 +112,24 @@ TEST_F(PresentationConnectionProxyTest, TestSendArrayBuffer) { |
run_loop.RunUntilIdle(); |
} |
+TEST_F(PresentationConnectionProxyTest, TestControllerConnectionCallsClose) { |
+ base::RunLoop run_loop; |
+ EXPECT_CALL(*controller_connection_, |
+ didChangeState(blink::WebPresentationConnectionState::Closed)); |
+ EXPECT_CALL(*receiver_connection_, |
+ didChangeState(blink::WebPresentationConnectionState::Closed)); |
+ controller_connection_proxy_->close(); |
+ run_loop.RunUntilIdle(); |
+} |
+ |
+TEST_F(PresentationConnectionProxyTest, TestReceiverConnectionCallsClose) { |
+ base::RunLoop run_loop; |
+ EXPECT_CALL(*controller_connection_, |
+ didChangeState(blink::WebPresentationConnectionState::Closed)); |
+ EXPECT_CALL(*receiver_connection_, |
+ didChangeState(blink::WebPresentationConnectionState::Closed)); |
+ receiver_connection_proxy_->close(); |
+ run_loop.RunUntilIdle(); |
+} |
+ |
} // namespace content |