| Index: remoting/protocol/fake_session.h
|
| diff --git a/remoting/protocol/fake_session.h b/remoting/protocol/fake_session.h
|
| index 90fe0c8d5d7b4fef6b7d37fc9902e6e0568e4e33..8daa59d5b0b639a82699f4f6f6a6f107a0e636c3 100644
|
| --- a/remoting/protocol/fake_session.h
|
| +++ b/remoting/protocol/fake_session.h
|
| @@ -10,6 +10,7 @@
|
| #include <vector>
|
|
|
| #include "base/memory/scoped_ptr.h"
|
| +#include "base/memory/weak_ptr.h"
|
| #include "net/base/completion_callback.h"
|
| #include "net/socket/socket.h"
|
| #include "net/socket/stream_socket.h"
|
| @@ -27,6 +28,11 @@ extern const char kTestJid[];
|
| // Read() reads data from another buffer that can be set with AppendInputData().
|
| // Pending reads are supported, so if there is a pending read AppendInputData()
|
| // calls the read callback.
|
| +//
|
| +// Two fake sockets can be connected to each other using the
|
| +// PairWith() method, e.g.: a->PairWith(b). After this all data
|
| +// written to |a| can be read from |b| and vica versa. Two connected
|
| +// sockets |a| and |b| must be created and used on the same thread.
|
| class FakeSocket : public net::StreamSocket {
|
| public:
|
| FakeSocket();
|
| @@ -34,7 +40,8 @@ class FakeSocket : public net::StreamSocket {
|
|
|
| const std::string& written_data() const { return written_data_; }
|
|
|
| - void AppendInputData(const char* data, int data_size);
|
| + void AppendInputData(const std::vector<char>& data);
|
| + void PairWith(FakeSocket* peer_socket);
|
| int input_pos() const { return input_pos_; }
|
| bool read_pending() const { return read_pending_; }
|
|
|
| @@ -47,7 +54,7 @@ class FakeSocket : public net::StreamSocket {
|
| virtual bool SetReceiveBufferSize(int32 size) OVERRIDE;
|
| virtual bool SetSendBufferSize(int32 size) OVERRIDE;
|
|
|
| - // net::StreamSocket implementation.
|
| + // net::StreamSocket interface.
|
| virtual int Connect(const net::CompletionCallback& callback) OVERRIDE;
|
| virtual void Disconnect() OVERRIDE;
|
| virtual bool IsConnected() const OVERRIDE;
|
| @@ -67,6 +74,7 @@ class FakeSocket : public net::StreamSocket {
|
| scoped_refptr<net::IOBuffer> read_buffer_;
|
| int read_buffer_size_;
|
| net::CompletionCallback read_callback_;
|
| + base::WeakPtr<FakeSocket> peer_socket_;
|
|
|
| std::string written_data_;
|
| std::string input_data_;
|
| @@ -75,6 +83,7 @@ class FakeSocket : public net::StreamSocket {
|
| net::BoundNetLog net_log_;
|
|
|
| MessageLoop* message_loop_;
|
| + base::WeakPtrFactory<FakeSocket> weak_factory_;
|
|
|
| DISALLOW_COPY_AND_ASSIGN(FakeSocket);
|
| };
|
|
|