Index: remoting/protocol/fake_session.h |
diff --git a/remoting/protocol/fake_session.h b/remoting/protocol/fake_session.h |
index 912cfb014238f0a0b9034db18903593e060d386f..f0d582871846971b8b38015305f4fefcbb764832 100644 |
--- a/remoting/protocol/fake_session.h |
+++ b/remoting/protocol/fake_session.h |
@@ -5,11 +5,13 @@ |
#ifndef REMOTING_PROTOCOL_FAKE_SESSION_H_ |
#define REMOTING_PROTOCOL_FAKE_SESSION_H_ |
+#include <map> |
#include <string> |
#include <vector> |
#include "base/memory/scoped_ptr.h" |
#include "net/socket/socket.h" |
+#include "net/socket/stream_socket.h" |
#include "remoting/protocol/session.h" |
namespace remoting { |
@@ -22,7 +24,7 @@ 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. |
-class FakeSocket : public net::Socket { |
+class FakeSocket : public net::StreamSocket { |
public: |
FakeSocket(); |
virtual ~FakeSocket(); |
@@ -42,6 +44,21 @@ class FakeSocket : public net::Socket { |
virtual bool SetReceiveBufferSize(int32 size); |
virtual bool SetSendBufferSize(int32 size); |
+ // net::StreamSocket interface. |
+ virtual int Connect(net::CompletionCallback* callback) OVERRIDE; |
+ virtual void Disconnect() OVERRIDE; |
+ virtual bool IsConnected() const OVERRIDE; |
+ virtual bool IsConnectedAndIdle() const OVERRIDE; |
+ virtual int GetPeerAddress(net::AddressList* address) const OVERRIDE; |
+ virtual int GetLocalAddress(net::IPEndPoint* address) const OVERRIDE; |
+ virtual const net::BoundNetLog& NetLog() const OVERRIDE; |
+ virtual void SetSubresourceSpeculation() OVERRIDE; |
+ virtual void SetOmniboxSpeculation() OVERRIDE; |
+ virtual bool WasEverUsed() const OVERRIDE; |
+ virtual bool UsingTCPFastOpen() const OVERRIDE; |
+ virtual int64 NumBytesRead() const OVERRIDE; |
+ virtual base::TimeDelta GetConnectTimeMicros() const OVERRIDE; |
+ |
private: |
bool read_pending_; |
scoped_refptr<net::IOBuffer> read_buffer_; |
@@ -51,6 +68,10 @@ class FakeSocket : public net::Socket { |
std::string written_data_; |
std::string input_data_; |
int input_pos_; |
+ |
+ net::BoundNetLog net_log_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(FakeSocket); |
}; |
// FakeUdpSocket is similar to FakeSocket but behaves as UDP socket. All written |
@@ -86,6 +107,8 @@ class FakeUdpSocket : public net::Socket { |
std::vector<std::string> written_packets_; |
std::vector<std::string> input_packets_; |
int input_pos_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(FakeUdpSocket); |
}; |
// FakeSession is a dummy protocol::Session that uses FakeSocket for all |
@@ -103,6 +126,10 @@ class FakeSession : public Session { |
bool is_closed() const { return closed_; } |
+ FakeSocket* GetStreamChannel(const std::string& name); |
+ FakeUdpSocket* GetDatagramChannel(const std::string& name); |
+ |
+ // Session interface. |
virtual void SetStateChangeCallback(StateChangeCallback* callback); |
virtual void CreateStreamChannel( |
@@ -112,10 +139,6 @@ class FakeSession : public Session { |
virtual FakeSocket* control_channel(); |
virtual FakeSocket* event_channel(); |
- virtual FakeSocket* video_channel(); |
- |
- virtual FakeUdpSocket* video_rtp_channel(); |
- virtual FakeUdpSocket* video_rtcp_channel(); |
virtual const std::string& jid(); |
@@ -140,9 +163,9 @@ class FakeSession : public Session { |
MessageLoop* message_loop_; |
FakeSocket control_channel_; |
FakeSocket event_channel_; |
- FakeSocket video_channel_; |
- FakeUdpSocket video_rtp_channel_; |
- FakeUdpSocket video_rtcp_channel_; |
+ |
+ std::map<std::string, FakeSocket*> stream_channels_; |
+ std::map<std::string, FakeUdpSocket*> datagram_channels_; |
std::string initiator_token_; |
std::string receiver_token_; |
@@ -151,6 +174,8 @@ class FakeSession : public Session { |
std::string jid_; |
bool closed_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(FakeSession); |
}; |
} // namespace protocol |