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