Index: net/tools/quic/test_tools/quic_test_client.h |
diff --git a/net/tools/quic/test_tools/quic_test_client.h b/net/tools/quic/test_tools/quic_test_client.h |
index 1c70c33a18644164525c951feb1350f32c5387dd..1201602bae6ea21aaee8dbe787705c4ea1449b81 100644 |
--- a/net/tools/quic/test_tools/quic_test_client.h |
+++ b/net/tools/quic/test_tools/quic_test_client.h |
@@ -14,6 +14,7 @@ |
#include "net/quic/quic_packet_creator.h" |
#include "net/quic/quic_protocol.h" |
#include "net/tools/balsa/balsa_frame.h" |
+#include "net/tools/epoll_server/epoll_server.h" |
#include "net/tools/quic/quic_client.h" |
#include "net/tools/quic/test_tools/simple_client.h" |
@@ -35,12 +36,14 @@ class MockableQuicClient : public QuicClient { |
public: |
MockableQuicClient(IPEndPoint server_address, |
const QuicServerId& server_id, |
- const QuicVersionVector& supported_versions); |
+ const QuicVersionVector& supported_versions, |
+ EpollServer* epoll_server); |
MockableQuicClient(IPEndPoint server_address, |
const QuicServerId& server_id, |
const QuicConfig& config, |
- const QuicVersionVector& supported_versions); |
+ const QuicVersionVector& supported_versions, |
+ EpollServer* epoll_server); |
virtual ~MockableQuicClient() OVERRIDE; |
virtual QuicPacketWriter* CreateQuicPacketWriter() OVERRIDE; |
@@ -157,6 +160,8 @@ class QuicTestClient : public SimpleClient, |
void WaitForWriteToFlush(); |
+ EpollServer* epoll_server() { return &epoll_server_; } |
+ |
protected: |
QuicTestClient(); |
@@ -165,6 +170,7 @@ class QuicTestClient : public SimpleClient, |
void set_client(MockableQuicClient* client) { client_.reset(client); } |
private: |
+ EpollServer epoll_server_; |
scoped_ptr<MockableQuicClient> client_; // The actual client |
QuicSpdyClientStream* stream_; |