diff options
| author | phajdan.jr@chromium.org <phajdan.jr@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2010-09-27 09:20:02 +0000 |
|---|---|---|
| committer | phajdan.jr@chromium.org <phajdan.jr@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2010-09-27 09:20:02 +0000 |
| commit | 073ed7b86bdbb6b90a305767b8de6ee9405fad7f (patch) | |
| tree | d10548ec6957aca4f581c03cfb6c823d197fb3f1 /chrome/browser/download/download_status_updater_unittest.cc | |
| parent | ebb84b2d96cb19aaa5349d92d1b2590dd09a93a0 (diff) | |
| download | chromium_src-073ed7b86bdbb6b90a305767b8de6ee9405fad7f.zip chromium_src-073ed7b86bdbb6b90a305767b8de6ee9405fad7f.tar.gz chromium_src-073ed7b86bdbb6b90a305767b8de6ee9405fad7f.tar.bz2 | |
Create DownloadStatusUpdater to keep track of download progress
and update the application icon.
We can't do that in DownloadManager because there are multiple
DownloadManagers (one per Profile) and they would overwrite
each other's updates.
BUG=none
TEST=none
Review URL: http://codereview.chromium.org/3454018
git-svn-id: svn://svn.chromium.org/chrome/trunk/src@60625 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'chrome/browser/download/download_status_updater_unittest.cc')
| -rw-r--r-- | chrome/browser/download/download_status_updater_unittest.cc | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/chrome/browser/download/download_status_updater_unittest.cc b/chrome/browser/download/download_status_updater_unittest.cc new file mode 100644 index 0000000..c74c0bf --- /dev/null +++ b/chrome/browser/download/download_status_updater_unittest.cc @@ -0,0 +1,159 @@ +// Copyright (c) 2010 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 "base/scoped_ptr.h" +#include "chrome/browser/download/download_status_updater.h" +#include "chrome/browser/download/download_status_updater_delegate.h" +#include "testing/gtest/include/gtest/gtest.h" + +namespace { + +class MockDelegate : public DownloadStatusUpdaterDelegate { + public: + explicit MockDelegate(DownloadStatusUpdater* updater) + : updater_(updater), + is_download_progress_known_(true), + in_progress_download_count_(0), + received_bytes_(0), + total_bytes_(0) { + updater_->AddDelegate(this); + } + + ~MockDelegate() { + updater_->RemoveDelegate(this); + } + + // Overriden from DownloadStatusUpdaterDelegate: + virtual bool IsDownloadProgressKnown() { + return is_download_progress_known_; + } + + virtual int64 GetInProgressDownloadCount() { + return in_progress_download_count_; + } + + virtual int64 GetReceivedDownloadBytes() { + return received_bytes_; + } + + virtual int64 GetTotalDownloadBytes() { + return total_bytes_; + } + + void set_is_download_progress_known(bool progress_known) { + is_download_progress_known_ = progress_known; + } + + void set_in_progress_download_count(int64 download_count) { + in_progress_download_count_ = download_count; + } + + void set_received_bytes(int64 received_bytes) { + received_bytes_ = received_bytes; + } + + void set_total_bytes(int64 total_bytes) { + total_bytes_ = total_bytes; + } + + private: + DownloadStatusUpdater* updater_; + + bool is_download_progress_known_; + int64 in_progress_download_count_; + int64 received_bytes_; + int64 total_bytes_; + + DISALLOW_COPY_AND_ASSIGN(MockDelegate); +}; + +} // namespace + +class DownloadStatusUpdaterTest : public testing::Test { + protected: + DownloadStatusUpdater updater_; +}; + +TEST_F(DownloadStatusUpdaterTest, Basic) { + float progress = -1; + EXPECT_TRUE(updater_.GetProgress(&progress)); + EXPECT_FLOAT_EQ(0, progress); + EXPECT_EQ(0, updater_.GetInProgressDownloadCount()); +} + +TEST_F(DownloadStatusUpdaterTest, OneDelegate) { + MockDelegate delegate(&updater_); + + { + float progress = -1; + EXPECT_TRUE(updater_.GetProgress(&progress)); + EXPECT_FLOAT_EQ(0, progress); + EXPECT_EQ(0, updater_.GetInProgressDownloadCount()); + } + + delegate.set_in_progress_download_count(1); + delegate.set_received_bytes(21); + delegate.set_total_bytes(42); + + { + float progress = -1; + EXPECT_TRUE(updater_.GetProgress(&progress)); + EXPECT_FLOAT_EQ(static_cast<float>(1) / 2, progress); + EXPECT_EQ(1, updater_.GetInProgressDownloadCount()); + } + + delegate.set_is_download_progress_known(false); + + { + float progress = -1; + EXPECT_FALSE(updater_.GetProgress(&progress)); + EXPECT_FLOAT_EQ(0, progress); + EXPECT_EQ(1, updater_.GetInProgressDownloadCount()); + } +} + +TEST_F(DownloadStatusUpdaterTest, MultipleDelegates) { + scoped_ptr<MockDelegate> delegate1(new MockDelegate(&updater_)); + scoped_ptr<MockDelegate> delegate2(new MockDelegate(&updater_)); + + { + float progress = -1; + EXPECT_TRUE(updater_.GetProgress(&progress)); + EXPECT_FLOAT_EQ(0, progress); + EXPECT_EQ(0, updater_.GetInProgressDownloadCount()); + } + + delegate1->set_in_progress_download_count(1); + delegate1->set_received_bytes(14); + delegate1->set_total_bytes(21); + + delegate2->set_in_progress_download_count(2); + delegate2->set_received_bytes(7); + delegate2->set_total_bytes(21); + + { + float progress = -1; + EXPECT_TRUE(updater_.GetProgress(&progress)); + EXPECT_FLOAT_EQ(static_cast<float>(1) / 2, progress); + EXPECT_EQ(3, updater_.GetInProgressDownloadCount()); + } + + delegate1->set_is_download_progress_known(false); + + { + float progress = -1; + EXPECT_FALSE(updater_.GetProgress(&progress)); + EXPECT_FLOAT_EQ(0, progress); + EXPECT_EQ(3, updater_.GetInProgressDownloadCount()); + } + + delegate1.reset(); + + { + float progress = -1; + EXPECT_TRUE(updater_.GetProgress(&progress)); + EXPECT_FLOAT_EQ(static_cast<float>(1) / 3, progress); + EXPECT_EQ(2, updater_.GetInProgressDownloadCount()); + } +} |