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/bind.h" | 5 #include "base/bind.h" |
6 #include "base/message_loop.h" | 6 #include "base/message_loop.h" |
7 #include "base/message_loop_proxy.h" | 7 #include "base/message_loop_proxy.h" |
8 #include "remoting/base/base_mock_objects.h" | 8 #include "remoting/base/base_mock_objects.h" |
9 #include "remoting/base/constants.h" | 9 #include "remoting/base/constants.h" |
10 #include "remoting/protocol/fake_session.h" | 10 #include "remoting/protocol/fake_session.h" |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
57 protocol::FakeSession* session_; | 57 protocol::FakeSession* session_; |
58 | 58 |
59 private: | 59 private: |
60 DISALLOW_COPY_AND_ASSIGN(ConnectionToClientTest); | 60 DISALLOW_COPY_AND_ASSIGN(ConnectionToClientTest); |
61 }; | 61 }; |
62 | 62 |
63 TEST_F(ConnectionToClientTest, SendUpdateStream) { | 63 TEST_F(ConnectionToClientTest, SendUpdateStream) { |
64 // Then send the actual data. | 64 // Then send the actual data. |
65 VideoPacket* packet = new VideoPacket(); | 65 VideoPacket* packet = new VideoPacket(); |
66 viewer_->video_stub()->ProcessVideoPacket( | 66 viewer_->video_stub()->ProcessVideoPacket( |
67 packet, base::Bind(&DeletePointer<VideoPacket>, packet)); | 67 packet, base::Bind(&base::DeletePointer<VideoPacket>, packet)); |
68 | 68 |
69 message_loop_.RunAllPending(); | 69 message_loop_.RunAllPending(); |
70 | 70 |
71 // Verify that something has been written. | 71 // Verify that something has been written. |
72 // TODO(sergeyu): Verify that the correct data has been written. | 72 // TODO(sergeyu): Verify that the correct data has been written. |
73 ASSERT_TRUE(session_->GetStreamChannel(kVideoChannelName)); | 73 ASSERT_TRUE(session_->GetStreamChannel(kVideoChannelName)); |
74 EXPECT_GT(session_->GetStreamChannel(kVideoChannelName)-> | 74 EXPECT_GT(session_->GetStreamChannel(kVideoChannelName)-> |
75 written_data().size(), 0u); | 75 written_data().size(), 0u); |
76 | 76 |
77 // And then close the connection to ConnectionToClient. | 77 // And then close the connection to ConnectionToClient. |
78 viewer_->Disconnect(); | 78 viewer_->Disconnect(); |
79 | 79 |
80 message_loop_.RunAllPending(); | 80 message_loop_.RunAllPending(); |
81 } | 81 } |
82 | 82 |
83 TEST_F(ConnectionToClientTest, NoWriteAfterDisconnect) { | 83 TEST_F(ConnectionToClientTest, NoWriteAfterDisconnect) { |
84 // Then send the actual data. | 84 // Then send the actual data. |
85 VideoPacket* packet = new VideoPacket(); | 85 VideoPacket* packet = new VideoPacket(); |
86 viewer_->video_stub()->ProcessVideoPacket( | 86 viewer_->video_stub()->ProcessVideoPacket( |
87 packet, base::Bind(&DeletePointer<VideoPacket>, packet)); | 87 packet, base::Bind(&base::DeletePointer<VideoPacket>, packet)); |
88 | 88 |
89 // And then close the connection to ConnectionToClient. | 89 // And then close the connection to ConnectionToClient. |
90 viewer_->Disconnect(); | 90 viewer_->Disconnect(); |
91 | 91 |
92 // The test will crash if data writer tries to write data to the | 92 // The test will crash if data writer tries to write data to the |
93 // channel socket. | 93 // channel socket. |
94 // TODO(sergeyu): Use MockSession to verify that no data is written? | 94 // TODO(sergeyu): Use MockSession to verify that no data is written? |
95 message_loop_.RunAllPending(); | 95 message_loop_.RunAllPending(); |
96 } | 96 } |
97 | 97 |
98 TEST_F(ConnectionToClientTest, StateChange) { | 98 TEST_F(ConnectionToClientTest, StateChange) { |
99 EXPECT_CALL(handler_, OnConnectionClosed(viewer_.get())); | 99 EXPECT_CALL(handler_, OnConnectionClosed(viewer_.get())); |
100 session_->state_change_callback().Run(protocol::Session::CLOSED); | 100 session_->state_change_callback().Run(protocol::Session::CLOSED); |
101 message_loop_.RunAllPending(); | 101 message_loop_.RunAllPending(); |
102 | 102 |
103 EXPECT_CALL(handler_, OnConnectionFailed( | 103 EXPECT_CALL(handler_, OnConnectionFailed( |
104 viewer_.get(), Session::SESSION_REJECTED)); | 104 viewer_.get(), Session::SESSION_REJECTED)); |
105 session_->set_error(Session::SESSION_REJECTED); | 105 session_->set_error(Session::SESSION_REJECTED); |
106 session_->state_change_callback().Run(protocol::Session::FAILED); | 106 session_->state_change_callback().Run(protocol::Session::FAILED); |
107 message_loop_.RunAllPending(); | 107 message_loop_.RunAllPending(); |
108 } | 108 } |
109 | 109 |
110 } // namespace protocol | 110 } // namespace protocol |
111 } // namespace remoting | 111 } // namespace remoting |
OLD | NEW |