DownloadManagerDelegate::ShouldCompleteDownload(callback)

content/browser/download/download_manager_impl_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <set>
 #include <string>
 
 #include "base/bind.h"
 #include "base/file_util.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 159 matching lines @@
   void SetFileSelectionExpectation(const FilePath& suggested_path,
                                    const FilePath& response) {
     expected_suggested_path_ = suggested_path;
     file_selection_response_ = response;
   }
 
   void SetMarkContentsDangerous(bool dangerous) {
     mark_content_dangerous_ = dangerous;
   }
 
-  virtual bool ShouldCompleteDownload(DownloadItem* item) {
+  virtual bool ShouldCompleteDownload(
+      DownloadItem* item,
+      const base::Closure& complete_callback) {
     if (mark_content_dangerous_) {
+      CHECK(!complete_callback.is_null());
       BrowserThread::PostTask(
           BrowserThread::UI, FROM_HERE,
           base::Bind(&TestDownloadManagerDelegate::MarkContentDangerous,
-                     base::Unretained(this), item->GetId()));
+                     base::Unretained(this), item->GetId(), complete_callback));
       mark_content_dangerous_ = false;
       return false;
     } else {
       return true;
     }
   }
 
  private:
-  void MarkContentDangerous(int32 download_id) {
+  void MarkContentDangerous(
+      int32 download_id,
+      const base::Closure& complete_callback) {
     DownloadItem* item = download_manager_->GetActiveDownloadItem(download_id);
     if (!item)
       return;
     item->SetDangerType(content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT);
-    item->MaybeCompleteDownload();
+    complete_callback.Run();
   }
 
   FilePath expected_suggested_path_;
   FilePath file_selection_response_;
   bool mark_content_dangerous_;
   bool prompt_user_for_save_location_;
   DownloadManager* download_manager_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestDownloadManagerDelegate);
 };
 
 } // namespace
 
 class DownloadManagerTest : public testing::Test {
  public:
   static const char* kTestData;
   static const size_t kTestDataLen;
 
   DownloadManagerTest()
@@ skipping 1093 matching lines @@
   EXPECT_FALSE(observer->hit_state(DownloadItem::INTERRUPTED));
   EXPECT_TRUE(observer->hit_state(DownloadItem::COMPLETE));
   EXPECT_FALSE(observer->hit_state(DownloadItem::REMOVING));
   EXPECT_TRUE(observer->was_updated());
   EXPECT_FALSE(observer->was_opened());
   EXPECT_TRUE(download->GetFileExternallyRemoved());
   EXPECT_EQ(DownloadItem::COMPLETE, download->GetState());
 
   EXPECT_FALSE(file_util::PathExists(new_path));
 }