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 e123cfaf0db922a7a6492d43de476d2977d6ac2f..eb8998ed5c84f9708cd03cb470848a29d253313f 100644 |
--- a/remoting/protocol/connection_to_client_unittest.cc |
+++ b/remoting/protocol/connection_to_client_unittest.cc |
@@ -11,6 +11,7 @@ |
#include "remoting/base/constants.h" |
#include "remoting/protocol/fake_session.h" |
#include "remoting/protocol/protocol_mock_objects.h" |
+#include "remoting/protocol/video_sender.h" |
#include "testing/gmock/include/gmock/gmock.h" |
using ::testing::_; |
@@ -63,7 +64,7 @@ class ConnectionToClientTest : public testing::Test { |
TEST_F(ConnectionToClientTest, SendUpdateStream) { |
scoped_ptr<VideoPacket> packet(new VideoPacket()); |
- viewer_->video_stub()->ProcessVideoPacket(packet.Pass(), base::Closure()); |
+ viewer_->video_sender()->ProcessVideoPacket(packet.Pass(), base::Closure()); |
base::RunLoop().RunUntilIdle(); |
@@ -82,7 +83,7 @@ TEST_F(ConnectionToClientTest, SendUpdateStream) { |
TEST_F(ConnectionToClientTest, NoWriteAfterDisconnect) { |
scoped_ptr<VideoPacket> packet(new VideoPacket()); |
- viewer_->video_stub()->ProcessVideoPacket(packet.Pass(), base::Closure()); |
+ viewer_->video_sender()->ProcessVideoPacket(packet.Pass(), base::Closure()); |
// And then close the connection to ConnectionToClient. |
viewer_->Disconnect(); |