Index: remoting/protocol/connection_tester.h |
diff --git a/remoting/protocol/connection_tester.h b/remoting/protocol/connection_tester.h |
index eda98b4e8ee0743ae8b0033fd8a99c5c14755d2d..fc63968367a519d7f35f86fbdd65cb988bf24a05 100644 |
--- a/remoting/protocol/connection_tester.h |
+++ b/remoting/protocol/connection_tester.h |
@@ -12,6 +12,7 @@ |
#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; |
@@ -26,7 +27,6 @@ class VideoPacket; |
namespace protocol { |
-class MessagePipe; |
class P2PDatagramSocket; |
class P2PStreamSocket; |
@@ -109,18 +109,20 @@ class DatagramConnectionTester { |
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_; |