Move download filename determintion into a separate class.

chrome/browser/download/download_test_file_activity_observer.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 "chrome/browser/download/download_test_file_activity_observer.h"
 
 #include "base/bind.h"
 #include "base/message_loop.h"
 #include "chrome/browser/download/chrome_download_manager_delegate.h"
+#include "chrome/browser/download/download_file_picker.h"
 #include "chrome/browser/download/download_service.h"
 #include "chrome/browser/download/download_service_factory.h"
 #include "chrome/browser/profiles/profile.h"
 
 namespace content {
 class DownloadItem;
 }
 
-// Test ChromeDownloadManagerDelegate that controls whether how file chooser
-// dialogs are handled, and how files are opend.
-// By default, file chooser dialogs are disabled.
-class DownloadTestFileActivityObserver::MockDownloadManagerDelegate
-    : public ChromeDownloadManagerDelegate {
+class DownloadTestFileActivityObserver::TestDownloadFilePickerFactory
+    : public DownloadFilePickerFactory {
  public:
-  explicit MockDownloadManagerDelegate(Profile* profile)
-      : ChromeDownloadManagerDelegate(profile),
-        file_chooser_enabled_(false),
-        file_chooser_displayed_(false) {}
+  TestDownloadFilePickerFactory()
+      : file_chooser_enabled_(false),
+        file_chooser_displayed_(false),
+        ALLOW_THIS_IN_INITIALIZER_LIST(weak_ptr_factory_(this)) {
+  }
+
+  // DownloadFilePicker
+  virtual void Create(
+      content::DownloadItem* download,
+      const base::FilePath& suggested_path,
+      const DownloadFilePicker::FileSelectedCallback& callback) OVERRIDE {
+    base::FilePath target_path =
+        file_chooser_enabled_ ? suggested_path : base::FilePath();
+    file_chooser_displayed_ = true;
+    MessageLoop::current()->PostTask(
+        FROM_HERE, base::Bind(callback, target_path, target_path));
+  }
 
   void EnableFileChooser(bool enable) {
     file_chooser_enabled_ = enable;
   }
 
   bool TestAndResetDidShowFileChooser() {
     bool did_show = file_chooser_displayed_;
     file_chooser_displayed_ = false;
     return did_show;
   }
 
- protected:
-
-  virtual void ChooseDownloadPath(content::DownloadItem* item,
-                                  const base::FilePath& suggested_path,
-                                  const FileSelectedCallback&
-                                      callback) OVERRIDE {
-    file_chooser_displayed_ = true;
-    MessageLoop::current()->PostTask(
-        FROM_HERE, base::Bind(callback, (file_chooser_enabled_ ? suggested_path
-                                         : base::FilePath())));
+  base::WeakPtr<TestDownloadFilePickerFactory> GetWeakPtr() {
+    return weak_ptr_factory_.GetWeakPtr();
   }
 
-  virtual void OpenDownload(content::DownloadItem* item) OVERRIDE {}
-
  private:
-  virtual ~MockDownloadManagerDelegate() {}
-
   bool file_chooser_enabled_;
   bool file_chooser_displayed_;
+  base::WeakPtrFactory<TestDownloadFilePickerFactory> weak_ptr_factory_;
 };
 
 DownloadTestFileActivityObserver::DownloadTestFileActivityObserver(
     Profile* profile) {
-  test_delegate_ = new MockDownloadManagerDelegate(profile);
-  DownloadServiceFactory::GetForProfile(profile)->
-      SetDownloadManagerDelegateForTesting(test_delegate_.get());
+  scoped_ptr<TestDownloadFilePickerFactory> file_picker_factory(
+      new TestDownloadFilePickerFactory);
+  test_file_picker_factory_ = file_picker_factory->GetWeakPtr();
+
+  DownloadService* service = DownloadServiceFactory::GetForProfile(profile);
+  service->GetDownloadManagerDelegate()->SetFilePickerFactoryForTesting(
+      file_picker_factory.PassAs<DownloadFilePickerFactory>());
 }
 
 DownloadTestFileActivityObserver::~DownloadTestFileActivityObserver() {
 }
 
 void DownloadTestFileActivityObserver::EnableFileChooser(bool enable) {
-  test_delegate_->EnableFileChooser(enable);
+  if (test_file_picker_factory_.get())
+    test_file_picker_factory_->EnableFileChooser(enable);
 }
 
 bool DownloadTestFileActivityObserver::TestAndResetDidShowFileChooser() {
-  return test_delegate_->TestAndResetDidShowFileChooser();
+  return test_file_picker_factory_.get() &&
+      test_file_picker_factory_->TestAndResetDidShowFileChooser();
 }