15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "base/memory/scoped_vector.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/weak_ptr.h" 79ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stl_util.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/download/download_status_updater.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/test/mock_download_item.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/test/mock_download_manager.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/test/test_browser_thread.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gmock/include/gmock/gmock.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using testing::AtLeast; 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using testing::Invoke; 182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using testing::Mock; 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using testing::Return; 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using testing::SetArgPointee; 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using testing::StrictMock; 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using testing::WithArg; 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)using testing::_; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class TestDownloadStatusUpdater : public DownloadStatusUpdater { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestDownloadStatusUpdater() : notification_count_(0), 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) acceptable_notification_item_(NULL) { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetAcceptableNotificationItem(content::DownloadItem* item) { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) acceptable_notification_item_ = item; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t NotificationCount() { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return notification_count_; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void UpdateAppIconDownloadProgress( 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::DownloadItem* download) OVERRIDE { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++notification_count_; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (acceptable_notification_item_) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(acceptable_notification_item_, download); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t notification_count_; 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::DownloadItem* acceptable_notification_item_; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DownloadStatusUpdaterTest : public testing::Test { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DownloadStatusUpdaterTest() 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : updater_(new TestDownloadStatusUpdater()), 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui_thread_(content::BrowserThread::UI, &loop_) {} 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DownloadStatusUpdaterTest() { 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t mgr_idx = 0; mgr_idx < managers_.size(); ++mgr_idx) { 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*Manager(mgr_idx), RemoveObserver(_)); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t item_idx = 0; item_idx < manager_items_[mgr_idx].size(); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++item_idx) { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*Item(mgr_idx, item_idx), RemoveObserver(_)); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delete updater_; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) updater_ = NULL; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) VerifyAndClearExpectations(); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) managers_.clear(); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (std::vector<Items>::iterator it = manager_items_.begin(); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it != manager_items_.end(); ++it) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) STLDeleteContainerPointers(it->begin(), it->end()); 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) loop_.RunUntilIdle(); // Allow DownloadManager destruction. 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Attach some number of DownloadManagers to the updater. 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetupManagers(int manager_count) { 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(0U, managers_.size()); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < manager_count; ++i) { 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::MockDownloadManager* mgr = 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new StrictMock<content::MockDownloadManager>; 8290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) managers_.push_back(mgr); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetObserver(content::DownloadManager::Observer* observer) { 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_observers_[manager_observer_index_] = observer; 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Hook the specified manager into the updater. 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void LinkManager(int i) { 9290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) content::MockDownloadManager* mgr = managers_[i]; 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_observer_index_ = i; 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (manager_observers_.size() <= static_cast<size_t>(i)) { 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_observers_.push_back(NULL); 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*mgr, AddObserver(_)) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(WithArg<0>(Invoke( 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &DownloadStatusUpdaterTest::SetObserver))); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) updater_->AddManager(mgr); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Add some number of Download items to a particular manager. 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddItems(int manager_index, int item_count, int in_progress_count) { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_GT(managers_.size(), static_cast<size_t>(manager_index)); 10690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) content::MockDownloadManager* manager = managers_[manager_index]; 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (manager_items_.size() <= static_cast<size_t>(manager_index)) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_items_.resize(manager_index+1); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<content::DownloadItem*> item_list; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < item_count; ++i) { 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::MockDownloadItem* item = 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new StrictMock<content::MockDownloadItem>; 11590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) content::DownloadItem::DownloadState state = 11690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) i < in_progress_count ? content::DownloadItem::IN_PROGRESS 11790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) : content::DownloadItem::CANCELLED; 11890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*item, GetState()).WillRepeatedly(Return(state)); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*item, AddObserver(_)) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(Return()); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_items_[manager_index].push_back(item); 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*manager, GetAllDownloads(_)) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(SetArgPointee<0>(manager_items_[manager_index])); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Return the specified manager. 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::MockDownloadManager* Manager(int manager_index) { 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_GT(managers_.size(), static_cast<size_t>(manager_index)); 13090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return managers_[manager_index]; 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Return the specified item. 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::MockDownloadItem* Item(int manager_index, int item_index) { 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_GT(manager_items_.size(), static_cast<size_t>(manager_index)); 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_GT(manager_items_[manager_index].size(), 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static_cast<size_t>(item_index)); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // All DownloadItems in manager_items_ are MockDownloadItems. 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return static_cast<content::MockDownloadItem*>( 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_items_[manager_index][item_index]); 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set return values relevant to |DownloadStatusUpdater::GetProgress()| 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // for the specified item. 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetItemValues(int manager_index, int item_index, 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int received_bytes, int total_bytes, bool notify) { 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::MockDownloadItem* item(Item(manager_index, item_index)); 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*item, GetReceivedBytes()) 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Return(received_bytes)); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*item, GetTotalBytes()) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Return(total_bytes)); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (notify) 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) updater_->OnDownloadUpdated(managers_[manager_index], item); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Transition specified item to completed. 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void CompleteItem(int manager_index, int item_index) { 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::MockDownloadItem* item(Item(manager_index, item_index)); 15990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*item, GetState()) 16090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) .WillRepeatedly(Return(content::DownloadItem::COMPLETE)); 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) updater_->OnDownloadUpdated(managers_[manager_index], item); 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify and clear all mocks expectations. 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void VerifyAndClearExpectations() { 16690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) for (ScopedVector<content::MockDownloadManager>::iterator it 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) = managers_.begin(); it != managers_.end(); ++it) 16890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) Mock::VerifyAndClearExpectations(*it); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (std::vector<Items>::iterator it = manager_items_.begin(); 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it != manager_items_.end(); ++it) 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (Items::iterator sit = it->begin(); sit != it->end(); ++sit) 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(*sit); 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 17590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ScopedVector<content::MockDownloadManager> managers_; 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // DownloadItem so that it can be assigned to the result of SearchDownloads. 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef std::vector<content::DownloadItem*> Items; 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<Items> manager_items_; 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int manager_observer_index_; 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<content::DownloadManager::Observer*> manager_observers_; 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Pointer so we can verify that destruction triggers appropriate 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // changes. 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestDownloadStatusUpdater *updater_; 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Thread so that the DownloadManager (which is a DeleteOnUIThread 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // object) can be deleted. 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rdsmith): This can be removed when the DownloadManager 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // is no longer required to be deleted on the UI thread. 19190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoop loop_; 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::TestBrowserThread ui_thread_; 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test null updater. 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(DownloadStatusUpdaterTest, Basic) { 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) float progress = -1; 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int download_count = -1; 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(updater_->GetProgress(&progress, &download_count)); 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FLOAT_EQ(0.0f, progress); 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, download_count); 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test updater with null manager. 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(DownloadStatusUpdaterTest, OneManagerNoItems) { 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupManagers(1); 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(0, 0, 0); 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(0); 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) VerifyAndClearExpectations(); 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) float progress = -1; 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int download_count = -1; 21390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*managers_[0], GetAllDownloads(_)) 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(SetArgPointee<0>(manager_items_[0])); 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(updater_->GetProgress(&progress, &download_count)); 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FLOAT_EQ(0.0f, progress); 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, download_count); 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test updater with non-null manager, including transition an item to 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |content::DownloadItem::COMPLETE| and adding a new item. 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(DownloadStatusUpdaterTest, OneManagerManyItems) { 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupManagers(1); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(0, 3, 2); 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(0); 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Prime items 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 0, 10, 20, false); 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 1, 50, 60, false); 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 2, 90, 90, false); 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) float progress = -1; 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int download_count = -1; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(updater_->GetProgress(&progress, &download_count)); 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FLOAT_EQ((10+50)/(20.0f+60), progress); 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(2, download_count); 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Transition one item to completed and confirm progress is updated 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // properly. 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompleteItem(0, 0); 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(updater_->GetProgress(&progress, &download_count)); 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FLOAT_EQ(50/60.0f, progress); 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, download_count); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Add a new item to manager and confirm progress is updated properly. 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(0, 1, 1); 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 3, 150, 200, false); 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_observers_[0]->OnDownloadCreated( 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) managers_[0], manager_items_[0][manager_items_[0].size()-1]); 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(updater_->GetProgress(&progress, &download_count)); 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FLOAT_EQ((50+150)/(60+200.0f), progress); 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(2, download_count); 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test to ensure that the download progress notification is called correctly. 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(DownloadStatusUpdaterTest, ProgressNotification) { 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t expected_notifications = updater_->NotificationCount(); 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupManagers(1); 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(0, 2, 2); 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(0); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Expect two notifications, one for each item; which item will come first 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // isn't defined so it cannot be tested. 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_notifications += 2; 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(expected_notifications, updater_->NotificationCount()); 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make progress on the first item. 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) updater_->SetAcceptableNotificationItem(Item(0, 0)); 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 0, 10, 20, true); 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++expected_notifications; 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(expected_notifications, updater_->NotificationCount()); 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Second item completes! 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) updater_->SetAcceptableNotificationItem(Item(0, 1)); 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompleteItem(0, 1); 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++expected_notifications; 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(expected_notifications, updater_->NotificationCount()); 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // First item completes. 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) updater_->SetAcceptableNotificationItem(Item(0, 0)); 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompleteItem(0, 0); 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++expected_notifications; 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(expected_notifications, updater_->NotificationCount()); 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) updater_->SetAcceptableNotificationItem(NULL); 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Confirm we recognize the situation where we have an unknown size. 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(DownloadStatusUpdaterTest, UnknownSize) { 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupManagers(1); 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(0, 2, 2); 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(0); 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Prime items 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 0, 10, 20, false); 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 1, 50, -1, false); 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) float progress = -1; 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int download_count = -1; 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(updater_->GetProgress(&progress, &download_count)); 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test many null managers. 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(DownloadStatusUpdaterTest, ManyManagersNoItems) { 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupManagers(1); 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(0, 0, 0); 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(0); 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) float progress = -1; 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int download_count = -1; 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(updater_->GetProgress(&progress, &download_count)); 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FLOAT_EQ(0.0f, progress); 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, download_count); 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test many managers with all items complete. 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(DownloadStatusUpdaterTest, ManyManagersEmptyItems) { 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupManagers(2); 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(0, 3, 0); 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(0); 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(1, 3, 0); 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(1); 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) float progress = -1; 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int download_count = -1; 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(updater_->GetProgress(&progress, &download_count)); 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FLOAT_EQ(0.0f, progress); 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, download_count); 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test many managers with some non-complete items. 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(DownloadStatusUpdaterTest, ManyManagersMixedItems) { 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupManagers(2); 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(0, 3, 2); 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(0); 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddItems(1, 3, 1); 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LinkManager(1); 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 0, 10, 20, false); 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(0, 1, 50, 60, false); 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetItemValues(1, 0, 80, 90, false); 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) float progress = -1; 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int download_count = -1; 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(updater_->GetProgress(&progress, &download_count)); 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FLOAT_EQ((10+50+80)/(20.0f+60+90), progress); 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(3, download_count); 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 350