Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(322)

Side by Side Diff: content/test/download_test_observer.cc

Issue 10855116: Move DownloadTestObserver and friends down into content. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 8 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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 #include "content/public/test/download_test_observer.h"
6
5 #include <vector> 7 #include <vector>
6 8
7 #include "base/bind.h" 9 #include "base/bind.h"
8 #include "base/logging.h" 10 #include "base/logging.h"
9 #include "base/message_loop.h" 11 #include "base/message_loop.h"
10 #include "base/stl_util.h" 12 #include "base/stl_util.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/download/download_test_observer.h"
15 #include "chrome/test/base/ui_test_utils.h"
16 #include "content/public/browser/browser_thread.h" 13 #include "content/public/browser/browser_thread.h"
17 #include "content/public/browser/download_url_parameters.h" 14 #include "content/public/browser/download_url_parameters.h"
15 #include "content/public/test/test_utils.h"
16 #include "testing/gtest/include/gtest/gtest.h"
18 17
19 using content::BrowserThread; 18 using content::BrowserThread;
20 using content::DownloadItem; 19 using content::DownloadItem;
21 using content::DownloadManager; 20 using content::DownloadManager;
22 21
23 namespace { 22 namespace {
24 23
25 // These functions take scoped_refptr's to DownloadManager because they 24 // These functions take scoped_refptr's to DownloadManager because they
26 // are posted to message queues, and hence may execute arbitrarily after 25 // are posted to message queues, and hence may execute arbitrarily after
27 // their actual posting. Once posted, there is no connection between 26 // their actual posting. Once posted, there is no connection between
(...skipping 12 matching lines...) Expand all
40 void DenyDangerousDownload(scoped_refptr<DownloadManager> download_manager, 39 void DenyDangerousDownload(scoped_refptr<DownloadManager> download_manager,
41 int32 download_id) { 40 int32 download_id) {
42 DownloadItem* download = download_manager->GetDownloadItem(download_id); 41 DownloadItem* download = download_manager->GetDownloadItem(download_id);
43 ASSERT_TRUE(download->IsPartialDownload()); 42 ASSERT_TRUE(download->IsPartialDownload());
44 download->Cancel(true); 43 download->Cancel(true);
45 download->Delete(DownloadItem::DELETE_DUE_TO_USER_DISCARD); 44 download->Delete(DownloadItem::DELETE_DUE_TO_USER_DISCARD);
46 } 45 }
47 46
48 } // namespace 47 } // namespace
49 48
49 namespace content {
50
50 DownloadTestObserver::DownloadTestObserver( 51 DownloadTestObserver::DownloadTestObserver(
51 DownloadManager* download_manager, 52 DownloadManager* download_manager,
52 size_t wait_count, 53 size_t wait_count,
53 DangerousDownloadAction dangerous_download_action) 54 DangerousDownloadAction dangerous_download_action)
54 : download_manager_(download_manager), 55 : download_manager_(download_manager),
55 wait_count_(wait_count), 56 wait_count_(wait_count),
56 finished_downloads_at_construction_(0), 57 finished_downloads_at_construction_(0),
57 waiting_(false), 58 waiting_(false),
58 dangerous_download_action_(dangerous_download_action) { 59 dangerous_download_action_(dangerous_download_action) {
59 } 60 }
60 61
61 DownloadTestObserver::~DownloadTestObserver() { 62 DownloadTestObserver::~DownloadTestObserver() {
62 for (DownloadSet::iterator it = downloads_observed_.begin(); 63 for (DownloadSet::iterator it = downloads_observed_.begin();
63 it != downloads_observed_.end(); ++it) 64 it != downloads_observed_.end(); ++it)
64 (*it)->RemoveObserver(this); 65 (*it)->RemoveObserver(this);
65 66
66 download_manager_->RemoveObserver(this); 67 download_manager_->RemoveObserver(this);
67 } 68 }
68 69
69 void DownloadTestObserver::Init() { 70 void DownloadTestObserver::Init() {
70 download_manager_->AddObserver(this); // Will call initial ModelChanged(). 71 download_manager_->AddObserver(this); // Will call initial ModelChanged().
71 finished_downloads_at_construction_ = finished_downloads_.size(); 72 finished_downloads_at_construction_ = finished_downloads_.size();
72 states_observed_.clear(); 73 states_observed_.clear();
73 } 74 }
74 75
75 void DownloadTestObserver::WaitForFinished() { 76 void DownloadTestObserver::WaitForFinished() {
76 if (!IsFinished()) { 77 if (!IsFinished()) {
77 waiting_ = true; 78 waiting_ = true;
78 content::RunMessageLoop(); 79 content::RunMessageLoop();
jam 2012/08/13 16:00:21 ditto
Randy Smith (Not in Mondays) 2012/08/13 18:57:49 Done.
79 waiting_ = false; 80 waiting_ = false;
80 } 81 }
81 } 82 }
82 83
83 bool DownloadTestObserver::IsFinished() const { 84 bool DownloadTestObserver::IsFinished() const {
84 return (finished_downloads_.size() - finished_downloads_at_construction_ >= 85 return (finished_downloads_.size() - finished_downloads_at_construction_ >=
85 wait_count_); 86 wait_count_);
86 } 87 }
87 88
88 void DownloadTestObserver::OnDownloadDestroyed(DownloadItem* download) { 89 void DownloadTestObserver::OnDownloadDestroyed(DownloadItem* download) {
(...skipping 292 matching lines...) Expand 10 before | Expand all | Expand 10 after
381 if (waiting_) 382 if (waiting_)
382 MessageLoopForUI::current()->Quit(); 383 MessageLoopForUI::current()->Quit();
383 } 384 }
384 385
385 const content::DownloadUrlParameters::OnStartedCallback 386 const content::DownloadUrlParameters::OnStartedCallback
386 DownloadTestItemCreationObserver::callback() { 387 DownloadTestItemCreationObserver::callback() {
387 return base::Bind( 388 return base::Bind(
388 &DownloadTestItemCreationObserver::DownloadItemCreationCallback, this); 389 &DownloadTestItemCreationObserver::DownloadItemCreationCallback, this);
389 } 390 }
390 391
391 namespace internal { 392 } // namespace content
392
393 // Test ChromeDownloadManagerDelegate that controls whether how file chooser
394 // dialogs are handled. By default, file chooser dialogs are disabled.
395 class MockFileChooserDownloadManagerDelegate
396 : public ChromeDownloadManagerDelegate {
397 public:
398 explicit MockFileChooserDownloadManagerDelegate(Profile* profile)
399 : ChromeDownloadManagerDelegate(profile),
400 file_chooser_enabled_(false),
401 file_chooser_displayed_(false) {}
402
403 void EnableFileChooser(bool enable) {
404 file_chooser_enabled_ = enable;
405 }
406
407 bool TestAndResetDidShowFileChooser() {
408 bool did_show = file_chooser_displayed_;
409 file_chooser_displayed_ = false;
410 return did_show;
411 }
412
413 private:
414 virtual ~MockFileChooserDownloadManagerDelegate() {}
415
416 virtual void ChooseDownloadPath(DownloadItem* item,
417 const FilePath& suggested_path,
418 const FileSelectedCallback&
419 callback) OVERRIDE {
420 file_chooser_displayed_ = true;
421 MessageLoop::current()->PostTask(
422 FROM_HERE,
423 base::Bind(callback,
424 (file_chooser_enabled_ ? suggested_path : FilePath())));
425 }
426
427 bool file_chooser_enabled_;
428 bool file_chooser_displayed_;
429 };
430
431 } // namespace internal
432
433 DownloadTestFileChooserObserver::DownloadTestFileChooserObserver(
434 Profile* profile) {
435 test_delegate_ =
436 new internal::MockFileChooserDownloadManagerDelegate(profile);
437 DownloadServiceFactory::GetForProfile(profile)->
438 SetDownloadManagerDelegateForTesting(test_delegate_.get());
439 }
440
441 DownloadTestFileChooserObserver::~DownloadTestFileChooserObserver() {
442 }
443
444 void DownloadTestFileChooserObserver::EnableFileChooser(bool enable) {
445 test_delegate_->EnableFileChooser(enable);
446 }
447
448 bool DownloadTestFileChooserObserver::TestAndResetDidShowFileChooser() {
449 return test_delegate_->TestAndResetDidShowFileChooser();
450 }
OLDNEW
« content/public/test/download_test_observer.h ('K') | « content/public/test/download_test_observer.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698