1 // Copyright 2013 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.
4
5 #include "chrome/browser/extensions/extension_notification_observer.h"
6
7 #include <vector>
8
9 #include "base/logging.h"
10 #include "base/strings/stringprintf.h"
11 #include "extensions/common/extension.h"
12
13 namespace extensions {
14
15 namespace {
16
Str(const std::vector<chrome::NotificationType> & types)17 std::string Str(const std::vector<chrome::NotificationType>& types) {
18 std::string str = "[";
19 bool needs_comma = false;
20 for (std::vector<chrome::NotificationType>::const_iterator it =
21 types.begin(); it != types.end(); ++it) {
22 if (needs_comma)
23 str += ",";
24 needs_comma = true;
25 str += base::StringPrintf("%d", *it);
26 }
27 str += "]";
28 return str;
29 }
30
31 } // namespace
32
ExtensionNotificationObserver(content::NotificationSource source,const std::set<std::string> & extension_ids)33 ExtensionNotificationObserver::ExtensionNotificationObserver(
34 content::NotificationSource source,
35 const std::set<std::string>& extension_ids)
36 : extension_ids_(extension_ids) {
37 registrar_.Add(
38 this, chrome::NOTIFICATION_EXTENSION_LOADED_DEPRECATED, source);
39 registrar_.Add(
40 this, chrome::NOTIFICATION_EXTENSION_INSTALLED_DEPRECATED, source);
41 registrar_.Add(
42 this, chrome::NOTIFICATION_EXTENSION_UNLOADED_DEPRECATED, source);
43 }
44
~ExtensionNotificationObserver()45 ExtensionNotificationObserver::~ExtensionNotificationObserver() {}
46
CheckNotifications()47 testing::AssertionResult ExtensionNotificationObserver::CheckNotifications() {
48 return CheckNotifications(std::vector<chrome::NotificationType>());
49 }
50
CheckNotifications(chrome::NotificationType type)51 testing::AssertionResult ExtensionNotificationObserver::CheckNotifications(
52 chrome::NotificationType type) {
53 return CheckNotifications(std::vector<chrome::NotificationType>(1, type));
54 }
55
CheckNotifications(chrome::NotificationType t1,chrome::NotificationType t2)56 testing::AssertionResult ExtensionNotificationObserver::CheckNotifications(
57 chrome::NotificationType t1,
58 chrome::NotificationType t2) {
59 std::vector<chrome::NotificationType> types;
60 types.push_back(t1);
61 types.push_back(t2);
62 return CheckNotifications(types);
63 }
64
CheckNotifications(chrome::NotificationType t1,chrome::NotificationType t2,chrome::NotificationType t3)65 testing::AssertionResult ExtensionNotificationObserver::CheckNotifications(
66 chrome::NotificationType t1,
67 chrome::NotificationType t2,
68 chrome::NotificationType t3) {
69 std::vector<chrome::NotificationType> types;
70 types.push_back(t1);
71 types.push_back(t2);
72 types.push_back(t3);
73 return CheckNotifications(types);
74 }
75
CheckNotifications(chrome::NotificationType t1,chrome::NotificationType t2,chrome::NotificationType t3,chrome::NotificationType t4,chrome::NotificationType t5,chrome::NotificationType t6)76 testing::AssertionResult ExtensionNotificationObserver::CheckNotifications(
77 chrome::NotificationType t1,
78 chrome::NotificationType t2,
79 chrome::NotificationType t3,
80 chrome::NotificationType t4,
81 chrome::NotificationType t5,
82 chrome::NotificationType t6) {
83 std::vector<chrome::NotificationType> types;
84 types.push_back(t1);
85 types.push_back(t2);
86 types.push_back(t3);
87 types.push_back(t4);
88 types.push_back(t5);
89 types.push_back(t6);
90 return CheckNotifications(types);
91 }
92
93 // content::NotificationObserver implementation.
Observe(int type,const content::NotificationSource & source,const content::NotificationDetails & details)94 void ExtensionNotificationObserver::Observe(
95 int type,
96 const content::NotificationSource& source,
97 const content::NotificationDetails& details) {
98 switch (type) {
99 case chrome::NOTIFICATION_EXTENSION_INSTALLED_DEPRECATED: {
100 const Extension* extension =
101 content::Details<const InstalledExtensionInfo>(details)->extension;
102 if (extension_ids_.count(extension->id()))
103 notifications_.push_back(static_cast<chrome::NotificationType>(type));
104 break;
105 }
106
107 case chrome::NOTIFICATION_EXTENSION_LOADED_DEPRECATED: {
108 const Extension* extension =
109 content::Details<const Extension>(details).ptr();
110 if (extension_ids_.count(extension->id()))
111 notifications_.push_back(static_cast<chrome::NotificationType>(type));
112 break;
113 }
114
115 case chrome::NOTIFICATION_EXTENSION_UNLOADED_DEPRECATED: {
116 UnloadedExtensionInfo* reason =
117 content::Details<UnloadedExtensionInfo>(details).ptr();
118 if (extension_ids_.count(reason->extension->id())) {
119 notifications_.push_back(static_cast<chrome::NotificationType>(type));
120 // The only way that extensions are unloaded in these tests is
121 // by blacklisting.
122 EXPECT_EQ(UnloadedExtensionInfo::REASON_BLACKLIST,
123 reason->reason);
124 }
125 break;
126 }
127
128 default:
129 NOTREACHED();
130 break;
131 }
132 }
133
CheckNotifications(const std::vector<chrome::NotificationType> & types)134 testing::AssertionResult ExtensionNotificationObserver::CheckNotifications(
135 const std::vector<chrome::NotificationType>& types) {
136 testing::AssertionResult result = (notifications_ == types) ?
137 testing::AssertionSuccess() :
138 testing::AssertionFailure() << "Expected " << Str(types) << ", " <<
139 "Got " << Str(notifications_);
140 notifications_.clear();
141 return result;
142 }
143
144 } // namespace extensions
145