[Presentation API] Add layout test for connection.close() and fix test failures

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 96 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));
+  EXPECT_CALL(*controller_connection_, didClose());
+  EXPECT_CALL(*receiver_connection_, didClose());
   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));
+  EXPECT_CALL(*controller_connection_, didClose());
+  EXPECT_CALL(*receiver_connection_, didClose());
   receiver_connection_proxy_->close();
   run_loop.RunUntilIdle();
 }
 
 }  // namespace content