| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <string> | 5 #include <string> |
| 6 | 6 |
| 7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
| 8 #include "base/memory/singleton.h" | 8 #include "base/memory/singleton.h" |
| 9 #include "base/path_service.h" | 9 #include "base/path_service.h" |
| 10 #include "base/stl_util.h" | 10 #include "base/stl_util.h" |
| 11 #include "base/strings/string_number_conversions.h" | 11 #include "base/strings/string_number_conversions.h" |
| 12 #include "base/strings/string_piece.h" | 12 #include "base/strings/string_piece.h" |
| 13 #include "net/spdy/spdy_framer.h" | 13 #include "net/spdy/spdy_framer.h" |
| 14 #include "net/tools/quic/quic_in_memory_cache.h" | 14 #include "net/tools/quic/quic_http_response_cache.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 16 |
| 17 using base::ContainsKey; | 17 using base::ContainsKey; |
| 18 using base::IntToString; | 18 using base::IntToString; |
| 19 using base::StringPiece; | 19 using base::StringPiece; |
| 20 using net::SpdyHeaderBlock; | 20 using net::SpdyHeaderBlock; |
| 21 using std::list; | 21 using std::list; |
| 22 using std::string; | 22 using std::string; |
| 23 | 23 |
| 24 namespace net { | 24 namespace net { |
| 25 namespace test { | 25 namespace test { |
| 26 | 26 |
| 27 namespace { | 27 namespace { |
| 28 typedef QuicInMemoryCache::Response Response; | 28 typedef QuicHttpResponseCache::Response Response; |
| 29 typedef QuicInMemoryCache::ServerPushInfo ServerPushInfo; | 29 typedef QuicHttpResponseCache::ServerPushInfo ServerPushInfo; |
| 30 }; // namespace | 30 }; // namespace |
| 31 | 31 |
| 32 class QuicInMemoryCacheTest : public ::testing::Test { | 32 class QuicHttpResponseCacheTest : public ::testing::Test { |
| 33 protected: | 33 protected: |
| 34 void CreateRequest(string host, string path, SpdyHeaderBlock* headers) { | 34 void CreateRequest(string host, string path, SpdyHeaderBlock* headers) { |
| 35 (*headers)[":method"] = "GET"; | 35 (*headers)[":method"] = "GET"; |
| 36 (*headers)[":path"] = path; | 36 (*headers)[":path"] = path; |
| 37 (*headers)[":authority"] = host; | 37 (*headers)[":authority"] = host; |
| 38 (*headers)[":scheme"] = "https"; | 38 (*headers)[":scheme"] = "https"; |
| 39 } | 39 } |
| 40 | 40 |
| 41 string CacheDirectory() { | 41 string CacheDirectory() { |
| 42 base::FilePath path; | 42 base::FilePath path; |
| 43 PathService::Get(base::DIR_SOURCE_ROOT, &path); | 43 PathService::Get(base::DIR_SOURCE_ROOT, &path); |
| 44 path = path.AppendASCII("net").AppendASCII("data").AppendASCII( | 44 path = path.AppendASCII("net").AppendASCII("data").AppendASCII( |
| 45 "quic_in_memory_cache_data"); | 45 "quic_http_response_cache_data"); |
| 46 // The file path is known to be an ascii string. | 46 // The file path is known to be an ascii string. |
| 47 return path.MaybeAsASCII(); | 47 return path.MaybeAsASCII(); |
| 48 } | 48 } |
| 49 | 49 |
| 50 QuicInMemoryCache cache_; | 50 QuicHttpResponseCache cache_; |
| 51 }; | 51 }; |
| 52 | 52 |
| 53 TEST_F(QuicInMemoryCacheTest, GetResponseNoMatch) { | 53 TEST_F(QuicHttpResponseCacheTest, GetResponseNoMatch) { |
| 54 const QuicInMemoryCache::Response* response = | 54 const QuicHttpResponseCache::Response* response = |
| 55 cache_.GetResponse("mail.google.com", "/index.html"); | 55 cache_.GetResponse("mail.google.com", "/index.html"); |
| 56 ASSERT_FALSE(response); | 56 ASSERT_FALSE(response); |
| 57 } | 57 } |
| 58 | 58 |
| 59 TEST_F(QuicInMemoryCacheTest, AddSimpleResponseGetResponse) { | 59 TEST_F(QuicHttpResponseCacheTest, AddSimpleResponseGetResponse) { |
| 60 string response_body("hello response"); | 60 string response_body("hello response"); |
| 61 cache_.AddSimpleResponse("www.google.com", "/", 200, response_body); | 61 cache_.AddSimpleResponse("www.google.com", "/", 200, response_body); |
| 62 | 62 |
| 63 SpdyHeaderBlock request_headers; | 63 SpdyHeaderBlock request_headers; |
| 64 CreateRequest("www.google.com", "/", &request_headers); | 64 CreateRequest("www.google.com", "/", &request_headers); |
| 65 const QuicInMemoryCache::Response* response = | 65 const QuicHttpResponseCache::Response* response = |
| 66 cache_.GetResponse("www.google.com", "/"); | 66 cache_.GetResponse("www.google.com", "/"); |
| 67 ASSERT_TRUE(response); | 67 ASSERT_TRUE(response); |
| 68 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 68 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
| 69 EXPECT_EQ("200", response->headers().find(":status")->second); | 69 EXPECT_EQ("200", response->headers().find(":status")->second); |
| 70 EXPECT_EQ(response_body.size(), response->body().length()); | 70 EXPECT_EQ(response_body.size(), response->body().length()); |
| 71 } | 71 } |
| 72 | 72 |
| 73 TEST_F(QuicInMemoryCacheTest, AddResponse) { | 73 TEST_F(QuicHttpResponseCacheTest, AddResponse) { |
| 74 const string kRequestHost = "www.foo.com"; | 74 const string kRequestHost = "www.foo.com"; |
| 75 const string kRequestPath = "/"; | 75 const string kRequestPath = "/"; |
| 76 const string kResponseBody("hello response"); | 76 const string kResponseBody("hello response"); |
| 77 | 77 |
| 78 SpdyHeaderBlock response_headers; | 78 SpdyHeaderBlock response_headers; |
| 79 response_headers[":version"] = "HTTP/1.1"; | 79 response_headers[":version"] = "HTTP/1.1"; |
| 80 response_headers[":status"] = "200"; | 80 response_headers[":status"] = "200"; |
| 81 response_headers["content-length"] = IntToString(kResponseBody.size()); | 81 response_headers["content-length"] = IntToString(kResponseBody.size()); |
| 82 | 82 |
| 83 SpdyHeaderBlock response_trailers; | 83 SpdyHeaderBlock response_trailers; |
| 84 response_trailers["key-1"] = "value-1"; | 84 response_trailers["key-1"] = "value-1"; |
| 85 response_trailers["key-2"] = "value-2"; | 85 response_trailers["key-2"] = "value-2"; |
| 86 response_trailers["key-3"] = "value-3"; | 86 response_trailers["key-3"] = "value-3"; |
| 87 | 87 |
| 88 cache_.AddResponse(kRequestHost, "/", response_headers.Clone(), kResponseBody, | 88 cache_.AddResponse(kRequestHost, "/", response_headers.Clone(), kResponseBody, |
| 89 response_trailers.Clone()); | 89 response_trailers.Clone()); |
| 90 | 90 |
| 91 const QuicInMemoryCache::Response* response = | 91 const QuicHttpResponseCache::Response* response = |
| 92 cache_.GetResponse(kRequestHost, kRequestPath); | 92 cache_.GetResponse(kRequestHost, kRequestPath); |
| 93 EXPECT_EQ(response->headers(), response_headers); | 93 EXPECT_EQ(response->headers(), response_headers); |
| 94 EXPECT_EQ(response->body(), kResponseBody); | 94 EXPECT_EQ(response->body(), kResponseBody); |
| 95 EXPECT_EQ(response->trailers(), response_trailers); | 95 EXPECT_EQ(response->trailers(), response_trailers); |
| 96 } | 96 } |
| 97 | 97 |
| 98 TEST_F(QuicInMemoryCacheTest, ReadsCacheDir) { | 98 TEST_F(QuicHttpResponseCacheTest, ReadsCacheDir) { |
| 99 cache_.InitializeFromDirectory(CacheDirectory()); | 99 cache_.InitializeFromDirectory(CacheDirectory()); |
| 100 const QuicInMemoryCache::Response* response = | 100 const QuicHttpResponseCache::Response* response = |
| 101 cache_.GetResponse("quic.test.url", "/index.html"); | 101 cache_.GetResponse("quic.test.url", "/index.html"); |
| 102 ASSERT_TRUE(response); | 102 ASSERT_TRUE(response); |
| 103 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 103 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
| 104 EXPECT_EQ("200", response->headers().find(":status")->second); | 104 EXPECT_EQ("200", response->headers().find(":status")->second); |
| 105 // Connection headers are not valid in HTTP/2. | 105 // Connection headers are not valid in HTTP/2. |
| 106 EXPECT_FALSE(ContainsKey(response->headers(), "connection")); | 106 EXPECT_FALSE(ContainsKey(response->headers(), "connection")); |
| 107 EXPECT_LT(0U, response->body().length()); | 107 EXPECT_LT(0U, response->body().length()); |
| 108 } | 108 } |
| 109 | 109 |
| 110 TEST_F(QuicInMemoryCacheTest, ReadsCacheDirWithServerPushResource) { | 110 TEST_F(QuicHttpResponseCacheTest, ReadsCacheDirWithServerPushResource) { |
| 111 cache_.InitializeFromDirectory(CacheDirectory() + "_with_push"); | 111 cache_.InitializeFromDirectory(CacheDirectory() + "_with_push"); |
| 112 list<ServerPushInfo> resources = | 112 list<ServerPushInfo> resources = |
| 113 cache_.GetServerPushResources("quic.test.url/"); | 113 cache_.GetServerPushResources("quic.test.url/"); |
| 114 ASSERT_EQ(1UL, resources.size()); | 114 ASSERT_EQ(1UL, resources.size()); |
| 115 } | 115 } |
| 116 | 116 |
| 117 TEST_F(QuicInMemoryCacheTest, ReadsCacheDirWithServerPushResources) { | 117 TEST_F(QuicHttpResponseCacheTest, ReadsCacheDirWithServerPushResources) { |
| 118 cache_.InitializeFromDirectory(CacheDirectory() + "_with_push"); | 118 cache_.InitializeFromDirectory(CacheDirectory() + "_with_push"); |
| 119 list<ServerPushInfo> resources = | 119 list<ServerPushInfo> resources = |
| 120 cache_.GetServerPushResources("quic.test.url/index2.html"); | 120 cache_.GetServerPushResources("quic.test.url/index2.html"); |
| 121 ASSERT_EQ(2UL, resources.size()); | 121 ASSERT_EQ(2UL, resources.size()); |
| 122 } | 122 } |
| 123 | 123 |
| 124 TEST_F(QuicInMemoryCacheTest, UsesOriginalUrl) { | 124 TEST_F(QuicHttpResponseCacheTest, UsesOriginalUrl) { |
| 125 cache_.InitializeFromDirectory(CacheDirectory()); | 125 cache_.InitializeFromDirectory(CacheDirectory()); |
| 126 const QuicInMemoryCache::Response* response = | 126 const QuicHttpResponseCache::Response* response = |
| 127 cache_.GetResponse("quic.test.url", "/index.html"); | 127 cache_.GetResponse("quic.test.url", "/index.html"); |
| 128 ASSERT_TRUE(response); | 128 ASSERT_TRUE(response); |
| 129 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 129 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
| 130 EXPECT_EQ("200", response->headers().find(":status")->second); | 130 EXPECT_EQ("200", response->headers().find(":status")->second); |
| 131 // Connection headers are not valid in HTTP/2. | 131 // Connection headers are not valid in HTTP/2. |
| 132 EXPECT_FALSE(ContainsKey(response->headers(), "connection")); | 132 EXPECT_FALSE(ContainsKey(response->headers(), "connection")); |
| 133 EXPECT_LT(0U, response->body().length()); | 133 EXPECT_LT(0U, response->body().length()); |
| 134 } | 134 } |
| 135 | 135 |
| 136 TEST_F(QuicInMemoryCacheTest, DefaultResponse) { | 136 TEST_F(QuicHttpResponseCacheTest, DefaultResponse) { |
| 137 // Verify GetResponse returns nullptr when no default is set. | 137 // Verify GetResponse returns nullptr when no default is set. |
| 138 const QuicInMemoryCache::Response* response = | 138 const QuicHttpResponseCache::Response* response = |
| 139 cache_.GetResponse("www.google.com", "/"); | 139 cache_.GetResponse("www.google.com", "/"); |
| 140 ASSERT_FALSE(response); | 140 ASSERT_FALSE(response); |
| 141 | 141 |
| 142 // Add a default response. | 142 // Add a default response. |
| 143 SpdyHeaderBlock response_headers; | 143 SpdyHeaderBlock response_headers; |
| 144 response_headers[":version"] = "HTTP/1.1"; | 144 response_headers[":version"] = "HTTP/1.1"; |
| 145 response_headers[":status"] = "200"; | 145 response_headers[":status"] = "200"; |
| 146 response_headers["content-length"] = "0"; | 146 response_headers["content-length"] = "0"; |
| 147 QuicInMemoryCache::Response* default_response = | 147 QuicHttpResponseCache::Response* default_response = |
| 148 new QuicInMemoryCache::Response; | 148 new QuicHttpResponseCache::Response; |
| 149 default_response->set_headers(std::move(response_headers)); | 149 default_response->set_headers(std::move(response_headers)); |
| 150 cache_.AddDefaultResponse(default_response); | 150 cache_.AddDefaultResponse(default_response); |
| 151 | 151 |
| 152 // Now we should get the default response for the original request. | 152 // Now we should get the default response for the original request. |
| 153 response = cache_.GetResponse("www.google.com", "/"); | 153 response = cache_.GetResponse("www.google.com", "/"); |
| 154 ASSERT_TRUE(response); | 154 ASSERT_TRUE(response); |
| 155 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 155 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
| 156 EXPECT_EQ("200", response->headers().find(":status")->second); | 156 EXPECT_EQ("200", response->headers().find(":status")->second); |
| 157 | 157 |
| 158 // Now add a set response for / and make sure it is returned | 158 // Now add a set response for / and make sure it is returned |
| 159 cache_.AddSimpleResponse("www.google.com", "/", 302, ""); | 159 cache_.AddSimpleResponse("www.google.com", "/", 302, ""); |
| 160 response = cache_.GetResponse("www.google.com", "/"); | 160 response = cache_.GetResponse("www.google.com", "/"); |
| 161 ASSERT_TRUE(response); | 161 ASSERT_TRUE(response); |
| 162 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 162 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
| 163 EXPECT_EQ("302", response->headers().find(":status")->second); | 163 EXPECT_EQ("302", response->headers().find(":status")->second); |
| 164 | 164 |
| 165 // We should get the default response for other requests. | 165 // We should get the default response for other requests. |
| 166 response = cache_.GetResponse("www.google.com", "/asd"); | 166 response = cache_.GetResponse("www.google.com", "/asd"); |
| 167 ASSERT_TRUE(response); | 167 ASSERT_TRUE(response); |
| 168 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 168 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
| 169 EXPECT_EQ("200", response->headers().find(":status")->second); | 169 EXPECT_EQ("200", response->headers().find(":status")->second); |
| 170 } | 170 } |
| 171 | 171 |
| 172 TEST_F(QuicInMemoryCacheTest, AddSimpleResponseWithServerPushResources) { | 172 TEST_F(QuicHttpResponseCacheTest, AddSimpleResponseWithServerPushResources) { |
| 173 string request_host = "www.foo.com"; | 173 string request_host = "www.foo.com"; |
| 174 string response_body("hello response"); | 174 string response_body("hello response"); |
| 175 const size_t kNumResources = 5; | 175 const size_t kNumResources = 5; |
| 176 int NumResources = 5; | 176 int NumResources = 5; |
| 177 list<QuicInMemoryCache::ServerPushInfo> push_resources; | 177 list<QuicHttpResponseCache::ServerPushInfo> push_resources; |
| 178 string scheme = "http"; | 178 string scheme = "http"; |
| 179 for (int i = 0; i < NumResources; ++i) { | 179 for (int i = 0; i < NumResources; ++i) { |
| 180 string path = "/server_push_src" + base::IntToString(i); | 180 string path = "/server_push_src" + base::IntToString(i); |
| 181 string url = scheme + "://" + request_host + path; | 181 string url = scheme + "://" + request_host + path; |
| 182 GURL resource_url(url); | 182 GURL resource_url(url); |
| 183 string body = "This is server push response body for " + path; | 183 string body = "This is server push response body for " + path; |
| 184 SpdyHeaderBlock response_headers; | 184 SpdyHeaderBlock response_headers; |
| 185 response_headers[":version"] = "HTTP/1.1"; | 185 response_headers[":version"] = "HTTP/1.1"; |
| 186 response_headers[":status"] = "200"; | 186 response_headers[":status"] = "200"; |
| 187 response_headers["content-length"] = base::UintToString(body.size()); | 187 response_headers["content-length"] = base::UintToString(body.size()); |
| 188 push_resources.push_back( | 188 push_resources.push_back( |
| 189 ServerPushInfo(resource_url, response_headers.Clone(), i, body)); | 189 ServerPushInfo(resource_url, response_headers.Clone(), i, body)); |
| 190 } | 190 } |
| 191 | 191 |
| 192 cache_.AddSimpleResponseWithServerPushResources( | 192 cache_.AddSimpleResponseWithServerPushResources( |
| 193 request_host, "/", 200, response_body, push_resources); | 193 request_host, "/", 200, response_body, push_resources); |
| 194 string request_url = request_host + "/"; | 194 string request_url = request_host + "/"; |
| 195 list<ServerPushInfo> resources = cache_.GetServerPushResources(request_url); | 195 list<ServerPushInfo> resources = cache_.GetServerPushResources(request_url); |
| 196 ASSERT_EQ(kNumResources, resources.size()); | 196 ASSERT_EQ(kNumResources, resources.size()); |
| 197 for (const auto& push_resource : push_resources) { | 197 for (const auto& push_resource : push_resources) { |
| 198 ServerPushInfo resource = resources.front(); | 198 ServerPushInfo resource = resources.front(); |
| 199 EXPECT_EQ(resource.request_url.spec(), push_resource.request_url.spec()); | 199 EXPECT_EQ(resource.request_url.spec(), push_resource.request_url.spec()); |
| 200 EXPECT_EQ(resource.priority, push_resource.priority); | 200 EXPECT_EQ(resource.priority, push_resource.priority); |
| 201 resources.pop_front(); | 201 resources.pop_front(); |
| 202 } | 202 } |
| 203 } | 203 } |
| 204 | 204 |
| 205 TEST_F(QuicInMemoryCacheTest, GetServerPushResourcesAndPushResponses) { | 205 TEST_F(QuicHttpResponseCacheTest, GetServerPushResourcesAndPushResponses) { |
| 206 string request_host = "www.foo.com"; | 206 string request_host = "www.foo.com"; |
| 207 string response_body("hello response"); | 207 string response_body("hello response"); |
| 208 const size_t kNumResources = 4; | 208 const size_t kNumResources = 4; |
| 209 int NumResources = 4; | 209 int NumResources = 4; |
| 210 string scheme = "http"; | 210 string scheme = "http"; |
| 211 string push_response_status[kNumResources] = {"200", "200", "301", "404"}; | 211 string push_response_status[kNumResources] = {"200", "200", "301", "404"}; |
| 212 list<QuicInMemoryCache::ServerPushInfo> push_resources; | 212 list<QuicHttpResponseCache::ServerPushInfo> push_resources; |
| 213 for (int i = 0; i < NumResources; ++i) { | 213 for (int i = 0; i < NumResources; ++i) { |
| 214 string path = "/server_push_src" + base::IntToString(i); | 214 string path = "/server_push_src" + base::IntToString(i); |
| 215 string url = scheme + "://" + request_host + path; | 215 string url = scheme + "://" + request_host + path; |
| 216 GURL resource_url(url); | 216 GURL resource_url(url); |
| 217 string body = "This is server push response body for " + path; | 217 string body = "This is server push response body for " + path; |
| 218 SpdyHeaderBlock response_headers; | 218 SpdyHeaderBlock response_headers; |
| 219 response_headers[":version"] = "HTTP/1.1"; | 219 response_headers[":version"] = "HTTP/1.1"; |
| 220 response_headers[":status"] = push_response_status[i]; | 220 response_headers[":status"] = push_response_status[i]; |
| 221 response_headers["content-length"] = base::UintToString(body.size()); | 221 response_headers["content-length"] = base::UintToString(body.size()); |
| 222 push_resources.push_back( | 222 push_resources.push_back( |
| 223 ServerPushInfo(resource_url, response_headers.Clone(), i, body)); | 223 ServerPushInfo(resource_url, response_headers.Clone(), i, body)); |
| 224 } | 224 } |
| 225 cache_.AddSimpleResponseWithServerPushResources( | 225 cache_.AddSimpleResponseWithServerPushResources( |
| 226 request_host, "/", 200, response_body, push_resources); | 226 request_host, "/", 200, response_body, push_resources); |
| 227 string request_url = request_host + "/"; | 227 string request_url = request_host + "/"; |
| 228 list<ServerPushInfo> resources = cache_.GetServerPushResources(request_url); | 228 list<ServerPushInfo> resources = cache_.GetServerPushResources(request_url); |
| 229 ASSERT_EQ(kNumResources, resources.size()); | 229 ASSERT_EQ(kNumResources, resources.size()); |
| 230 int i = 0; | 230 int i = 0; |
| 231 for (const auto& push_resource : push_resources) { | 231 for (const auto& push_resource : push_resources) { |
| 232 GURL url = resources.front().request_url; | 232 GURL url = resources.front().request_url; |
| 233 string host = url.host(); | 233 string host = url.host(); |
| 234 string path = url.path(); | 234 string path = url.path(); |
| 235 const QuicInMemoryCache::Response* response = | 235 const QuicHttpResponseCache::Response* response = |
| 236 cache_.GetResponse(host, path); | 236 cache_.GetResponse(host, path); |
| 237 ASSERT_TRUE(response); | 237 ASSERT_TRUE(response); |
| 238 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 238 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
| 239 EXPECT_EQ(push_response_status[i++], | 239 EXPECT_EQ(push_response_status[i++], |
| 240 response->headers().find(":status")->second); | 240 response->headers().find(":status")->second); |
| 241 EXPECT_EQ(push_resource.body, response->body()); | 241 EXPECT_EQ(push_resource.body, response->body()); |
| 242 resources.pop_front(); | 242 resources.pop_front(); |
| 243 } | 243 } |
| 244 } | 244 } |
| 245 | 245 |
| 246 } // namespace test | 246 } // namespace test |
| 247 } // namespace net | 247 } // namespace net |
| OLD | NEW |