Implement video frame acknowledgements in the chromoting protocol.

remoting/protocol/protocol_mock_objects.h

 // 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.
 
 #ifndef REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_
 #define REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_
 
 #include <map>
 #include <string>
 
 #include "base/location.h"
 #include "base/single_thread_task_runner.h"
 #include "base/values.h"
 #include "net/base/ip_endpoint.h"
 #include "remoting/proto/internal.pb.h"
 #include "remoting/proto/video.pb.h"
 #include "remoting/protocol/authenticator.h"
 #include "remoting/protocol/client_stub.h"
 #include "remoting/protocol/clipboard_stub.h"
 #include "remoting/protocol/connection_to_client.h"
 #include "remoting/protocol/host_stub.h"
 #include "remoting/protocol/input_stub.h"
 #include "remoting/protocol/pairing_registry.h"
 #include "remoting/protocol/session.h"
 #include "remoting/protocol/session_manager.h"
 #include "remoting/protocol/transport.h"
-#include "remoting/protocol/video_stub.h"
+#include "remoting/protocol/video_sender.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace remoting {
 namespace protocol {
 
 class MockConnectionToClient : public ConnectionToClient {
  public:
   MockConnectionToClient(Session* session,
                          HostStub* host_stub);
   virtual ~MockConnectionToClient();
 
   MOCK_METHOD1(Init, void(Session* session));
-  MOCK_METHOD0(video_stub, VideoStub*());
+  MOCK_METHOD0(video_sender, VideoSender*());
   MOCK_METHOD0(client_stub, ClientStub*());
   MOCK_METHOD0(session, Session*());
   MOCK_METHOD0(Disconnect, void());
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockConnectionToClient);
 };
 
 class MockConnectionToClientEventHandler :
       public ConnectionToClient::EventHandler {
@@ skipping 104 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(MockCursorShapeStub);
 };
 
 class MockVideoStub : public VideoStub {
  public:
   MockVideoStub();
   virtual ~MockVideoStub();
 
   MOCK_METHOD2(ProcessVideoPacketPtr, void(const VideoPacket* video_packet,
                                            const base::Closure& done));
-  virtual void ProcessVideoPacket(scoped_ptr<VideoPacket> video_packet,
-                                  const base::Closure& done) {
+  void ProcessVideoPacket(scoped_ptr<VideoPacket> video_packet,
+                                  const base::Closure& done) override {
     ProcessVideoPacketPtr(video_packet.get(), done);
   }
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockVideoStub);
 };
 
+class MockVideoSender : public VideoSender {
+ public:
+  MockVideoSender();
+  virtual ~MockVideoSender();
+
+  MOCK_METHOD2(ProcessVideoPacketPtr, void(const VideoPacket* video_packet,
+                                           const base::Closure& done));
+  void ProcessVideoPacket(scoped_ptr<VideoPacket> video_packet,
+                                  const base::Closure& done) override {
+    ProcessVideoPacketPtr(video_packet.get(), done);
+  }
+  bool SupportsAcks() override { return true; }
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockVideoSender);
+};
+
 class MockSession : public Session {
  public:
   MockSession();
   virtual ~MockSession();
 
   MOCK_METHOD1(SetEventHandler, void(Session::EventHandler* event_handler));
   MOCK_METHOD0(error, ErrorCode());
   MOCK_METHOD0(GetTransportChannelFactory, StreamChannelFactory*());
   MOCK_METHOD0(GetMultiplexedChannelFactory, StreamChannelFactory*());
   MOCK_METHOD0(jid, const std::string&());
@@ skipping 70 matching lines @@
   // Runs tasks synchronously instead of posting them to |task_runner|.
   void PostTask(const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
                 const tracked_objects::Location& from_here,
                 const base::Closure& task) override;
 };
 
 }  // namespace protocol
 }  // namespace remoting
 
 #endif  // REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_