Index: remoting/protocol/connection_to_client_unittest.cc |
diff --git a/remoting/protocol/connection_to_client_unittest.cc b/remoting/protocol/connection_to_client_unittest.cc |
index 360951131e9e5ca683f61bcea988b44629e13d8e..9c2b6af40efd61efad49f1a06ea4b9757f99f2f2 100644 |
--- a/remoting/protocol/connection_to_client_unittest.cc |
+++ b/remoting/protocol/connection_to_client_unittest.cc |
@@ -4,6 +4,7 @@ |
#include "base/message_loop.h" |
#include "remoting/base/base_mock_objects.h" |
+#include "remoting/base/constants.h" |
#include "remoting/protocol/fake_session.h" |
#include "remoting/protocol/connection_to_client.h" |
#include "remoting/protocol/protocol_mock_objects.h" |
@@ -34,6 +35,8 @@ class ConnectionToClientTest : public testing::Test { |
EXPECT_CALL(handler_, OnConnectionOpened(viewer_.get())); |
session_->state_change_callback()->Run( |
protocol::Session::CONNECTED); |
+ session_->state_change_callback()->Run( |
+ protocol::Session::CONNECTED_CHANNELS); |
message_loop_.RunAllPending(); |
} |
@@ -60,7 +63,9 @@ TEST_F(ConnectionToClientTest, SendUpdateStream) { |
// Verify that something has been written. |
// TODO(sergeyu): Verify that the correct data has been written. |
- EXPECT_GT(session_->video_channel()->written_data().size(), 0u); |
+ ASSERT_TRUE(session_->GetStreamChannel(kVideoChannelName)); |
+ EXPECT_GT(session_->GetStreamChannel(kVideoChannelName)-> |
+ written_data().size(), 0u); |
// And then close the connection to ConnectionToClient. |
viewer_->Disconnect(); |