chrome/browser/download/download_test_file_activity_observer.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/download/download_test_file_activity_observer.h"
   6
   7 #include "base/bind.h"
   8 #include "base/location.h"
   9 #include "base/single_thread_task_runner.h"
  10 #include "base/thread_task_runner_handle.h"
  11 #include "chrome/browser/download/chrome_download_manager_delegate.h"
  12 #include "chrome/browser/download/download_service.h"
  13 #include "chrome/browser/download/download_service_factory.h"
  14 #include "chrome/browser/profiles/profile.h"
  15
  16 namespace content {
  17 class DownloadItem;
  18 }
  19
  20 // Test ChromeDownloadManagerDelegate that controls whether how file chooser
  21 // dialogs are handled, and how files are opend.
  22 // By default, file chooser dialogs are disabled.
  23 class DownloadTestFileActivityObserver::MockDownloadManagerDelegate
  24     : public ChromeDownloadManagerDelegate {
  25  public:
  26   explicit MockDownloadManagerDelegate(Profile* profile)
  27       : ChromeDownloadManagerDelegate(profile),
  28         file_chooser_enabled_(false),
  29         file_chooser_displayed_(false),
  30         weak_ptr_factory_(this) {
  31     if (!profile->IsOffTheRecord())
  32       GetDownloadIdReceiverCallback().Run(
  33           content::DownloadItem::kInvalidId + 1);
  34   }
  35
  36   ~MockDownloadManagerDelegate() override {}
  37
  38   void EnableFileChooser(bool enable) {
  39     file_chooser_enabled_ = enable;
  40   }
  41
  42   bool TestAndResetDidShowFileChooser() {
  43     bool did_show = file_chooser_displayed_;
  44     file_chooser_displayed_ = false;
  45     return did_show;
  46   }
  47
  48   base::WeakPtr<MockDownloadManagerDelegate> GetWeakPtr() {
  49     return weak_ptr_factory_.GetWeakPtr();
  50   }
  51
  52  protected:
  53   void PromptUserForDownloadPath(
  54       content::DownloadItem* item,
  55       const base::FilePath& suggested_path,
  56       const FileSelectedCallback& callback) override {
  57     file_chooser_displayed_ = true;
  58     base::ThreadTaskRunnerHandle::Get()->PostTask(
  59         FROM_HERE,
  60         base::Bind(callback, (file_chooser_enabled_ ? suggested_path
  61                                                     : base::FilePath())));
  62   }
  63
  64   void OpenDownload(content::DownloadItem* item) override {}
  65
  66  private:
  67   bool file_chooser_enabled_;
  68   bool file_chooser_displayed_;
  69   base::WeakPtrFactory<MockDownloadManagerDelegate> weak_ptr_factory_;
  70 };
  71
  72 DownloadTestFileActivityObserver::DownloadTestFileActivityObserver(
  73     Profile* profile) {
  74   scoped_ptr<MockDownloadManagerDelegate> mock_delegate(
  75       new MockDownloadManagerDelegate(profile));
  76   test_delegate_ = mock_delegate->GetWeakPtr();
  77   DownloadServiceFactory::GetForBrowserContext(profile)
  78       ->SetDownloadManagerDelegateForTesting(mock_delegate.Pass());
  79 }
  80
  81 DownloadTestFileActivityObserver::~DownloadTestFileActivityObserver() {
  82 }
  83
  84 void DownloadTestFileActivityObserver::EnableFileChooser(bool enable) {
  85   if (test_delegate_.get())
  86     test_delegate_->EnableFileChooser(enable);
  87 }
  88
  89 bool DownloadTestFileActivityObserver::TestAndResetDidShowFileChooser() {
  90   return test_delegate_.get() &&
  91       test_delegate_->TestAndResetDidShowFileChooser();
  92 }