1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/download/content/public/all_download_item_notifier.h"
#include <memory>
#include "components/download/public/common/mock_download_item.h"
#include "content/public/test/mock_download_manager.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using testing::NiceMock;
using testing::SetArgPointee;
using testing::_;
namespace download {
namespace {
class MockNotifierObserver : public AllDownloadItemNotifier::Observer {
public:
MockNotifierObserver() = default;
MockNotifierObserver(const MockNotifierObserver&) = delete;
MockNotifierObserver& operator=(const MockNotifierObserver&) = delete;
~MockNotifierObserver() override = default;
MOCK_METHOD(void,
OnDownloadCreated,
(content::DownloadManager * manager, DownloadItem* item),
(override));
MOCK_METHOD(void,
OnDownloadUpdated,
(content::DownloadManager * manager, DownloadItem* item),
(override));
MOCK_METHOD(void,
OnDownloadOpened,
(content::DownloadManager * manager, DownloadItem* item),
(override));
MOCK_METHOD(void,
OnDownloadRemoved,
(content::DownloadManager * manager, DownloadItem* item),
(override));
MOCK_METHOD(void,
OnDownloadDestroyed,
(content::DownloadManager * manager, DownloadItem* item),
(override));
};
class AllDownloadItemNotifierTest : public testing::Test {
public:
AllDownloadItemNotifierTest()
: download_manager_(new content::MockDownloadManager) {}
AllDownloadItemNotifierTest(const AllDownloadItemNotifierTest&) = delete;
AllDownloadItemNotifierTest& operator=(const AllDownloadItemNotifierTest&) =
delete;
~AllDownloadItemNotifierTest() override = default;
content::MockDownloadManager& manager() { return *download_manager_; }
download::MockDownloadItem& item() { return item_; }
DownloadItem::Observer* NotifierAsItemObserver() const {
return notifier_.get();
}
content::DownloadManager::Observer* NotifierAsManagerObserver() const {
return notifier_.get();
}
MockNotifierObserver& observer() { return observer_; }
void SetNotifier() {
EXPECT_CALL(*download_manager_, AddObserver(_));
notifier_ = std::make_unique<AllDownloadItemNotifier>(
download_manager_.get(), &observer_);
}
void ClearNotifier() { notifier_.reset(); }
private:
NiceMock<download::MockDownloadItem> item_;
std::unique_ptr<content::MockDownloadManager> download_manager_;
std::unique_ptr<AllDownloadItemNotifier> notifier_;
NiceMock<MockNotifierObserver> observer_;
};
} // namespace
TEST_F(AllDownloadItemNotifierTest, AllDownloadItemNotifierTest_0) {
content::DownloadManager::DownloadVector items;
items.push_back(&item());
EXPECT_CALL(manager(), GetAllDownloads(_)).WillOnce(SetArgPointee<0>(items));
SetNotifier();
EXPECT_CALL(observer(), OnDownloadUpdated(&manager(), &item()));
NotifierAsItemObserver()->OnDownloadUpdated(&item());
EXPECT_CALL(observer(), OnDownloadOpened(&manager(), &item()));
NotifierAsItemObserver()->OnDownloadOpened(&item());
EXPECT_CALL(observer(), OnDownloadRemoved(&manager(), &item()));
NotifierAsItemObserver()->OnDownloadRemoved(&item());
EXPECT_CALL(observer(), OnDownloadDestroyed(&manager(), &item()));
NotifierAsItemObserver()->OnDownloadDestroyed(&item());
EXPECT_CALL(manager(), RemoveObserver(NotifierAsManagerObserver()));
ClearNotifier();
}
TEST_F(AllDownloadItemNotifierTest, AllDownloadItemNotifierTest_1) {
EXPECT_CALL(manager(), GetAllDownloads(_));
SetNotifier();
EXPECT_CALL(observer(), OnDownloadCreated(&manager(), &item()));
NotifierAsManagerObserver()->OnDownloadCreated(&manager(), &item());
EXPECT_CALL(manager(), RemoveObserver(NotifierAsManagerObserver()));
NotifierAsManagerObserver()->ManagerGoingDown(&manager());
ClearNotifier();
}
} // namespace download
|