Index: remoting/protocol/connection_tester.cc |
diff --git a/remoting/protocol/connection_tester.cc b/remoting/protocol/connection_tester.cc |
index 7df32a4b80e2eb0d4fba41cfe88e91d60837ff7f..46c95542a23ab709b475f90c318b74b3a2de9ddb 100644 |
--- a/remoting/protocol/connection_tester.cc |
+++ b/remoting/protocol/connection_tester.cc |
@@ -8,14 +8,14 @@ |
#include "base/message_loop/message_loop.h" |
#include "net/base/io_buffer.h" |
#include "net/base/net_errors.h" |
-#include "net/socket/stream_socket.h" |
+#include "remoting/protocol/p2p_socket.h" |
#include "testing/gtest/include/gtest/gtest.h" |
namespace remoting { |
namespace protocol { |
-StreamConnectionTester::StreamConnectionTester(net::StreamSocket* client_socket, |
- net::StreamSocket* host_socket, |
+StreamConnectionTester::StreamConnectionTester(P2PStreamSocket* client_socket, |
+ P2PStreamSocket* host_socket, |
int message_size, |
int message_count) |
: message_loop_(base::MessageLoop::current()), |
@@ -127,11 +127,12 @@ void StreamConnectionTester::HandleReadResult(int result) { |
} |
} |
-DatagramConnectionTester::DatagramConnectionTester(net::Socket* client_socket, |
- net::Socket* host_socket, |
- int message_size, |
- int message_count, |
- int delay_ms) |
+DatagramConnectionTester::DatagramConnectionTester( |
+ P2PDatagramSocket* client_socket, |
+ P2PDatagramSocket* host_socket, |
+ int message_size, |
+ int message_count, |
+ int delay_ms) |
: message_loop_(base::MessageLoop::current()), |
host_socket_(host_socket), |
client_socket_(client_socket), |