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
|
// 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 <memory>
#include "base/memory/raw_ptr.h"
#include "chrome/browser/lifetime/application_lifetime.h"
#include "chrome/browser/notifications/notification_system_observer.h"
#include "chrome/browser/notifications/notification_test_util.h"
#include "chrome/browser/notifications/notification_ui_manager.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/testing_profile_manager.h"
#include "components/crx_file/id_util.h"
#include "content/public/test/browser_task_environment.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/browser/unloaded_extension_reason.h"
#include "extensions/common/extension_builder.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
scoped_refptr<const extensions::Extension> CreateGoodExtension(
const std::string& name) {
return extensions::ExtensionBuilder(name).Build();
}
} // namespace
class NotificationSystemObserverTest : public testing::Test {
protected:
NotificationSystemObserverTest() = default;
~NotificationSystemObserverTest() override = default;
// testing::Test:
void SetUp() override {
profile_manager_ = std::make_unique<TestingProfileManager>(
TestingBrowserProcess::GetGlobal());
ASSERT_TRUE(profile_manager_->SetUp());
profile_ = profile_manager_->CreateTestingProfile("test-profile");
ui_manager_ = std::make_unique<StubNotificationUIManager>();
notification_observer_ =
std::make_unique<NotificationSystemObserver>(ui_manager_.get());
}
TestingProfileManager* profile_manager() { return profile_manager_.get(); }
TestingProfile* profile() { return profile_; }
StubNotificationUIManager* ui_manager() { return ui_manager_.get(); }
private:
content::BrowserTaskEnvironment task_environment_;
std::unique_ptr<TestingProfileManager> profile_manager_;
raw_ptr<TestingProfile> profile_ = nullptr;
std::unique_ptr<StubNotificationUIManager> ui_manager_;
std::unique_ptr<NotificationSystemObserver> notification_observer_;
};
TEST_F(NotificationSystemObserverTest, MultiProfileExtensionUnloaded) {
scoped_refptr<const extensions::Extension> extension =
CreateGoodExtension("foo");
ASSERT_NE(extension->url(), ui_manager()->last_canceled_source());
// Claim the extension has been unloaded.
extensions::ExtensionRegistry::Get(profile())->TriggerOnUnloaded(
extension.get(), extensions::UnloadedExtensionReason::UNINSTALL);
// Check the NotificationUIManger has canceled the source coming from the
// unloaded extension.
EXPECT_EQ(extension->url(), ui_manager()->last_canceled_source());
TestingProfile* profile2 =
profile_manager()->CreateTestingProfile("test-profile2");
scoped_refptr<const extensions::Extension> extension2 =
CreateGoodExtension("bar");
// Claim the extension has been unloaded with anoter profile.
extensions::ExtensionRegistry::Get(profile2)->TriggerOnUnloaded(
extension2.get(), extensions::UnloadedExtensionReason::UNINSTALL);
EXPECT_EQ(extension2->url(), ui_manager()->last_canceled_source());
}
|
