Move some download tests to content from chrome.

content/shell/shell_download_manager_delegate.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 "content/shell/shell_download_manager_delegate.h"
 
 #if defined(TOOLKIT_GTK)
 #include <gtk/gtk.h>
 #endif
 
@@ skipping 10 matching lines @@
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/download_manager.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_view.h"
 #include "net/base/net_util.h"
 
 namespace content {
 
 ShellDownloadManagerDelegate::ShellDownloadManagerDelegate()
-    : download_manager_(NULL) {
+    : download_manager_(NULL),
+      suppress_prompting_(false),
+      last_download_db_handle_(DownloadItem::kUninitializedHandle) {
   // Balanced in Shutdown();
   AddRef();
 }
 
 ShellDownloadManagerDelegate::~ShellDownloadManagerDelegate(){
 }
 
 
 void ShellDownloadManagerDelegate::SetDownloadManager(
     DownloadManager* download_manager) {
   download_manager_ = download_manager;
 }
 
 void ShellDownloadManagerDelegate::Shutdown() {
   Release();
 }
 
 bool ShellDownloadManagerDelegate::DetermineDownloadTarget(
     DownloadItem* download,
     const DownloadTargetCallback& callback) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  // This assignment needs to be here because even at the call to
+  // SetDownloadManager, the system is not fully initialized.
+  if (default_download_path_.empty()) {
+    default_download_path_ = download_manager_->GetBrowserContext()->GetPath().
+        Append(FILE_PATH_LITERAL("Downloads"));
+  }
+
   if (!download->GetForcedFilePath().empty()) {
     callback.Run(download->GetForcedFilePath(),
                  DownloadItem::TARGET_DISPOSITION_OVERWRITE,
                  DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
                  download->GetForcedFilePath());
     return true;
   }
 
   FilePath generated_name = net::GenerateFileName(
       download->GetURL(),
       download->GetContentDisposition(),
       download->GetReferrerCharset(),
       download->GetSuggestedFilename(),
       download->GetMimeType(),
       "download");
 
   BrowserThread::PostTask(
       BrowserThread::FILE,
       FROM_HERE,
       base::Bind(
           &ShellDownloadManagerDelegate::GenerateFilename,
-          this, download->GetId(), callback, generated_name));
-  return false;
+          this, download->GetId(), callback, generated_name,
+          default_download_path_));
+  return true;
 }
 
 void ShellDownloadManagerDelegate::GenerateFilename(
     int32 download_id,
     const DownloadTargetCallback& callback,
-    const FilePath& generated_name) {
-  FilePath suggested_path = download_manager_->GetBrowserContext()->GetPath().
-      Append(FILE_PATH_LITERAL("Downloads"));
-  if (!file_util::PathExists(suggested_path))
-    file_util::CreateDirectory(suggested_path);
+    const FilePath& generated_name,
+    const FilePath& suggested_directory) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+  if (!file_util::PathExists(suggested_directory))
+    file_util::CreateDirectory(suggested_directory);
 
-  suggested_path = suggested_path.Append(generated_name);
+  FilePath suggested_path(suggested_directory.Append(generated_name));
   BrowserThread::PostTask(
       BrowserThread::UI,
       FROM_HERE,
       base::Bind(
-          &ShellDownloadManagerDelegate::ChooseDownloadPath,
+          &ShellDownloadManagerDelegate::OnDownloadPathGenerated,
           this, download_id, callback, suggested_path));
 }
 
+void ShellDownloadManagerDelegate::OnDownloadPathGenerated(
+    int32 download_id,
+    const DownloadTargetCallback& callback,
+    const FilePath& suggested_path) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  if (suppress_prompting_) {
+    // Testing exit.
+    callback.Run(suggested_path, DownloadItem::TARGET_DISPOSITION_OVERWRITE,
+                 DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, suggested_path);
+    return;
+  }
+
+  ChooseDownloadPath(download_id, callback, suggested_path);
+}
+
 void ShellDownloadManagerDelegate::ChooseDownloadPath(
     int32 download_id,
     const DownloadTargetCallback& callback,
     const FilePath& suggested_path) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   DownloadItem* item =
       download_manager_->GetActiveDownloadItem(download_id);
   if (!item)
     return;
 
   FilePath result;
 #if defined(OS_WIN) && !defined(USE_AURA)
   std::wstring file_part = FilePath(suggested_path).BaseName().value();
   wchar_t file_name[MAX_PATH];
   base::wcslcpy(file_name, file_part.c_str(), arraysize(file_name));
@@ skipping 38 matching lines @@
   }
   gtk_widget_destroy(dialog);
 #else
   NOTIMPLEMENTED();
 #endif
 
   callback.Run(result, DownloadItem::TARGET_DISPOSITION_PROMPT,
                DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, result);
 }
 
+void ShellDownloadManagerDelegate::AddItemToPersistentStore(
+    DownloadItem* item) {
+  download_manager_->OnItemAddedToPersistentStore(
+      item->GetId(), --last_download_db_handle_);
+}
+
+void ShellDownloadManagerDelegate::SetDownloadBehaviorForTesting(
+    const FilePath& default_download_path) {
+  default_download_path_ = default_download_path;
+  suppress_prompting_ = true;
+}
+
 }  // namespace content