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 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/device_notifications/device_pinned_notification_unittest.h"
#include <string>
#include "base/strings/stringprintf.h"
#include "build/build_config.h"
#include "chrome/browser/device_notifications/device_pinned_notification_renderer.h"
#include "chrome/browser/device_notifications/device_system_tray_icon.h"
#include "chrome/browser/notifications/notification_display_service_tester.h"
#include "chrome/browser/notifications/system_notification_helper.h"
#include "chrome/test/base/testing_browser_process.h"
#include "extensions/common/constants.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#if BUILDFLAG(IS_CHROMEOS)
#include "ash/constants/ash_features.h"
#endif // BUILDFLAG(IS_CHROMEOS)
#if BUILDFLAG(ENABLE_EXTENSIONS)
#include "extensions/common/constants.h"
#include "extensions/common/extension.h"
#endif // BUILDFLAG(ENABLE_EXTENSIONS)
DevicePinnedNotificationTestBase::DevicePinnedNotificationTestBase(
std::u16string device_content_settings_label)
: device_content_settings_label_(std::move(device_content_settings_label)) {
}
DevicePinnedNotificationTestBase::~DevicePinnedNotificationTestBase() = default;
void DevicePinnedNotificationTestBase::SetUp() {
DeviceSystemTrayIconTestBase::SetUp();
TestingBrowserProcess::GetGlobal()->SetSystemNotificationHelper(
std::make_unique<SystemNotificationHelper>());
display_service_ =
std::make_unique<NotificationDisplayServiceTester>(/*profile=*/nullptr);
}
void DevicePinnedNotificationTestBase::TearDown() {
TestingBrowserProcess::GetGlobal()->SetSystemNotificationHelper(nullptr);
DeviceSystemTrayIconTestBase::TearDown();
}
void DevicePinnedNotificationTestBase::CheckIcon(
const std::vector<DeviceSystemTrayIconTestBase::ProfileItem>&
profile_connection_counts) {
EXPECT_FALSE(display_service_
->GetDisplayedNotificationsForType(
NotificationHandler::Type::TRANSIENT)
.empty());
// Check each button label and behavior of clicking the button.
for (const auto& [profile, origin_items] : profile_connection_counts) {
size_t total_connection_count = 0;
for (const auto& [origin, connection_count, name] : origin_items) {
total_connection_count += connection_count;
}
auto* device_pinned_notification_renderer =
GetDevicePinnedNotificationRenderer();
auto* connection_tracker = GetDeviceConnectionTracker(profile,
/*create=*/false);
ASSERT_TRUE(connection_tracker);
auto maybe_notification = display_service_->GetNotification(
device_pinned_notification_renderer->GetNotificationId(profile));
ASSERT_TRUE(maybe_notification);
EXPECT_EQ(maybe_notification->title(),
GetExpectedTitle(origin_items.size(), total_connection_count));
EXPECT_EQ(maybe_notification->message(), GetExpectedMessage(origin_items));
EXPECT_EQ(maybe_notification->priority(), message_center::LOW_PRIORITY);
ASSERT_EQ(maybe_notification->rich_notification_data().buttons.size(), 1u);
#if BUILDFLAG(IS_CHROMEOS)
if (!ash::features::AreOngoingProcessesEnabled()) {
EXPECT_EQ(maybe_notification->rich_notification_data().buttons[0].title,
device_content_settings_label_);
}
#else
EXPECT_EQ(maybe_notification->rich_notification_data().buttons[0].title,
device_content_settings_label_);
#endif // BUILDFLAG(IS_CHROMEOS)
EXPECT_TRUE(maybe_notification->delegate());
EXPECT_CALL(*GetMockDeviceConnectionTracker(connection_tracker),
ShowContentSettingsExceptions());
SimulateButtonClick(profile);
}
}
void DevicePinnedNotificationTestBase::CheckIconHidden() {
EXPECT_TRUE(display_service_
->GetDisplayedNotificationsForType(
NotificationHandler::Type::TRANSIENT)
.empty());
}
void DevicePinnedNotificationTestBase::SimulateButtonClick(Profile* profile) {
auto* device_pinned_notification_renderer =
GetDevicePinnedNotificationRenderer();
display_service_->SimulateClick(
NotificationHandler::Type::TRANSIENT,
device_pinned_notification_renderer->GetNotificationId(profile),
/*action_index=*/0, /*reply=*/std::nullopt);
}
#if BUILDFLAG(ENABLE_EXTENSIONS)
// Test message in pinned notification for up to 5 extensions.
void DevicePinnedNotificationTestBase::
TestMultipleExtensionsNotificationMessage() {
Profile* profile = CreateTestingProfile("user");
auto* connection_tracker = GetDeviceConnectionTracker(profile,
/*create=*/true);
std::vector<DeviceSystemTrayIconTestBase::OriginItem> origin_items;
for (size_t idx = 0; idx < 5; idx++) {
auto extension =
CreateExtensionWithName(base::StringPrintf("Test Extension %zu", idx));
AddExtensionToProfile(profile, extension.get());
connection_tracker->IncrementConnectionCount(extension.get()->origin());
origin_items.emplace_back(extension->origin(), 1, extension->name());
CheckIcon({{profile, origin_items}});
}
}
#endif // BUILDFLAG(ENABLE_EXTENSIONS)
|
