extension_action_unittest.cc revision ca12bfac764ba476d6cd062bf1dde12cc64c3f40 
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.
4
5#include "base/message_loop/message_loop.h"
6#include "chrome/browser/extensions/extension_action.h"
7#include "chrome/common/extensions/api/extension_action/action_info.h"
8#include "testing/gtest/include/gtest/gtest.h"
9#include "url/gurl.h"
10
11namespace {
12
13using extensions::ActionInfo;
14
15TEST(ExtensionActionTest, Title) {
16  ActionInfo action_info;
17  action_info.default_title = "Initial Title";
18  ExtensionAction action(std::string(), ActionInfo::TYPE_PAGE, action_info);
19
20  ASSERT_EQ("Initial Title", action.GetTitle(1));
21  action.SetTitle(ExtensionAction::kDefaultTabId, "foo");
22  ASSERT_EQ("foo", action.GetTitle(1));
23  ASSERT_EQ("foo", action.GetTitle(100));
24  action.SetTitle(100, "bar");
25  ASSERT_EQ("foo", action.GetTitle(1));
26  ASSERT_EQ("bar", action.GetTitle(100));
27  action.SetTitle(ExtensionAction::kDefaultTabId, "baz");
28  ASSERT_EQ("baz", action.GetTitle(1));
29  action.ClearAllValuesForTab(100);
30  ASSERT_EQ("baz", action.GetTitle(100));
31}
32
33TEST(ExtensionActionTest, Visibility) {
34  ExtensionAction action(std::string(), ActionInfo::TYPE_PAGE, ActionInfo());
35
36  ASSERT_FALSE(action.GetIsVisible(1));
37  action.SetAppearance(ExtensionAction::kDefaultTabId, ExtensionAction::ACTIVE);
38  ASSERT_TRUE(action.GetIsVisible(1));
39  ASSERT_TRUE(action.GetIsVisible(100));
40
41  action.SetAppearance(ExtensionAction::kDefaultTabId,
42                       ExtensionAction::INVISIBLE);
43  ASSERT_FALSE(action.GetIsVisible(1));
44  ASSERT_FALSE(action.GetIsVisible(100));
45  action.SetAppearance(100, ExtensionAction::ACTIVE);
46  ASSERT_FALSE(action.GetIsVisible(1));
47  ASSERT_TRUE(action.GetIsVisible(100));
48  EXPECT_FALSE(action.GetIconAnimation(100))
49      << "Page actions should not animate.";
50
51  action.ClearAllValuesForTab(100);
52  ASSERT_FALSE(action.GetIsVisible(1));
53  ASSERT_FALSE(action.GetIsVisible(100));
54
55  ExtensionAction browser_action(
56      std::string(), ActionInfo::TYPE_BROWSER, ActionInfo());
57  ASSERT_TRUE(browser_action.GetIsVisible(1));
58}
59
60TEST(ExtensionActionTest, ScriptBadgeAnimation) {
61  // Supports the icon animation.
62  base::MessageLoop message_loop;
63
64  ExtensionAction script_badge(
65      std::string(), ActionInfo::TYPE_SCRIPT_BADGE, ActionInfo());
66  EXPECT_FALSE(script_badge.GetIconAnimation(ExtensionAction::kDefaultTabId));
67  script_badge.SetAppearance(ExtensionAction::kDefaultTabId,
68                             ExtensionAction::ACTIVE);
69  EXPECT_FALSE(script_badge.GetIconAnimation(ExtensionAction::kDefaultTabId))
70      << "Showing the default tab should not animate script badges.";
71
72  script_badge.SetAppearance(ExtensionAction::kDefaultTabId,
73                             ExtensionAction::INVISIBLE);
74  EXPECT_FALSE(script_badge.GetIconAnimation(1))
75      << "Making a script badge invisible should not show its animation.";
76  script_badge.SetAppearance(1, ExtensionAction::ACTIVE);
77  EXPECT_TRUE(script_badge.GetIconAnimation(1))
78      << "Making a script badge visible should show its animation.";
79
80  script_badge.ClearAllValuesForTab(1);
81  EXPECT_FALSE(script_badge.GetIconAnimation(100));
82}
83
84TEST(ExtensionActionTest, GetAttention) {
85  // Supports the icon animation.
86  scoped_ptr<base::MessageLoop> message_loop(new base::MessageLoop);
87
88  ExtensionAction script_badge(
89      std::string(), ActionInfo::TYPE_SCRIPT_BADGE, ActionInfo());
90  EXPECT_FALSE(script_badge.GetIsVisible(1));
91  EXPECT_FALSE(script_badge.GetIconAnimation(1));
92  script_badge.SetAppearance(1, ExtensionAction::WANTS_ATTENTION);
93  EXPECT_TRUE(script_badge.GetIsVisible(1));
94  EXPECT_TRUE(script_badge.GetIconAnimation(1));
95
96  // Simulate waiting long enough for the animation to end.
97  message_loop.reset();  // Can't have 2 MessageLoops alive at once.
98  message_loop.reset(new base::MessageLoop);
99  EXPECT_FALSE(script_badge.GetIconAnimation(1));  // Sanity check.
100
101  script_badge.SetAppearance(1, ExtensionAction::ACTIVE);
102  EXPECT_FALSE(script_badge.GetIconAnimation(1))
103      << "The animation should not play again if the icon was already visible.";
104}
105
106TEST(ExtensionActionTest, Icon) {
107  ActionInfo action_info;
108  action_info.default_icon.Add(16, "icon16.png");
109  ExtensionAction page_action(
110      std::string(), ActionInfo::TYPE_PAGE, action_info);
111  ASSERT_TRUE(page_action.default_icon());
112  EXPECT_EQ("icon16.png",
113            page_action.default_icon()->Get(
114                16, ExtensionIconSet::MATCH_EXACTLY));
115  EXPECT_EQ("",
116            page_action.default_icon()->Get(
117                17, ExtensionIconSet::MATCH_BIGGER));
118}
119
120TEST(ExtensionActionTest, Badge) {
121  ExtensionAction action(std::string(), ActionInfo::TYPE_PAGE, ActionInfo());
122  ASSERT_EQ("", action.GetBadgeText(1));
123  action.SetBadgeText(ExtensionAction::kDefaultTabId, "foo");
124  ASSERT_EQ("foo", action.GetBadgeText(1));
125  ASSERT_EQ("foo", action.GetBadgeText(100));
126  action.SetBadgeText(100, "bar");
127  ASSERT_EQ("foo", action.GetBadgeText(1));
128  ASSERT_EQ("bar", action.GetBadgeText(100));
129  action.SetBadgeText(ExtensionAction::kDefaultTabId, "baz");
130  ASSERT_EQ("baz", action.GetBadgeText(1));
131  action.ClearAllValuesForTab(100);
132  ASSERT_EQ("baz", action.GetBadgeText(100));
133}
134
135TEST(ExtensionActionTest, BadgeTextColor) {
136  ExtensionAction action(std::string(), ActionInfo::TYPE_PAGE, ActionInfo());
137  ASSERT_EQ(0x00000000u, action.GetBadgeTextColor(1));
138  action.SetBadgeTextColor(ExtensionAction::kDefaultTabId, 0xFFFF0000u);
139  ASSERT_EQ(0xFFFF0000u, action.GetBadgeTextColor(1));
140  ASSERT_EQ(0xFFFF0000u, action.GetBadgeTextColor(100));
141  action.SetBadgeTextColor(100, 0xFF00FF00);
142  ASSERT_EQ(0xFFFF0000u, action.GetBadgeTextColor(1));
143  ASSERT_EQ(0xFF00FF00u, action.GetBadgeTextColor(100));
144  action.SetBadgeTextColor(ExtensionAction::kDefaultTabId, 0xFF0000FFu);
145  ASSERT_EQ(0xFF0000FFu, action.GetBadgeTextColor(1));
146  action.ClearAllValuesForTab(100);
147  ASSERT_EQ(0xFF0000FFu, action.GetBadgeTextColor(100));
148}
149
150TEST(ExtensionActionTest, BadgeBackgroundColor) {
151  ExtensionAction action(std::string(), ActionInfo::TYPE_PAGE, ActionInfo());
152  ASSERT_EQ(0x00000000u, action.GetBadgeBackgroundColor(1));
153  action.SetBadgeBackgroundColor(ExtensionAction::kDefaultTabId,
154                                 0xFFFF0000u);
155  ASSERT_EQ(0xFFFF0000u, action.GetBadgeBackgroundColor(1));
156  ASSERT_EQ(0xFFFF0000u, action.GetBadgeBackgroundColor(100));
157  action.SetBadgeBackgroundColor(100, 0xFF00FF00);
158  ASSERT_EQ(0xFFFF0000u, action.GetBadgeBackgroundColor(1));
159  ASSERT_EQ(0xFF00FF00u, action.GetBadgeBackgroundColor(100));
160  action.SetBadgeBackgroundColor(ExtensionAction::kDefaultTabId,
161                                 0xFF0000FFu);
162  ASSERT_EQ(0xFF0000FFu, action.GetBadgeBackgroundColor(1));
163  action.ClearAllValuesForTab(100);
164  ASSERT_EQ(0xFF0000FFu, action.GetBadgeBackgroundColor(100));
165}
166
167TEST(ExtensionActionTest, PopupUrl) {
168  GURL url_unset;
169  GURL url_foo("http://www.example.com/foo.html");
170  GURL url_bar("http://www.example.com/bar.html");
171  GURL url_baz("http://www.example.com/baz.html");
172
173  ActionInfo action_info;
174  action_info.default_popup_url = url_foo;
175  ExtensionAction action(std::string(), ActionInfo::TYPE_PAGE, action_info);
176
177  ASSERT_EQ(url_foo, action.GetPopupUrl(1));
178  ASSERT_EQ(url_foo, action.GetPopupUrl(100));
179  ASSERT_TRUE(action.HasPopup(1));
180  ASSERT_TRUE(action.HasPopup(100));
181
182  action.SetPopupUrl(ExtensionAction::kDefaultTabId, url_unset);
183  ASSERT_EQ(url_unset, action.GetPopupUrl(1));
184  ASSERT_EQ(url_unset, action.GetPopupUrl(100));
185  ASSERT_FALSE(action.HasPopup(1));
186  ASSERT_FALSE(action.HasPopup(100));
187
188  action.SetPopupUrl(100, url_bar);
189  ASSERT_EQ(url_unset, action.GetPopupUrl(1));
190  ASSERT_EQ(url_bar, action.GetPopupUrl(100));
191
192  action.SetPopupUrl(ExtensionAction::kDefaultTabId, url_baz);
193  ASSERT_EQ(url_baz, action.GetPopupUrl(1));
194  ASSERT_EQ(url_bar, action.GetPopupUrl(100));
195
196  action.ClearAllValuesForTab(100);
197  ASSERT_EQ(url_baz, action.GetPopupUrl(1));
198  ASSERT_EQ(url_baz, action.GetPopupUrl(100));
199}
200
201}  // namespace
202