diff options
Diffstat (limited to 'remoting/protocol/connection_to_client_unittest.cc')
| -rw-r--r-- | remoting/protocol/connection_to_client_unittest.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/remoting/protocol/connection_to_client_unittest.cc b/remoting/protocol/connection_to_client_unittest.cc index 3609511..8a8e1e8 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" @@ -60,7 +61,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(); |