Landing recent QUIC changes until 6/24/2016 14:00 UTC

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 <stddef.h>
 #include <stdint.h>
 
@@ skipping 184 matching lines @@
   void set_allow_bidirectional_data(bool value) {
     allow_bidirectional_data_ = value;
   }
 
   bool allow_bidirectional_data() const { return allow_bidirectional_data_; }
 
   size_t num_requests() const { return num_requests_; }
 
   size_t num_responses() const { return num_responses_; }
 
+  // Explicitly set the SNI value for this client, overriding the default
+  // behavior which extracts the SNI value from the request URL.
+  void OverrideSni(const std::string& sni) {
+    override_sni_set_ = true;
+    override_sni_ = sni;
+  }
+
  protected:
   QuicTestClient();
 
   void Initialize();
 
   void set_client(MockableQuicClient* client) { client_.reset(client); }
 
  private:
   class TestClientDataToResend : public QuicClient::QuicDataToResend {
    public:
@@ skipping 52 matching lines @@
   // When true allows the sending of a request to continue while the response is
   // arriving.
   bool allow_bidirectional_data_;
   // For async push promise rendezvous, validation may fail in which
   // case the request should be retried.
   std::unique_ptr<TestClientDataToResend> push_promise_data_to_resend_;
   // Number of requests/responses this client has sent/received.
   size_t num_requests_;
   size_t num_responses_;
 
+  // If set, this value is used for the connection SNI, overriding the usual
+  // logic which extracts the SNI from the request URL.
+  bool override_sni_set_ = false;
+  std::string override_sni_;
+
   DISALLOW_COPY_AND_ASSIGN(QuicTestClient);
 };
 
 }  // namespace test
 
 }  // namespace net
 
 #endif  // NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_CLIENT_H_