| 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_;
|
|
|