| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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 "base/memory/scoped_ptr.h" | |
| 6 #include "base/memory/weak_ptr.h" | |
| 7 #include "chrome/browser/download/download_status_updater.h" | |
| 8 #include "chrome/browser/download/download_status_updater_delegate.h" | |
| 9 #include "testing/gtest/include/gtest/gtest.h" | |
| 10 | |
| 11 namespace { | |
| 12 | |
| 13 class MockDelegate : public DownloadStatusUpdaterDelegate { | |
| 14 public: | |
| 15 explicit MockDelegate(DownloadStatusUpdater* updater) | |
| 16 : updater_(updater->AsWeakPtr()), | |
| 17 is_download_progress_known_(true), | |
| 18 in_progress_download_count_(0), | |
| 19 received_bytes_(0), | |
| 20 total_bytes_(0) { | |
| 21 EXPECT_TRUE(updater); | |
| 22 if (updater_) | |
| 23 updater_->AddDelegate(this); | |
| 24 } | |
| 25 | |
| 26 ~MockDelegate() { | |
| 27 EXPECT_TRUE(updater_); | |
| 28 if (updater_) | |
| 29 updater_->RemoveDelegate(this); | |
| 30 } | |
| 31 | |
| 32 // Overriden from DownloadStatusUpdaterDelegate: | |
| 33 virtual bool IsDownloadProgressKnown() { | |
| 34 return is_download_progress_known_; | |
| 35 } | |
| 36 | |
| 37 virtual int64 GetInProgressDownloadCount() { | |
| 38 return in_progress_download_count_; | |
| 39 } | |
| 40 | |
| 41 virtual int64 GetReceivedDownloadBytes() { | |
| 42 return received_bytes_; | |
| 43 } | |
| 44 | |
| 45 virtual int64 GetTotalDownloadBytes() { | |
| 46 return total_bytes_; | |
| 47 } | |
| 48 | |
| 49 void set_is_download_progress_known(bool progress_known) { | |
| 50 is_download_progress_known_ = progress_known; | |
| 51 } | |
| 52 | |
| 53 void set_in_progress_download_count(int64 download_count) { | |
| 54 in_progress_download_count_ = download_count; | |
| 55 } | |
| 56 | |
| 57 void set_received_bytes(int64 received_bytes) { | |
| 58 received_bytes_ = received_bytes; | |
| 59 } | |
| 60 | |
| 61 void set_total_bytes(int64 total_bytes) { | |
| 62 total_bytes_ = total_bytes; | |
| 63 } | |
| 64 | |
| 65 private: | |
| 66 base::WeakPtr<DownloadStatusUpdater> updater_; | |
| 67 | |
| 68 bool is_download_progress_known_; | |
| 69 int64 in_progress_download_count_; | |
| 70 int64 received_bytes_; | |
| 71 int64 total_bytes_; | |
| 72 | |
| 73 DISALLOW_COPY_AND_ASSIGN(MockDelegate); | |
| 74 }; | |
| 75 | |
| 76 } // namespace | |
| 77 | |
| 78 class DownloadStatusUpdaterTest : public testing::Test { | |
| 79 protected: | |
| 80 DownloadStatusUpdater updater_; | |
| 81 }; | |
| 82 | |
| 83 TEST_F(DownloadStatusUpdaterTest, Basic) { | |
| 84 float progress = -1; | |
| 85 EXPECT_TRUE(updater_.GetProgress(&progress)); | |
| 86 EXPECT_FLOAT_EQ(0, progress); | |
| 87 EXPECT_EQ(0, updater_.GetInProgressDownloadCount()); | |
| 88 } | |
| 89 | |
| 90 TEST_F(DownloadStatusUpdaterTest, OneDelegate) { | |
| 91 MockDelegate delegate(&updater_); | |
| 92 | |
| 93 { | |
| 94 float progress = -1; | |
| 95 EXPECT_TRUE(updater_.GetProgress(&progress)); | |
| 96 EXPECT_FLOAT_EQ(0, progress); | |
| 97 EXPECT_EQ(0, updater_.GetInProgressDownloadCount()); | |
| 98 } | |
| 99 | |
| 100 delegate.set_in_progress_download_count(1); | |
| 101 delegate.set_received_bytes(21); | |
| 102 delegate.set_total_bytes(42); | |
| 103 | |
| 104 { | |
| 105 float progress = -1; | |
| 106 EXPECT_TRUE(updater_.GetProgress(&progress)); | |
| 107 EXPECT_FLOAT_EQ(static_cast<float>(1) / 2, progress); | |
| 108 EXPECT_EQ(1, updater_.GetInProgressDownloadCount()); | |
| 109 } | |
| 110 | |
| 111 delegate.set_is_download_progress_known(false); | |
| 112 | |
| 113 { | |
| 114 float progress = -1; | |
| 115 EXPECT_FALSE(updater_.GetProgress(&progress)); | |
| 116 EXPECT_FLOAT_EQ(0, progress); | |
| 117 EXPECT_EQ(1, updater_.GetInProgressDownloadCount()); | |
| 118 } | |
| 119 } | |
| 120 | |
| 121 TEST_F(DownloadStatusUpdaterTest, MultipleDelegates) { | |
| 122 scoped_ptr<MockDelegate> delegate1(new MockDelegate(&updater_)); | |
| 123 scoped_ptr<MockDelegate> delegate2(new MockDelegate(&updater_)); | |
| 124 | |
| 125 { | |
| 126 float progress = -1; | |
| 127 EXPECT_TRUE(updater_.GetProgress(&progress)); | |
| 128 EXPECT_FLOAT_EQ(0, progress); | |
| 129 EXPECT_EQ(0, updater_.GetInProgressDownloadCount()); | |
| 130 } | |
| 131 | |
| 132 delegate1->set_in_progress_download_count(1); | |
| 133 delegate1->set_received_bytes(14); | |
| 134 delegate1->set_total_bytes(21); | |
| 135 | |
| 136 delegate2->set_in_progress_download_count(2); | |
| 137 delegate2->set_received_bytes(7); | |
| 138 delegate2->set_total_bytes(21); | |
| 139 | |
| 140 { | |
| 141 float progress = -1; | |
| 142 EXPECT_TRUE(updater_.GetProgress(&progress)); | |
| 143 EXPECT_FLOAT_EQ(static_cast<float>(1) / 2, progress); | |
| 144 EXPECT_EQ(3, updater_.GetInProgressDownloadCount()); | |
| 145 } | |
| 146 | |
| 147 delegate1->set_is_download_progress_known(false); | |
| 148 | |
| 149 { | |
| 150 float progress = -1; | |
| 151 EXPECT_FALSE(updater_.GetProgress(&progress)); | |
| 152 EXPECT_FLOAT_EQ(0, progress); | |
| 153 EXPECT_EQ(3, updater_.GetInProgressDownloadCount()); | |
| 154 } | |
| 155 | |
| 156 delegate1.reset(); | |
| 157 | |
| 158 { | |
| 159 float progress = -1; | |
| 160 EXPECT_TRUE(updater_.GetProgress(&progress)); | |
| 161 EXPECT_FLOAT_EQ(static_cast<float>(1) / 3, progress); | |
| 162 EXPECT_EQ(2, updater_.GetInProgressDownloadCount()); | |
| 163 } | |
| 164 } | |
| OLD | NEW |