1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "chrome/browser/extensions/extension_warning_badge_service.h"
7 #include "chrome/app/chrome_command_ids.h"
8 #include "chrome/browser/extensions/extension_warning_set.h"
9 #include "chrome/browser/profiles/profile.h"
10 #include "chrome/browser/ui/global_error/global_error_service.h"
11 #include "chrome/browser/ui/global_error/global_error_service_factory.h"
12 #include "chrome/test/base/testing_profile.h"
13 #include "testing/gtest/include/gtest/gtest.h"
15 namespace extensions
{
19 class TestExtensionWarningSet
: public ExtensionWarningService
{
21 explicit TestExtensionWarningSet(Profile
* profile
)
22 : ExtensionWarningService(profile
) {
24 virtual ~TestExtensionWarningSet() {}
26 void AddWarning(const ExtensionWarning
& warning
) {
27 ExtensionWarningSet warnings
;
28 warnings
.insert(warning
);
29 AddWarnings(warnings
);
33 class TestExtensionWarningBadgeService
: public ExtensionWarningBadgeService
{
35 TestExtensionWarningBadgeService(Profile
* profile
,
36 ExtensionWarningService
* warning_service
)
37 : ExtensionWarningBadgeService(profile
),
38 warning_service_(warning_service
) {}
39 virtual ~TestExtensionWarningBadgeService() {}
41 virtual const std::set
<ExtensionWarning
>&
42 GetCurrentWarnings() const OVERRIDE
{
43 return warning_service_
->warnings();
47 ExtensionWarningService
* warning_service_
;
50 bool HasBadge(Profile
* profile
) {
51 GlobalErrorService
* service
=
52 GlobalErrorServiceFactory::GetForProfile(profile
);
53 return service
->GetGlobalErrorByMenuItemCommandID(IDC_EXTENSION_ERRORS
) !=
57 const char* ext1_id
= "extension1";
58 const char* ext2_id
= "extension2";
59 const ExtensionWarning::WarningType warning_1
=
60 ExtensionWarning::kNetworkDelay
;
61 const ExtensionWarning::WarningType warning_2
=
62 ExtensionWarning::kNetworkConflict
;
66 // Check that no badge appears if it has been suppressed for a specific
68 TEST(ExtensionWarningBadgeServiceTest
, SuppressBadgeForCurrentWarnings
) {
69 TestingProfile profile
;
70 TestExtensionWarningSet
warnings(&profile
);
71 TestExtensionWarningBadgeService
badge_service(&profile
, &warnings
);
72 warnings
.AddObserver(&badge_service
);
74 // Insert first warning.
75 warnings
.AddWarning(ExtensionWarning::CreateNetworkDelayWarning(ext1_id
));
76 EXPECT_TRUE(HasBadge(&profile
));
78 // Suppress first warning.
79 badge_service
.SuppressCurrentWarnings();
80 EXPECT_FALSE(HasBadge(&profile
));
82 // Simulate deinstallation of extension.
83 std::set
<ExtensionWarning::WarningType
> to_clear
=
84 warnings
.GetWarningTypesAffectingExtension(ext1_id
);
85 warnings
.ClearWarnings(to_clear
);
86 EXPECT_FALSE(HasBadge(&profile
));
88 // Set first warning again and verify that not badge is shown this time.
89 warnings
.AddWarning(ExtensionWarning::CreateNetworkDelayWarning(ext1_id
));
90 EXPECT_FALSE(HasBadge(&profile
));
92 // Set second warning and verify that it shows a badge.
93 warnings
.AddWarning(ExtensionWarning::CreateNetworkConflictWarning(ext2_id
));
94 EXPECT_TRUE(HasBadge(&profile
));
96 warnings
.RemoveObserver(&badge_service
);
99 } // namespace extensions