Custom hosts in QuicTestPacketMaker.

net/quic/test_tools/quic_test_packet_maker.cc

 // Copyright 2013 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.
 
 #include "net/quic/test_tools/quic_test_packet_maker.h"
 
 #include <list>
 
 #include "net/quic/quic_framer.h"
 #include "net/quic/quic_http_utils.h"
 #include "net/quic/quic_utils.h"
 #include "net/quic/test_tools/quic_test_utils.h"
 
 using std::make_pair;
 
 namespace net {
 namespace test {
 
 QuicTestPacketMaker::QuicTestPacketMaker(QuicVersion version,
                                          QuicConnectionId connection_id,
-                                         MockClock* clock)
+                                         MockClock* clock,
+                                         const std::string& host)
     : version_(version),
       connection_id_(connection_id),
       clock_(clock),
+      host_(host),
       spdy_request_framer_(SPDY4),
       spdy_response_framer_(SPDY4) {
 }
 
 QuicTestPacketMaker::~QuicTestPacketMaker() {
 }
 
+void QuicTestPacketMaker::set_hostname(const std::string& host) {
+  host_.assign(host);
+}
+
 scoped_ptr<QuicEncryptedPacket> QuicTestPacketMaker::MakeRstPacket(
     QuicPacketSequenceNumber num,
     bool include_version,
     QuicStreamId stream_id,
     QuicRstStreamErrorCode error_code) {
   QuicPacketHeader header;
   header.public_header.connection_id = connection_id_;
   header.public_header.reset_flag = false;
   header.public_header.version_flag = include_version;
   header.public_header.sequence_number_length = PACKET_1BYTE_SEQUENCE_NUMBER;
@@ skipping 177 matching lines @@
   return MakePacket(header_, QuicFrame(&frame));
 }
 
 SpdyHeaderBlock QuicTestPacketMaker::GetRequestHeaders(
     const std::string& method,
     const std::string& scheme,
     const std::string& path) {
   SpdyHeaderBlock headers;
   headers[":method"] = method;
   if (version_ <= QUIC_VERSION_24) {
-    headers[":host"] = "www.google.com";
+    headers[":host"] = host_;
   } else {
-    headers[":authority"] = "www.google.com";
+    headers[":authority"] = host_;
   }
   headers[":path"] = path;
   headers[":scheme"] = scheme;
   if (version_ <= QUIC_VERSION_24) {
     headers[":version"] = "HTTP/1.1";
   }
   return headers;
 }
 
 SpdyHeaderBlock QuicTestPacketMaker::GetResponseHeaders(
@@ skipping 32 matching lines @@
   header_.public_header.version_flag = should_include_version;
   header_.public_header.sequence_number_length = PACKET_1BYTE_SEQUENCE_NUMBER;
   header_.packet_sequence_number = sequence_number;
   header_.fec_group = 0;
   header_.entropy_flag = false;
   header_.fec_flag = false;
 }
 
 }  // namespace test
 }  // namespace net