Land Recent QUIC Changes.

net/tools/quic/test_tools/quic_test_client.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_
 #define NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/memory/scoped_ptr.h"
 #include "net/base/ip_endpoint.h"
 #include "net/quic/quic_framer.h"
 #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"
 
 namespace net {
 
 class ProofVerifier;
 
 namespace tools {
 
 class QuicPacketWriterWrapper;
 
 namespace test {
 
 class HTTPMessage;
 class MockableQuicClient;
 
 // A quic client which allows mocking out writes.
 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;
   virtual QuicConnectionId GenerateConnectionId() OVERRIDE;
   void UseWriter(QuicPacketWriterWrapper* writer);
   void UseConnectionId(QuicConnectionId connection_id);
 
  private:
   QuicConnectionId override_connection_id_;  // ConnectionId to use, if nonzero
   QuicPacketWriterWrapper* test_writer_;
@@ skipping 18 matching lines @@
                  const QuicConfig& config,
                  const QuicVersionVector& supported_versions);
 
   virtual ~QuicTestClient();
 
   // ExpectCertificates controls whether the server is expected to provide
   // certificates. The certificates, if any, are not verified, but the common
   // name is recorded and available with |cert_common_name()|.
   void ExpectCertificates(bool on);
 
+  // Sets the |user_agent_id| of the |client_|.
+  void SetUserAgentID(const string& user_agent_id);
+
   // Wraps data in a quic packet and sends it.
   ssize_t SendData(string data, bool last_data);
 
   // From SimpleClient
   // Clears any outstanding state and sends a simple GET of 'uri' to the
   // server.  Returns 0 if the request failed and no bytes were written.
   virtual ssize_t SendRequest(const string& uri) OVERRIDE;
   virtual ssize_t SendMessage(const HTTPMessage& message) OVERRIDE;
   virtual string SendCustomSynchronousRequest(
       const HTTPMessage& message) OVERRIDE;
@@ skipping 58 matching lines @@
   void set_auto_reconnect(bool reconnect) { auto_reconnect_ = reconnect; }
 
   void set_priority(QuicPriority priority) { priority_ = priority; }
 
   // Sets client's FEC policy. This policy applies to the data stream(s), and
   // also to the headers and crypto streams.
   void SetFecPolicy(FecPolicy fec_policy);
 
   void WaitForWriteToFlush();
 
+  EpollServer* epoll_server() { return &epoll_server_; }
+
  protected:
   QuicTestClient();
 
   void Initialize(bool secure);
 
   void set_client(MockableQuicClient* client) { client_.reset(client); }
 
  private:
+  EpollServer epoll_server_;
   scoped_ptr<MockableQuicClient> client_;  // The actual client
   QuicSpdyClientStream* stream_;
 
   QuicRstStreamErrorCode stream_error_;
 
   bool response_complete_;
   bool response_headers_complete_;
   BalsaHeaders headers_;
   QuicPriority priority_;
   string response_;
@@ skipping 20 matching lines @@
 
   DISALLOW_COPY_AND_ASSIGN(QuicTestClient);
 };
 
 }  // namespace test
 
 }  // namespace tools
 }  // namespace net
 
 #endif  // NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_