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
|
// Copyright 2012 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/extensions/warning_badge_service.h"
#include "base/memory/raw_ptr.h"
#include "chrome/app/chrome_command_ids.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/global_error/global_error_service.h"
#include "chrome/browser/ui/global_error/global_error_service_factory.h"
#include "chrome/test/base/testing_profile.h"
#include "content/public/test/browser_task_environment.h"
#include "extensions/browser/warning_service.h"
#include "extensions/browser/warning_set.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions {
namespace {
class TestExtensionWarningSet : public WarningService {
public:
explicit TestExtensionWarningSet(Profile* profile)
: WarningService(profile) {}
~TestExtensionWarningSet() override = default;
void AddWarning(const Warning& warning) {
WarningSet warnings;
warnings.insert(warning);
AddWarnings(warnings);
}
};
class TestWarningBadgeService : public WarningBadgeService {
public:
TestWarningBadgeService(Profile* profile, WarningService* warning_service)
: WarningBadgeService(profile), warning_service_(warning_service) {}
~TestWarningBadgeService() override = default;
const std::set<Warning>& GetCurrentWarnings() const override {
return warning_service_->warnings();
}
private:
raw_ptr<WarningService> warning_service_;
};
bool HasBadge(Profile* profile) {
GlobalErrorService* service =
GlobalErrorServiceFactory::GetForProfile(profile);
return service->GetGlobalErrorByMenuItemCommandID(IDC_EXTENSION_ERRORS) !=
nullptr;
}
const char ext1_id[] = "extension1";
const char ext2_id[] = "extension2";
} // namespace
// Check that no badge appears if it has been suppressed for a specific
// warning.
TEST(WarningBadgeServiceTest, SuppressBadgeForCurrentWarnings) {
content::BrowserTaskEnvironment task_environment;
TestingProfile profile;
TestExtensionWarningSet warnings(&profile);
TestWarningBadgeService badge_service(&profile, &warnings);
warnings.AddObserver(&badge_service);
// Insert first warning.
warnings.AddWarning(Warning::CreateNetworkDelayWarning(ext1_id));
EXPECT_TRUE(HasBadge(&profile));
// Suppress first warning.
badge_service.SuppressCurrentWarnings();
EXPECT_FALSE(HasBadge(&profile));
// Simulate deinstallation of extension.
std::set<Warning::WarningType> to_clear =
warnings.GetWarningTypesAffectingExtension(ext1_id);
warnings.ClearWarnings(to_clear);
EXPECT_FALSE(HasBadge(&profile));
// Set first warning again and verify that not badge is shown this time.
warnings.AddWarning(Warning::CreateNetworkDelayWarning(ext1_id));
EXPECT_FALSE(HasBadge(&profile));
// Set second warning and verify that it shows a badge.
warnings.AddWarning(Warning::CreateRepeatedCacheFlushesWarning(ext2_id));
EXPECT_TRUE(HasBadge(&profile));
warnings.RemoveObserver(&badge_service);
}
} // namespace extensions
|
