Implement video frame acknowledgements in the chromoting protocol.

remoting/protocol/connection_to_client_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "remoting/protocol/connection_to_client.h"
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/message_loop/message_loop_proxy.h"
 #include "base/run_loop.h"
 #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::_;
 using ::testing::NotNull;
 using ::testing::StrictMock;
 
 namespace remoting {
 namespace protocol {
 
 class ConnectionToClientTest : public testing::Test {
@@ skipping 32 matching lines @@
 
   // Owned by |viewer_|.
   FakeSession* session_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(ConnectionToClientTest);
 };
 
 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();
 
   // Verify that something has been written.
   // TODO(sergeyu): Verify that the correct data has been written.
   ASSERT_TRUE(
       session_->fake_channel_factory().GetFakeChannel(kVideoChannelName));
   EXPECT_FALSE(session_->fake_channel_factory()
                    .GetFakeChannel(kVideoChannelName)->written_data().empty());
 
   // And then close the connection to ConnectionToClient.
   viewer_->Disconnect();
 
   base::RunLoop().RunUntilIdle();
 }
 
 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();
 
   // The test will crash if data writer tries to write data to the
   // channel socket.
   // TODO(sergeyu): Use MockSession to verify that no data is written?
   base::RunLoop().RunUntilIdle();
 }
 
 TEST_F(ConnectionToClientTest, StateChange) {
   EXPECT_CALL(handler_, OnConnectionClosed(viewer_.get(), OK));
   session_->event_handler()->OnSessionStateChange(Session::CLOSED);
   base::RunLoop().RunUntilIdle();
 
   EXPECT_CALL(handler_, OnConnectionClosed(viewer_.get(), SESSION_REJECTED));
   session_->set_error(SESSION_REJECTED);
   session_->event_handler()->OnSessionStateChange(Session::FAILED);
   base::RunLoop().RunUntilIdle();
 }
 
 }  // namespace protocol
 }  // namespace remoting