Make some changes to the QuicInMemory server to clarify the interface.

net/tools/quic/quic_in_memory_cache.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_QUIC_IN_MEMORY_CACHE_H_
 #define NET_TOOLS_QUIC_QUIC_IN_MEMORY_CACHE_H_
 
 #include <string>
 
 #include "base/containers/hash_tables.h"
@@ skipping 33 matching lines @@
     Response() : response_type_(REGULAR_RESPONSE) {}
     ~Response() {}
 
     SpecialResponseType response_type() const { return response_type_; }
     const BalsaHeaders& headers() const { return headers_; }
     const base::StringPiece body() const { return base::StringPiece(body_); }
 
    private:
     friend class QuicInMemoryCache;
 
+    void set_response_type(SpecialResponseType response_type) {
+      response_type_ = response_type;
+    }
     void set_headers(const BalsaHeaders& headers) {
       headers_.CopyFrom(headers);
     }
     void set_body(base::StringPiece body) {
       body.CopyToString(&body_);
     }
 
     SpecialResponseType response_type_;
     BalsaHeaders headers_;
     std::string body_;
 
     DISALLOW_COPY_AND_ASSIGN(Response);
   };
 
   // Returns the singleton instance of the cache.
   static QuicInMemoryCache* GetInstance();
 
-  // Retrieve a response from this cache for a given request.
+  // Retrieve a response from this cache for a given host and path..
   // If no appropriate response exists, nullptr is returned.
-  // Currently, responses are selected based on request URI only.
-  const Response* GetResponse(const BalsaHeaders& request_headers) const;
+  const Response* GetResponse(base::StringPiece host,
+                              base::StringPiece path) const;
 
   // Adds a simple response to the cache.  The response headers will
-  // only contain the "content-length" header with the lenght of |body|.
-  void AddSimpleResponse(base::StringPiece method,
+  // only contain the "content-length" header with the length of |body|.
+  void AddSimpleResponse(base::StringPiece host,
                          base::StringPiece path,
-                         base::StringPiece version,
-                         base::StringPiece response_code,
+                         int response_code,
                          base::StringPiece response_detail,
                          base::StringPiece body);
 
   // Add a response to the cache.
-  void AddResponse(const BalsaHeaders& request_headers,
+  void AddResponse(base::StringPiece host,
+                   base::StringPiece path,
                    const BalsaHeaders& response_headers,
                    base::StringPiece response_body);
 
   // Simulate a special behavior at a particular path.
-  void AddSpecialResponse(base::StringPiece method,
+  void AddSpecialResponse(base::StringPiece host,
                           base::StringPiece path,
-                          base::StringPiece version,
                           SpecialResponseType response_type);
 
  private:
   typedef base::hash_map<std::string, Response*> ResponseMap;
   friend struct DefaultSingletonTraits<QuicInMemoryCache>;
   friend class test::QuicInMemoryCachePeer;
 
   QuicInMemoryCache();
   ~QuicInMemoryCache();
 
   void ResetForTests();
 
   void Initialize();
 
-  std::string GetKey(const BalsaHeaders& response_headers) const;
+  void AddResponseImpl(base::StringPiece host,
+                       base::StringPiece path,
+                       SpecialResponseType response_type,
+                       const BalsaHeaders& response_headers,
+                       base::StringPiece response_body);
+
+  std::string GetKey(base::StringPiece host, base::StringPiece path) const;
 
   // Cached responses.
   ResponseMap responses_;
 
   DISALLOW_COPY_AND_ASSIGN(QuicInMemoryCache);
 };
 
 }  // namespace tools
 }  // namespace net
 
 #endif  // NET_TOOLS_QUIC_QUIC_IN_MEMORY_CACHE_H_