| 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 "content/public/test/download_test_observer.h" | 5 #include "content/public/test/download_test_observer.h" |
| 6 | 6 |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| (...skipping 120 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 131 DownloadInFinalState(download); | 131 DownloadInFinalState(download); |
| 132 } | 132 } |
| 133 | 133 |
| 134 void DownloadTestObserver::ModelChanged(DownloadManager* manager) { | 134 void DownloadTestObserver::ModelChanged(DownloadManager* manager) { |
| 135 DCHECK_EQ(manager, download_manager_); | 135 DCHECK_EQ(manager, download_manager_); |
| 136 | 136 |
| 137 // Regenerate DownloadItem observers. If there are any download items | 137 // Regenerate DownloadItem observers. If there are any download items |
| 138 // in our final state, note them in |finished_downloads_| | 138 // in our final state, note them in |finished_downloads_| |
| 139 // (done by |OnDownloadUpdated()|). | 139 // (done by |OnDownloadUpdated()|). |
| 140 std::vector<DownloadItem*> downloads; | 140 std::vector<DownloadItem*> downloads; |
| 141 download_manager_->GetAllDownloads(FilePath(), &downloads); | 141 download_manager_->GetAllDownloads(&downloads); |
| 142 // As a test class, we're generally interested in whatever's there, | |
| 143 // so we include temporary downloads. | |
| 144 download_manager_->GetTemporaryDownloads(FilePath(), &downloads); | |
| 145 | 142 |
| 146 for (std::vector<DownloadItem*>::iterator it = downloads.begin(); | 143 for (std::vector<DownloadItem*>::iterator it = downloads.begin(); |
| 147 it != downloads.end(); ++it) { | 144 it != downloads.end(); ++it) { |
| 148 OnDownloadUpdated(*it); // Safe to call multiple times; checks state. | 145 OnDownloadUpdated(*it); // Safe to call multiple times; checks state. |
| 149 | 146 |
| 150 DownloadSet::const_iterator finished_it(finished_downloads_.find(*it)); | 147 DownloadSet::const_iterator finished_it(finished_downloads_.find(*it)); |
| 151 DownloadSet::iterator observed_it(downloads_observed_.find(*it)); | 148 DownloadSet::iterator observed_it(downloads_observed_.find(*it)); |
| 152 | 149 |
| 153 // If it isn't finished and we're aren't observing it, start. | 150 // If it isn't finished and we're aren't observing it, start. |
| 154 if (finished_it == finished_downloads_.end() && | 151 if (finished_it == finished_downloads_.end() && |
| (...skipping 131 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 286 | 283 |
| 287 // If we're waiting for that flush point, check the number | 284 // If we're waiting for that flush point, check the number |
| 288 // of downloads in the IN_PROGRESS state and take appropriate | 285 // of downloads in the IN_PROGRESS state and take appropriate |
| 289 // action. If requested, also observes all downloads while iterating. | 286 // action. If requested, also observes all downloads while iterating. |
| 290 void DownloadTestFlushObserver::CheckDownloadsInProgress( | 287 void DownloadTestFlushObserver::CheckDownloadsInProgress( |
| 291 bool observe_downloads) { | 288 bool observe_downloads) { |
| 292 if (waiting_for_zero_inprogress_) { | 289 if (waiting_for_zero_inprogress_) { |
| 293 int count = 0; | 290 int count = 0; |
| 294 | 291 |
| 295 std::vector<DownloadItem*> downloads; | 292 std::vector<DownloadItem*> downloads; |
| 296 download_manager_->SearchDownloads(string16(), &downloads); | 293 download_manager_->GetAllDownloads(&downloads); |
| 297 for (std::vector<DownloadItem*>::iterator it = downloads.begin(); | 294 for (std::vector<DownloadItem*>::iterator it = downloads.begin(); |
| 298 it != downloads.end(); ++it) { | 295 it != downloads.end(); ++it) { |
| 299 if ((*it)->GetState() == DownloadItem::IN_PROGRESS) | 296 if ((*it)->GetState() == DownloadItem::IN_PROGRESS) |
| 300 count++; | 297 count++; |
| 301 if (observe_downloads) { | 298 if (observe_downloads) { |
| 302 if (downloads_observed_.find(*it) == downloads_observed_.end()) { | 299 if (downloads_observed_.find(*it) == downloads_observed_.end()) { |
| 303 (*it)->AddObserver(this); | 300 (*it)->AddObserver(this); |
| 304 downloads_observed_.insert(*it); | 301 downloads_observed_.insert(*it); |
| 305 } | 302 } |
| 306 // Download items are forever, and we don't want to make | 303 // Download items are forever, and we don't want to make |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 379 MessageLoopForUI::current()->Quit(); | 376 MessageLoopForUI::current()->Quit(); |
| 380 } | 377 } |
| 381 | 378 |
| 382 const DownloadUrlParameters::OnStartedCallback | 379 const DownloadUrlParameters::OnStartedCallback |
| 383 DownloadTestItemCreationObserver::callback() { | 380 DownloadTestItemCreationObserver::callback() { |
| 384 return base::Bind( | 381 return base::Bind( |
| 385 &DownloadTestItemCreationObserver::DownloadItemCreationCallback, this); | 382 &DownloadTestItemCreationObserver::DownloadItemCreationCallback, this); |
| 386 } | 383 } |
| 387 | 384 |
| 388 } // namespace content | 385 } // namespace content |
| OLD | NEW |