[Presentation API] Use connection proxy to change connection state to 'closed'

content/renderer/presentation/presentation_connection_proxy_unittest.cc

 // Copyright 2017 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.
 
 #include <memory>
 #include <utility>
 
 #include "base/run_loop.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "content/renderer/presentation/presentation_connection_proxy.h"
@@ skipping 94 matching lines @@
           }));
 
   controller_connection_proxy_->SendConnectionMessage(
       std::move(session_message),
       base::Bind(
           &PresentationConnectionProxyTest::ExpectSendConnectionMessageCallback,
           base::Unretained(this)));
   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