Index: content/public/test/download_test_observer.cc |
diff --git a/content/public/test/download_test_observer.cc b/content/public/test/download_test_observer.cc |
index ee5961fe531356e821d75a9f07fedb124803f8b3..cc26ff02a45a7ef0ec013ccc80bdf72eed4a341a 100644 |
--- a/content/public/test/download_test_observer.cc |
+++ b/content/public/test/download_test_observer.cc |
@@ -221,7 +221,7 @@ void DownloadTestObserver::AcceptDangerousDownload(int32 download_id) { |
if (!download_manager_) |
return; |
DownloadItem* download = download_manager_->GetDownload(download_id); |
- if (download && (download->GetState() == DownloadItem::IN_PROGRESS)) |
+ if (download && !download->IsDone()) |
download->ValidateDangerousDownload(); |
} |
@@ -231,9 +231,8 @@ void DownloadTestObserver::DenyDangerousDownload(int32 download_id) { |
if (!download_manager_) |
return; |
DownloadItem* download = download_manager_->GetDownload(download_id); |
- if (download && (download->GetState() == DownloadItem::IN_PROGRESS)) { |
+ if (download && !download->IsDone()) |
download->Remove(); |
- } |
} |
DownloadTestObserverTerminal::DownloadTestObserverTerminal( |