Index: remoting/protocol/connection_tester.h |
diff --git a/remoting/protocol/connection_tester.h b/remoting/protocol/connection_tester.h |
index 6c95c71f81fb36ae9a52abf5377463aa32dbab44..de63a62da50cfc900f77749efd6f5ddc87d9c711 100644 |
--- a/remoting/protocol/connection_tester.h |
+++ b/remoting/protocol/connection_tester.h |
@@ -17,20 +17,21 @@ namespace net { |
class DrainableIOBuffer; |
class GrowableIOBuffer; |
class IOBuffer; |
-class Socket; |
-class StreamSocket; |
} // namespace net |
namespace remoting { |
namespace protocol { |
+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(net::StreamSocket* client_socket, |
- net::StreamSocket* host_socket, |
+ StreamConnectionTester(P2PStreamSocket* client_socket, |
+ P2PStreamSocket* host_socket, |
int message_size, |
int message_count); |
~StreamConnectionTester(); |
@@ -51,8 +52,8 @@ class StreamConnectionTester { |
private: |
base::MessageLoop* message_loop_; |
- net::StreamSocket* host_socket_; |
- net::StreamSocket* client_socket_; |
+ P2PStreamSocket* host_socket_; |
+ P2PStreamSocket* client_socket_; |
int message_size_; |
int test_data_size_; |
bool done_; |
@@ -66,8 +67,8 @@ class StreamConnectionTester { |
class DatagramConnectionTester { |
public: |
- DatagramConnectionTester(net::Socket* client_socket, |
- net::Socket* host_socket, |
+ DatagramConnectionTester(P2PDatagramSocket* client_socket, |
+ P2PDatagramSocket* host_socket, |
int message_size, |
int message_count, |
int delay_ms); |
@@ -86,8 +87,8 @@ class DatagramConnectionTester { |
void HandleReadResult(int result); |
base::MessageLoop* message_loop_; |
- net::Socket* host_socket_; |
- net::Socket* client_socket_; |
+ P2PDatagramSocket* host_socket_; |
+ P2PDatagramSocket* client_socket_; |
int message_size_; |
int message_count_; |
int delay_ms_; |