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/tools/balsa/balsa_headers.h" | 13 #include "net/tools/balsa/balsa_headers.h" |
14 #include "net/tools/quic/quic_in_memory_cache.h" | 14 #include "net/tools/quic/quic_in_memory_cache.h" |
15 #include "net/tools/quic/test_tools/quic_in_memory_cache_peer.h" | 15 #include "net/tools/quic/test_tools/quic_in_memory_cache_peer.h" |
16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
17 | 17 |
18 using base::IntToString; | 18 using base::IntToString; |
19 using base::StringPiece; | 19 using base::StringPiece; |
20 using std::string; | 20 using std::string; |
21 | 21 |
22 namespace net { | 22 namespace net { |
23 namespace tools { | 23 namespace tools { |
24 namespace test { | 24 namespace test { |
25 | 25 |
26 class QuicInMemoryCacheTest : public ::testing::Test { | 26 class QuicInMemoryCacheTest : public ::testing::Test { |
27 protected: | 27 protected: |
28 QuicInMemoryCacheTest() { | 28 QuicInMemoryCacheTest() { |
| 29 QuicInMemoryCachePeer::ResetForTests(); |
| 30 } |
| 31 |
| 32 ~QuicInMemoryCacheTest() override { QuicInMemoryCachePeer::ResetForTests(); } |
| 33 |
| 34 void CreateRequest(string host, string path, BalsaHeaders* headers) { |
| 35 headers->SetRequestFirstlineFromStringPieces("GET", path, "HTTP/1.1"); |
| 36 headers->ReplaceOrAppendHeader("host", host); |
| 37 } |
| 38 |
| 39 string CacheDirectory() { |
29 base::FilePath path; | 40 base::FilePath path; |
30 PathService::Get(base::DIR_SOURCE_ROOT, &path); | 41 PathService::Get(base::DIR_SOURCE_ROOT, &path); |
31 path = path.AppendASCII("net").AppendASCII("data") | 42 path = path.AppendASCII("net").AppendASCII("data") |
32 .AppendASCII("quic_in_memory_cache_data"); | 43 .AppendASCII("quic_in_memory_cache_data"); |
33 // The file path is known to be an ascii string. | 44 // The file path is known to be an ascii string. |
34 FLAGS_quic_in_memory_cache_dir = path.MaybeAsASCII(); | 45 return path.MaybeAsASCII(); |
35 } | 46 } |
| 47 }; |
36 | 48 |
37 ~QuicInMemoryCacheTest() override { QuicInMemoryCachePeer::ResetForTests(); } | 49 TEST_F(QuicInMemoryCacheTest, GetResponseNoMatch) { |
38 | 50 const QuicInMemoryCache::Response* response = |
39 void CreateRequest(StringPiece host, | 51 QuicInMemoryCache::GetInstance()->GetResponse("mail.google.com", |
40 StringPiece path, | 52 "/index.html"); |
41 BalsaHeaders* headers) { | 53 ASSERT_FALSE(response); |
42 headers->SetRequestFirstlineFromStringPieces("GET", path, "HTTP/1.1"); | 54 } |
43 headers->ReplaceOrAppendHeader("host", host); | |
44 } | |
45 | |
46 void SetUp() override { QuicInMemoryCachePeer::ResetForTests(); } | |
47 | |
48 }; | |
49 | 55 |
50 TEST_F(QuicInMemoryCacheTest, AddSimpleResponseGetResponse) { | 56 TEST_F(QuicInMemoryCacheTest, AddSimpleResponseGetResponse) { |
51 string response_body("hello response"); | 57 string response_body("hello response"); |
52 QuicInMemoryCache* cache = QuicInMemoryCache::GetInstance(); | 58 QuicInMemoryCache* cache = QuicInMemoryCache::GetInstance(); |
53 cache->AddSimpleResponse("www.google.com", "/", 200, "OK", response_body); | 59 cache->AddSimpleResponse("www.google.com", "/", 200, "OK", response_body); |
54 | 60 |
55 BalsaHeaders request_headers; | 61 BalsaHeaders request_headers; |
56 CreateRequest("www.google.com", "/", &request_headers); | 62 CreateRequest("www.google.com", "/", &request_headers); |
57 const QuicInMemoryCache::Response* response = | 63 const QuicInMemoryCache::Response* response = |
58 cache->GetResponse("www.google.com", "/"); | 64 cache->GetResponse("www.google.com", "/"); |
59 ASSERT_TRUE(response); | 65 ASSERT_TRUE(response); |
60 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 66 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
61 EXPECT_EQ("200 OK", response->headers().find(":status")->second); | 67 EXPECT_EQ("200 OK", response->headers().find(":status")->second); |
62 EXPECT_EQ(response_body.size(), response->body().length()); | 68 EXPECT_EQ(response_body.size(), response->body().length()); |
63 } | 69 } |
64 | 70 |
65 TEST_F(QuicInMemoryCacheTest, ReadsCacheDir) { | 71 TEST_F(QuicInMemoryCacheTest, ReadsCacheDir) { |
| 72 QuicInMemoryCache::GetInstance()->InitializeFromDirectory(CacheDirectory()); |
66 const QuicInMemoryCache::Response* response = | 73 const QuicInMemoryCache::Response* response = |
67 QuicInMemoryCache::GetInstance()->GetResponse("quic.test.url", | 74 QuicInMemoryCache::GetInstance()->GetResponse("quic.test.url", |
68 "/index.html"); | 75 "/index.html"); |
69 ASSERT_TRUE(response); | 76 ASSERT_TRUE(response); |
70 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 77 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
71 EXPECT_EQ("200 OK", response->headers().find(":status")->second); | 78 EXPECT_EQ("200 OK", response->headers().find(":status")->second); |
72 ASSERT_TRUE(ContainsKey(response->headers(), "connection")); | 79 ASSERT_TRUE(ContainsKey(response->headers(), "connection")); |
73 EXPECT_EQ("close", response->headers().find("connection")->second); | 80 EXPECT_EQ("close", response->headers().find("connection")->second); |
74 EXPECT_LT(0U, response->body().length()); | 81 EXPECT_LT(0U, response->body().length()); |
75 } | 82 } |
76 | 83 |
77 TEST_F(QuicInMemoryCacheTest, UsesOriginalUrl) { | 84 TEST_F(QuicInMemoryCacheTest, UsesOriginalUrl) { |
| 85 QuicInMemoryCache::GetInstance()->InitializeFromDirectory(CacheDirectory()); |
78 const QuicInMemoryCache::Response* response = | 86 const QuicInMemoryCache::Response* response = |
79 QuicInMemoryCache::GetInstance()->GetResponse("quic.test.url", | 87 QuicInMemoryCache::GetInstance()->GetResponse("quic.test.url", |
80 "/index.html"); | 88 "/index.html"); |
81 ASSERT_TRUE(response); | 89 ASSERT_TRUE(response); |
82 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); | 90 ASSERT_TRUE(ContainsKey(response->headers(), ":status")); |
83 EXPECT_EQ("200 OK", response->headers().find(":status")->second); | 91 EXPECT_EQ("200 OK", response->headers().find(":status")->second); |
84 ASSERT_TRUE(ContainsKey(response->headers(), "connection")); | 92 ASSERT_TRUE(ContainsKey(response->headers(), "connection")); |
85 EXPECT_EQ("close", response->headers().find("connection")->second); | 93 EXPECT_EQ("close", response->headers().find("connection")->second); |
86 EXPECT_LT(0U, response->body().length()); | 94 EXPECT_LT(0U, response->body().length()); |
87 } | 95 } |
88 | 96 |
89 TEST_F(QuicInMemoryCacheTest, GetResponseNoMatch) { | |
90 const QuicInMemoryCache::Response* response = | |
91 QuicInMemoryCache::GetInstance()->GetResponse("mail.google.com", | |
92 "/index.html"); | |
93 ASSERT_FALSE(response); | |
94 } | |
95 | |
96 } // namespace test | 97 } // namespace test |
97 } // namespace tools | 98 } // namespace tools |
98 } // namespace net | 99 } // namespace net |
OLD | NEW |