Add support for dynamic channels in WebrtcTransport.

remoting/protocol/connection_tester.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_CONNECTION_TESTER_H_
 #define REMOTING_PROTOCOL_CONNECTION_TESTER_H_
 
 #include <list>
 #include <memory>
 #include <vector>
 
 #include "base/memory/ref_counted.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
+#include "remoting/protocol/message_pipe.h"
 
 namespace net {
 class DrainableIOBuffer;
 class GrowableIOBuffer;
 class IOBuffer;
 }  // namespace net
 
 namespace remoting {
 
 class CompoundBuffer;
 class VideoPacket;
 
 namespace protocol {
 
-class MessagePipe;
 class P2PDatagramSocket;
 class P2PStreamSocket;
 
 // This class is used by unit tests to verify that a connection
 // between two sockets works properly, i.e. data is delivered from one
 // end to the other.
 class StreamConnectionTester {
  public:
   StreamConnectionTester(P2PStreamSocket* client_socket,
                          P2PStreamSocket* host_socket,
@@ skipping 62 matching lines @@
   std::vector<scoped_refptr<net::IOBuffer> > sent_packets_;
   scoped_refptr<net::IOBuffer> read_buffer_;
 
   int write_errors_;
   int read_errors_;
   int packets_sent_;
   int packets_received_;
   int bad_packets_received_;
 };
 
-class MessagePipeConnectionTester {
+class MessagePipeConnectionTester : public MessagePipe::EventHandler {
  public:
   MessagePipeConnectionTester(MessagePipe* client_pipe,
                               MessagePipe* host_pipe,
                               int message_size,
                               int message_count);
-  ~MessagePipeConnectionTester();
+  ~MessagePipeConnectionTester() override;
 
   void RunAndCheckResults();
 
  protected:
-  void OnMessageReceived(std::unique_ptr<CompoundBuffer> message);
+  // MessagePipe::EventHandler interface.
+  void OnMessageReceived(std::unique_ptr<CompoundBuffer> message) override;
+  void OnMessagePipeClosed() override;
 
  private:
   base::RunLoop run_loop_;
   MessagePipe* host_pipe_;
   MessagePipe* client_pipe_;
   int message_size_;
   int message_count_;
 
   std::vector<std::unique_ptr<VideoPacket>> sent_messages_;
   std::vector<std::unique_ptr<VideoPacket>> received_messages_;
 };
 
 }  // namespace protocol
 }  // namespace remoting
 
 #endif  // REMOTING_PROTOCOL_CONNECTION_TESTER_H_