| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/message_loop.h" | 5 #include "base/message_loop.h" |
| 6 #include "remoting/base/base_mock_objects.h" | 6 #include "remoting/base/base_mock_objects.h" |
| 7 #include "remoting/protocol/fake_session.h" | 7 #include "remoting/protocol/fake_session.h" |
| 8 #include "remoting/protocol/connection_to_client.h" | 8 #include "remoting/protocol/connection_to_client.h" |
| 9 #include "remoting/protocol/protocol_mock_objects.h" | 9 #include "remoting/protocol/protocol_mock_objects.h" |
| 10 #include "testing/gmock/include/gmock/gmock.h" | 10 #include "testing/gmock/include/gmock/gmock.h" |
| (...skipping 79 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 90 | 90 |
| 91 EXPECT_CALL(handler_, OnConnectionFailed(viewer_.get())); | 91 EXPECT_CALL(handler_, OnConnectionFailed(viewer_.get())); |
| 92 session_->state_change_callback()->Run(protocol::Session::FAILED); | 92 session_->state_change_callback()->Run(protocol::Session::FAILED); |
| 93 message_loop_.RunAllPending(); | 93 message_loop_.RunAllPending(); |
| 94 } | 94 } |
| 95 | 95 |
| 96 // Test that we can close client connection more than once. | 96 // Test that we can close client connection more than once. |
| 97 TEST_F(ConnectionToClientTest, Close) { | 97 TEST_F(ConnectionToClientTest, Close) { |
| 98 viewer_->Disconnect(); | 98 viewer_->Disconnect(); |
| 99 message_loop_.RunAllPending(); | 99 message_loop_.RunAllPending(); |
| 100 EXPECT_TRUE(session_->is_closed()); | |
| 101 | |
| 102 viewer_->Disconnect(); | 100 viewer_->Disconnect(); |
| 103 message_loop_.RunAllPending(); | 101 message_loop_.RunAllPending(); |
| 104 } | 102 } |
| 105 | 103 |
| 106 } // namespace protocol | 104 } // namespace protocol |
| 107 } // namespace remoting | 105 } // namespace remoting |
| OLD | NEW |