Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_BROWSER_DOWNLOAD_DOWNLOAD_TEST_OBSERVER_H_ | 5 #ifndef CONTENT_TEST_DOWNLOAD_TEST_OBSERVER_H_ |
| 6 #define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_TEST_OBSERVER_H_ | 6 #define CONTENT_TEST_DOWNLOAD_TEST_OBSERVER_H_ |
| 7 | 7 |
| 8 #include <set> | 8 #include <set> |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
| 12 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
| 13 #include "content/public/browser/download_item.h" | 13 #include "content/public/browser/download_item.h" |
| 14 #include "content/public/browser/download_manager.h" | 14 #include "content/public/browser/download_manager.h" |
| 15 #include "content/public/browser/download_url_parameters.h" | 15 #include "content/public/browser/download_url_parameters.h" |
| 16 #include "net/base/net_errors.h" | 16 #include "net/base/net_errors.h" |
| 17 | 17 |
| 18 namespace internal { | 18 namespace content { |
| 19 class MockFileChooserDownloadManagerDelegate; | |
| 20 } | |
| 21 | |
| 22 class Profile; | |
| 23 | 19 |
| 24 // Detects changes to the downloads after construction. | 20 // Detects changes to the downloads after construction. |
| 25 // Finishes when one of the following happens: | 21 // Finishes when one of the following happens: |
| 26 // - A specified number of downloads change to a terminal state (defined | 22 // - A specified number of downloads change to a terminal state (defined |
| 27 // in derived classes). | 23 // in derived classes). |
| 28 // - Specific events, such as a select file dialog. | 24 // - Specific events, such as a select file dialog. |
| 29 // Callers may either probe for the finished state, or wait on it. | 25 // Callers may either probe for the finished state, or wait on it. |
| 30 // | 26 // |
| 31 // TODO(rdsmith): Detect manager going down, remove pointer to | 27 // TODO(rdsmith): Detect manager going down, remove pointer to |
| 32 // DownloadManager, transition to finished. (For right now we | 28 // DownloadManager, transition to finished. (For right now we |
| 33 // just use a scoped_refptr<> to keep it around, but that may cause | 29 // just use a scoped_refptr<> to keep it around, but that may cause |
| 34 // timeouts on waiting if a DownloadManager::Shutdown() occurs which | 30 // timeouts on waiting if a DownloadManager::Shutdown() occurs which |
| 35 // cancels our in-progress downloads.) | 31 // cancels our in-progress downloads.) |
| 36 class DownloadTestObserver : public content::DownloadManager::Observer, | 32 class DownloadTestObserver : public content::DownloadManager::Observer, |
|
jam
2012/08/13 16:00:21
nit: get rid of content:: everywhere
Randy Smith (Not in Mondays)
2012/08/13 18:57:49
Done.
| |
| 37 public content::DownloadItem::Observer { | 33 public content::DownloadItem::Observer { |
| 38 public: | 34 public: |
| 39 // Action an observer should take if a dangerous download is encountered. | 35 // Action an observer should take if a dangerous download is encountered. |
| 40 enum DangerousDownloadAction { | 36 enum DangerousDownloadAction { |
| 41 ON_DANGEROUS_DOWNLOAD_ACCEPT, // Accept the download | 37 ON_DANGEROUS_DOWNLOAD_ACCEPT, // Accept the download |
| 42 ON_DANGEROUS_DOWNLOAD_DENY, // Deny the download | 38 ON_DANGEROUS_DOWNLOAD_DENY, // Deny the download |
| 43 ON_DANGEROUS_DOWNLOAD_FAIL // Fail if a dangerous download is seen | 39 ON_DANGEROUS_DOWNLOAD_FAIL // Fail if a dangerous download is seen |
| 44 }; | 40 }; |
| 45 | 41 |
| 46 // Create an object that will be considered finished when |wait_count| | 42 // Create an object that will be considered finished when |wait_count| |
| (...skipping 212 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 259 | 255 |
| 260 // Count of callbacks. | 256 // Count of callbacks. |
| 261 size_t called_back_count_; | 257 size_t called_back_count_; |
| 262 | 258 |
| 263 // We are in the message loop. | 259 // We are in the message loop. |
| 264 bool waiting_; | 260 bool waiting_; |
| 265 | 261 |
| 266 DISALLOW_COPY_AND_ASSIGN(DownloadTestItemCreationObserver); | 262 DISALLOW_COPY_AND_ASSIGN(DownloadTestItemCreationObserver); |
| 267 }; | 263 }; |
| 268 | 264 |
| 269 // Observes and overrides file chooser activity for a profile. By default, once | 265 } // namespace content` |
| 270 // attached to a profile, this class overrides the default file chooser by | |
| 271 // replacing the ChromeDownloadManagerDelegate associated with |profile|. | |
| 272 // NOTE: Again, this overrides the ChromeDownloadManagerDelegate for |profile|. | |
| 273 class DownloadTestFileChooserObserver { | |
| 274 public: | |
| 275 // Attaches to |profile|. By default file chooser dialogs will be disabled | |
| 276 // once attached. Call EnableFileChooser() to re-enable. | |
| 277 explicit DownloadTestFileChooserObserver(Profile* profile); | |
| 278 ~DownloadTestFileChooserObserver(); | |
| 279 | 266 |
| 280 // Sets whether the file chooser dialog is enabled. If |enable| is false, any | 267 #endif // CONTENT_TEST_DOWNLOAD_TEST_OBSERVER_H_ |
| 281 // attempt to display a file chooser dialog will cause the download to be | |
| 282 // canceled. Otherwise, attempting to display a file chooser dialog will | |
| 283 // result in the download continuing with the suggested path. | |
| 284 void EnableFileChooser(bool enable); | |
| 285 | |
| 286 // Returns true if a file chooser dialog was displayed since the last time | |
| 287 // this method was called. | |
| 288 bool TestAndResetDidShowFileChooser(); | |
| 289 | |
| 290 private: | |
| 291 scoped_refptr<internal::MockFileChooserDownloadManagerDelegate> | |
| 292 test_delegate_; | |
| 293 }; | |
| 294 | |
| 295 #endif // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_TEST_OBSERVER_H_ | |
| OLD | NEW |