OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/command_line.h" | 5 #include "base/command_line.h" |
6 #include "base/memory/ref_counted.h" | 6 #include "base/memory/ref_counted.h" |
7 #include "base/message_loop/message_loop_proxy.h" | 7 #include "base/message_loop/message_loop_proxy.h" |
8 #include "base/synchronization/waitable_event.h" | 8 #include "base/synchronization/waitable_event.h" |
9 #include "base/threading/thread.h" | 9 #include "base/threading/thread.h" |
10 #include "base/values.h" | 10 #include "base/values.h" |
(...skipping 20 matching lines...) Expand all Loading... |
31 : public net::TestURLRequestContextGetter { | 31 : public net::TestURLRequestContextGetter { |
32 public: | 32 public: |
33 explicit TrackingTestURLRequestContextGetter( | 33 explicit TrackingTestURLRequestContextGetter( |
34 base::MessageLoopProxy* io_message_loop_proxy, | 34 base::MessageLoopProxy* io_message_loop_proxy, |
35 net::URLRequestThrottlerManager* throttler_manager) | 35 net::URLRequestThrottlerManager* throttler_manager) |
36 : TestURLRequestContextGetter(io_message_loop_proxy), | 36 : TestURLRequestContextGetter(io_message_loop_proxy), |
37 throttler_manager_(throttler_manager) { | 37 throttler_manager_(throttler_manager) { |
38 g_request_context_getter_instances++; | 38 g_request_context_getter_instances++; |
39 } | 39 } |
40 | 40 |
41 virtual net::TestURLRequestContext* GetURLRequestContext() override { | 41 net::TestURLRequestContext* GetURLRequestContext() override { |
42 if (!context_.get()) { | 42 if (!context_.get()) { |
43 context_.reset(new net::TestURLRequestContext(true)); | 43 context_.reset(new net::TestURLRequestContext(true)); |
44 context_->set_throttler_manager(throttler_manager_); | 44 context_->set_throttler_manager(throttler_manager_); |
45 context_->Init(); | 45 context_->Init(); |
46 } | 46 } |
47 return context_.get(); | 47 return context_.get(); |
48 } | 48 } |
49 | 49 |
50 protected: | 50 protected: |
51 virtual ~TrackingTestURLRequestContextGetter() { | 51 ~TrackingTestURLRequestContextGetter() override { |
52 g_request_context_getter_instances--; | 52 g_request_context_getter_instances--; |
53 } | 53 } |
54 | 54 |
55 private: | 55 private: |
56 // Not owned here. | 56 // Not owned here. |
57 net::URLRequestThrottlerManager* throttler_manager_; | 57 net::URLRequestThrottlerManager* throttler_manager_; |
58 scoped_ptr<net::TestURLRequestContext> context_; | 58 scoped_ptr<net::TestURLRequestContext> context_; |
59 }; | 59 }; |
60 | 60 |
61 class TestCloudPrintURLFetcher : public CloudPrintURLFetcher { | 61 class TestCloudPrintURLFetcher : public CloudPrintURLFetcher { |
62 public: | 62 public: |
63 explicit TestCloudPrintURLFetcher( | 63 explicit TestCloudPrintURLFetcher( |
64 base::MessageLoopProxy* io_message_loop_proxy) | 64 base::MessageLoopProxy* io_message_loop_proxy) |
65 : io_message_loop_proxy_(io_message_loop_proxy) { | 65 : io_message_loop_proxy_(io_message_loop_proxy) { |
66 } | 66 } |
67 | 67 |
68 virtual net::URLRequestContextGetter* GetRequestContextGetter() override { | 68 net::URLRequestContextGetter* GetRequestContextGetter() override { |
69 return new TrackingTestURLRequestContextGetter( | 69 return new TrackingTestURLRequestContextGetter( |
70 io_message_loop_proxy_.get(), throttler_manager()); | 70 io_message_loop_proxy_.get(), throttler_manager()); |
71 } | 71 } |
72 | 72 |
73 net::URLRequestThrottlerManager* throttler_manager() { | 73 net::URLRequestThrottlerManager* throttler_manager() { |
74 return &throttler_manager_; | 74 return &throttler_manager_; |
75 } | 75 } |
76 | 76 |
77 private: | 77 private: |
78 virtual ~TestCloudPrintURLFetcher() {} | 78 ~TestCloudPrintURLFetcher() override {} |
79 | 79 |
80 scoped_refptr<base::MessageLoopProxy> io_message_loop_proxy_; | 80 scoped_refptr<base::MessageLoopProxy> io_message_loop_proxy_; |
81 | 81 |
82 // We set this as the throttler manager for the | 82 // We set this as the throttler manager for the |
83 // TestURLRequestContext we create. | 83 // TestURLRequestContext we create. |
84 net::URLRequestThrottlerManager throttler_manager_; | 84 net::URLRequestThrottlerManager throttler_manager_; |
85 }; | 85 }; |
86 | 86 |
87 class CloudPrintURLFetcherTest : public testing::Test, | 87 class CloudPrintURLFetcherTest : public testing::Test, |
88 public CloudPrintURLFetcherDelegate { | 88 public CloudPrintURLFetcherDelegate { |
89 public: | 89 public: |
90 CloudPrintURLFetcherTest() : max_retries_(0), fetcher_(NULL) { } | 90 CloudPrintURLFetcherTest() : max_retries_(0), fetcher_(NULL) { } |
91 | 91 |
92 // Creates a URLFetcher, using the program's main thread to do IO. | 92 // Creates a URLFetcher, using the program's main thread to do IO. |
93 virtual void CreateFetcher(const GURL& url, int max_retries); | 93 virtual void CreateFetcher(const GURL& url, int max_retries); |
94 | 94 |
95 // CloudPrintURLFetcher::Delegate | 95 // CloudPrintURLFetcher::Delegate |
96 virtual CloudPrintURLFetcher::ResponseAction HandleRawResponse( | 96 CloudPrintURLFetcher::ResponseAction HandleRawResponse( |
97 const net::URLFetcher* source, | 97 const net::URLFetcher* source, |
98 const GURL& url, | 98 const GURL& url, |
99 const net::URLRequestStatus& status, | 99 const net::URLRequestStatus& status, |
100 int response_code, | 100 int response_code, |
101 const net::ResponseCookies& cookies, | 101 const net::ResponseCookies& cookies, |
102 const std::string& data) override; | 102 const std::string& data) override; |
103 | 103 |
104 virtual CloudPrintURLFetcher::ResponseAction OnRequestAuthError() override { | 104 CloudPrintURLFetcher::ResponseAction OnRequestAuthError() override { |
105 ADD_FAILURE(); | 105 ADD_FAILURE(); |
106 return CloudPrintURLFetcher::STOP_PROCESSING; | 106 return CloudPrintURLFetcher::STOP_PROCESSING; |
107 } | 107 } |
108 | 108 |
109 virtual std::string GetAuthHeader() override { | 109 std::string GetAuthHeader() override { return std::string(); } |
110 return std::string(); | |
111 } | |
112 | 110 |
113 scoped_refptr<base::MessageLoopProxy> io_message_loop_proxy() { | 111 scoped_refptr<base::MessageLoopProxy> io_message_loop_proxy() { |
114 return io_message_loop_proxy_; | 112 return io_message_loop_proxy_; |
115 } | 113 } |
116 | 114 |
117 protected: | 115 protected: |
118 virtual void SetUp() { | 116 virtual void SetUp() { |
119 testing::Test::SetUp(); | 117 testing::Test::SetUp(); |
120 | 118 |
121 io_message_loop_proxy_ = base::MessageLoopProxy::current(); | 119 io_message_loop_proxy_ = base::MessageLoopProxy::current(); |
(...skipping 17 matching lines...) Expand all Loading... |
139 int max_retries_; | 137 int max_retries_; |
140 Time start_time_; | 138 Time start_time_; |
141 scoped_refptr<TestCloudPrintURLFetcher> fetcher_; | 139 scoped_refptr<TestCloudPrintURLFetcher> fetcher_; |
142 }; | 140 }; |
143 | 141 |
144 class CloudPrintURLFetcherBasicTest : public CloudPrintURLFetcherTest { | 142 class CloudPrintURLFetcherBasicTest : public CloudPrintURLFetcherTest { |
145 public: | 143 public: |
146 CloudPrintURLFetcherBasicTest() | 144 CloudPrintURLFetcherBasicTest() |
147 : handle_raw_response_(false), handle_raw_data_(false) { } | 145 : handle_raw_response_(false), handle_raw_data_(false) { } |
148 // CloudPrintURLFetcher::Delegate | 146 // CloudPrintURLFetcher::Delegate |
149 virtual CloudPrintURLFetcher::ResponseAction HandleRawResponse( | 147 CloudPrintURLFetcher::ResponseAction HandleRawResponse( |
150 const net::URLFetcher* source, | 148 const net::URLFetcher* source, |
151 const GURL& url, | 149 const GURL& url, |
152 const net::URLRequestStatus& status, | 150 const net::URLRequestStatus& status, |
153 int response_code, | 151 int response_code, |
154 const net::ResponseCookies& cookies, | 152 const net::ResponseCookies& cookies, |
155 const std::string& data) override; | 153 const std::string& data) override; |
156 | 154 |
157 virtual CloudPrintURLFetcher::ResponseAction HandleRawData( | 155 CloudPrintURLFetcher::ResponseAction HandleRawData( |
158 const net::URLFetcher* source, | 156 const net::URLFetcher* source, |
159 const GURL& url, | 157 const GURL& url, |
160 const std::string& data) override; | 158 const std::string& data) override; |
161 | 159 |
162 virtual CloudPrintURLFetcher::ResponseAction HandleJSONData( | 160 CloudPrintURLFetcher::ResponseAction HandleJSONData( |
163 const net::URLFetcher* source, | 161 const net::URLFetcher* source, |
164 const GURL& url, | 162 const GURL& url, |
165 base::DictionaryValue* json_data, | 163 base::DictionaryValue* json_data, |
166 bool succeeded) override; | 164 bool succeeded) override; |
167 | 165 |
168 void SetHandleRawResponse(bool handle_raw_response) { | 166 void SetHandleRawResponse(bool handle_raw_response) { |
169 handle_raw_response_ = handle_raw_response; | 167 handle_raw_response_ = handle_raw_response; |
170 } | 168 } |
171 void SetHandleRawData(bool handle_raw_data) { | 169 void SetHandleRawData(bool handle_raw_data) { |
172 handle_raw_data_ = handle_raw_data; | 170 handle_raw_data_ = handle_raw_data; |
173 } | 171 } |
174 private: | 172 private: |
175 bool handle_raw_response_; | 173 bool handle_raw_response_; |
176 bool handle_raw_data_; | 174 bool handle_raw_data_; |
177 }; | 175 }; |
178 | 176 |
179 // Version of CloudPrintURLFetcherTest that tests overload protection. | 177 // Version of CloudPrintURLFetcherTest that tests overload protection. |
180 class CloudPrintURLFetcherOverloadTest : public CloudPrintURLFetcherTest { | 178 class CloudPrintURLFetcherOverloadTest : public CloudPrintURLFetcherTest { |
181 public: | 179 public: |
182 CloudPrintURLFetcherOverloadTest() : response_count_(0) { | 180 CloudPrintURLFetcherOverloadTest() : response_count_(0) { |
183 } | 181 } |
184 | 182 |
185 // CloudPrintURLFetcher::Delegate | 183 // CloudPrintURLFetcher::Delegate |
186 virtual CloudPrintURLFetcher::ResponseAction HandleRawData( | 184 CloudPrintURLFetcher::ResponseAction HandleRawData( |
187 const net::URLFetcher* source, | 185 const net::URLFetcher* source, |
188 const GURL& url, | 186 const GURL& url, |
189 const std::string& data) override; | 187 const std::string& data) override; |
190 | 188 |
191 private: | 189 private: |
192 int response_count_; | 190 int response_count_; |
193 }; | 191 }; |
194 | 192 |
195 // Version of CloudPrintURLFetcherTest that tests backoff protection. | 193 // Version of CloudPrintURLFetcherTest that tests backoff protection. |
196 class CloudPrintURLFetcherRetryBackoffTest : public CloudPrintURLFetcherTest { | 194 class CloudPrintURLFetcherRetryBackoffTest : public CloudPrintURLFetcherTest { |
197 public: | 195 public: |
198 CloudPrintURLFetcherRetryBackoffTest() : response_count_(0) { | 196 CloudPrintURLFetcherRetryBackoffTest() : response_count_(0) { |
199 } | 197 } |
200 | 198 |
201 // CloudPrintURLFetcher::Delegate | 199 // CloudPrintURLFetcher::Delegate |
202 virtual CloudPrintURLFetcher::ResponseAction HandleRawData( | 200 CloudPrintURLFetcher::ResponseAction HandleRawData( |
203 const net::URLFetcher* source, | 201 const net::URLFetcher* source, |
204 const GURL& url, | 202 const GURL& url, |
205 const std::string& data) override; | 203 const std::string& data) override; |
206 | 204 |
207 virtual void OnRequestGiveUp() override; | 205 void OnRequestGiveUp() override; |
208 | 206 |
209 private: | 207 private: |
210 int response_count_; | 208 int response_count_; |
211 }; | 209 }; |
212 | 210 |
213 | 211 |
214 void CloudPrintURLFetcherTest::CreateFetcher(const GURL& url, int max_retries) { | 212 void CloudPrintURLFetcherTest::CreateFetcher(const GURL& url, int max_retries) { |
215 fetcher_ = new TestCloudPrintURLFetcher(io_message_loop_proxy().get()); | 213 fetcher_ = new TestCloudPrintURLFetcher(io_message_loop_proxy().get()); |
216 | 214 |
217 // Registers an entry for test url. It only allows 3 requests to be sent | 215 // Registers an entry for test url. It only allows 3 requests to be sent |
(...skipping 152 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
370 base::FilePath(kDocRoot)); | 368 base::FilePath(kDocRoot)); |
371 ASSERT_TRUE(test_server.Start()); | 369 ASSERT_TRUE(test_server.Start()); |
372 | 370 |
373 GURL url(test_server.GetURL("defaultresponse")); | 371 GURL url(test_server.GetURL("defaultresponse")); |
374 CreateFetcher(url, 11); | 372 CreateFetcher(url, 11); |
375 | 373 |
376 base::MessageLoop::current()->Run(); | 374 base::MessageLoop::current()->Run(); |
377 } | 375 } |
378 | 376 |
379 } // namespace cloud_print | 377 } // namespace cloud_print |
OLD | NEW |