OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/memory/scoped_ptr.h" | 5 #include "base/memory/scoped_ptr.h" |
6 #include "base/message_loop.h" | 6 #include "base/message_loop.h" |
7 #include "content/browser/browser_thread.h" | 7 #include "content/browser/browser_thread.h" |
8 #include "content/browser/mock_resource_context.h" | 8 #include "content/browser/mock_resource_context.h" |
9 #include "content/browser/renderer_host/dummy_resource_handler.h" | 9 #include "content/browser/renderer_host/dummy_resource_handler.h" |
10 #include "content/browser/renderer_host/global_request_id.h" | 10 #include "content/browser/renderer_host/global_request_id.h" |
(...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
71 } | 71 } |
72 | 72 |
73 virtual void Initialize(ResourceQueue* resource_queue) { | 73 virtual void Initialize(ResourceQueue* resource_queue) { |
74 resource_queue_ = resource_queue; | 74 resource_queue_ = resource_queue; |
75 } | 75 } |
76 | 76 |
77 virtual bool ShouldDelayRequest( | 77 virtual bool ShouldDelayRequest( |
78 net::URLRequest* request, | 78 net::URLRequest* request, |
79 const ResourceDispatcherHostRequestInfo& request_info, | 79 const ResourceDispatcherHostRequestInfo& request_info, |
80 const GlobalRequestID& request_id) { | 80 const GlobalRequestID& request_id) { |
| 81 delayed_requests_.push_back(request_id); |
81 return true; | 82 return true; |
82 } | 83 } |
83 | 84 |
84 virtual void WillShutdownResourceQueue() { | 85 virtual void WillShutdownResourceQueue() { |
85 resource_queue_ = NULL; | 86 resource_queue_ = NULL; |
86 } | 87 } |
87 | 88 |
88 void StartDelayedRequests() { | 89 void StartDelayedRequests() { |
89 if (resource_queue_) | 90 if (!resource_queue_) |
90 resource_queue_->StartDelayedRequests(this); | 91 return; |
| 92 |
| 93 for (RequestList::iterator i = delayed_requests_.begin(); |
| 94 i != delayed_requests_.end(); ++i) { |
| 95 resource_queue_->StartDelayedRequest(this, *i); |
| 96 } |
91 } | 97 } |
92 | 98 |
93 private: | 99 private: |
94 typedef std::vector<GlobalRequestID> RequestList; | 100 typedef std::vector<GlobalRequestID> RequestList; |
95 | 101 |
96 ResourceQueue* resource_queue_; | 102 ResourceQueue* resource_queue_; |
97 | 103 |
| 104 RequestList delayed_requests_; |
| 105 |
98 DISALLOW_COPY_AND_ASSIGN(AlwaysDelayingDelegate); | 106 DISALLOW_COPY_AND_ASSIGN(AlwaysDelayingDelegate); |
99 }; | 107 }; |
100 | 108 |
101 class ResourceQueueTest : public testing::Test, | 109 class ResourceQueueTest : public testing::Test, |
102 public net::URLRequest::Delegate { | 110 public net::URLRequest::Delegate { |
103 public: | 111 public: |
104 ResourceQueueTest() | 112 ResourceQueueTest() |
105 : response_started_count_(0), | 113 : response_started_count_(0), |
106 message_loop_(MessageLoop::TYPE_IO), | 114 message_loop_(MessageLoop::TYPE_IO), |
107 ui_thread_(BrowserThread::UI, &message_loop_), | 115 ui_thread_(BrowserThread::UI, &message_loop_), |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
202 queue.AddRequest(&request, *request_info.get()); | 210 queue.AddRequest(&request, *request_info.get()); |
203 MessageLoop::current()->RunAllPending(); | 211 MessageLoop::current()->RunAllPending(); |
204 EXPECT_EQ(0, response_started_count_); | 212 EXPECT_EQ(0, response_started_count_); |
205 always_delaying_delegate.StartDelayedRequests(); | 213 always_delaying_delegate.StartDelayedRequests(); |
206 MessageLoop::current()->RunAllPending(); | 214 MessageLoop::current()->RunAllPending(); |
207 EXPECT_EQ(1, response_started_count_); | 215 EXPECT_EQ(1, response_started_count_); |
208 | 216 |
209 queue.Shutdown(); | 217 queue.Shutdown(); |
210 } | 218 } |
211 | 219 |
212 TEST_F(ResourceQueueTest, TwoDelayingDelegates) { | |
213 ResourceQueue queue; | |
214 | |
215 AlwaysDelayingDelegate always_delaying_delegate1; | |
216 AlwaysDelayingDelegate always_delaying_delegate2; | |
217 InitializeQueue( | |
218 &queue, &always_delaying_delegate1, &always_delaying_delegate2); | |
219 | |
220 net::URLRequest request(GURL(kTestUrl), this); | |
221 scoped_ptr<ResourceDispatcherHostRequestInfo> request_info(GetRequestInfo(0)); | |
222 EXPECT_EQ(0, response_started_count_); | |
223 queue.AddRequest(&request, *request_info.get()); | |
224 MessageLoop::current()->RunAllPending(); | |
225 EXPECT_EQ(0, response_started_count_); | |
226 | |
227 always_delaying_delegate1.StartDelayedRequests(); | |
228 MessageLoop::current()->RunAllPending(); | |
229 EXPECT_EQ(0, response_started_count_); | |
230 | |
231 always_delaying_delegate2.StartDelayedRequests(); | |
232 MessageLoop::current()->RunAllPending(); | |
233 EXPECT_EQ(1, response_started_count_); | |
234 | |
235 queue.Shutdown(); | |
236 } | |
237 | |
238 TEST_F(ResourceQueueTest, RemoveRequest) { | 220 TEST_F(ResourceQueueTest, RemoveRequest) { |
239 ResourceQueue queue; | 221 ResourceQueue queue; |
240 | 222 |
241 AlwaysDelayingDelegate delegate; | 223 AlwaysDelayingDelegate delegate; |
242 InitializeQueue(&queue, &delegate); | 224 InitializeQueue(&queue, &delegate); |
243 | 225 |
244 net::URLRequest request(GURL(kTestUrl), this); | 226 net::URLRequest request(GURL(kTestUrl), this); |
245 scoped_ptr<ResourceDispatcherHostRequestInfo> request_info(GetRequestInfo(0)); | 227 scoped_ptr<ResourceDispatcherHostRequestInfo> request_info(GetRequestInfo(0)); |
246 GlobalRequestID request_id(request_info->child_id(), | 228 GlobalRequestID request_id(request_info->child_id(), |
247 request_info->request_id()); | 229 request_info->request_id()); |
248 EXPECT_EQ(0, response_started_count_); | 230 EXPECT_EQ(0, response_started_count_); |
249 queue.AddRequest(&request, *request_info.get()); | 231 queue.AddRequest(&request, *request_info.get()); |
250 MessageLoop::current()->RunAllPending(); | 232 MessageLoop::current()->RunAllPending(); |
251 EXPECT_EQ(0, response_started_count_); | 233 EXPECT_EQ(0, response_started_count_); |
252 queue.RemoveRequest(request_id); | 234 queue.RemoveRequest(request_id); |
253 delegate.StartDelayedRequests(); | 235 delegate.StartDelayedRequests(); |
254 MessageLoop::current()->RunAllPending(); | 236 MessageLoop::current()->RunAllPending(); |
255 EXPECT_EQ(0, response_started_count_); | 237 EXPECT_EQ(0, response_started_count_); |
256 | 238 |
257 queue.Shutdown(); | 239 queue.Shutdown(); |
258 | 240 |
259 MessageLoop::current()->RunAllPending(); | 241 MessageLoop::current()->RunAllPending(); |
260 EXPECT_EQ(0, response_started_count_); | 242 EXPECT_EQ(0, response_started_count_); |
261 } | 243 } |
262 | 244 |
263 } // namespace | 245 } // namespace |
OLD | NEW |