| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/common/extensions/api/extension_action/action_info.h" |
| 6 #include "chrome/common/extensions/api/extension_action/script_badge_handler.h" |
| 5 #include "chrome/common/extensions/extension_builder.h" | 7 #include "chrome/common/extensions/extension_builder.h" |
| 6 #include "chrome/common/extensions/extension_constants.h" | 8 #include "chrome/common/extensions/extension_constants.h" |
| 7 #include "chrome/common/extensions/extension_icon_set.h" | 9 #include "chrome/common/extensions/extension_icon_set.h" |
| 8 #include "chrome/common/extensions/extension_manifest_constants.h" | 10 #include "chrome/common/extensions/extension_manifest_constants.h" |
| 11 #include "chrome/common/extensions/manifest_handler.h" |
| 9 #include "chrome/common/extensions/manifest_tests/extension_manifest_test.h" | 12 #include "chrome/common/extensions/manifest_tests/extension_manifest_test.h" |
| 10 #include "testing/gmock/include/gmock/gmock.h" | 13 #include "testing/gmock/include/gmock/gmock.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 12 | 15 |
| 13 namespace errors = extension_manifest_errors; | 16 namespace errors = extension_manifest_errors; |
| 14 using extensions::DictionaryBuilder; | 17 |
| 15 using extensions::Extension; | 18 namespace extensions { |
| 16 using extensions::ExtensionBuilder; | |
| 17 | 19 |
| 18 namespace { | 20 namespace { |
| 19 | 21 |
| 20 std::vector<Extension::InstallWarning> StripMissingFlagWarning( | 22 std::vector<Extension::InstallWarning> StripMissingFlagWarning( |
| 21 const std::vector<Extension::InstallWarning>& install_warnings) { | 23 const std::vector<Extension::InstallWarning>& install_warnings) { |
| 22 std::vector<Extension::InstallWarning> result; | 24 std::vector<Extension::InstallWarning> result; |
| 23 for (size_t i = 0; i < install_warnings.size(); ++i) { | 25 for (size_t i = 0; i < install_warnings.size(); ++i) { |
| 24 if (install_warnings[i].message != errors::kScriptBadgeRequiresFlag) | 26 if (install_warnings[i].message != errors::kScriptBadgeRequiresFlag) |
| 25 result.push_back(install_warnings[i]); | 27 result.push_back(install_warnings[i]); |
| 26 } | 28 } |
| 27 return result; | 29 return result; |
| 28 } | 30 } |
| 29 | 31 |
| 30 TEST_F(ExtensionManifestTest, ScriptBadgeBasic) { | 32 class ScriptBadgeManifestTest : public ExtensionManifestTest { |
| 33 protected: |
| 34 virtual void SetUp() OVERRIDE { |
| 35 ManifestHandler::Register(extension_manifest_keys::kScriptBadge, |
| 36 new ScriptBadgeHandler); |
| 37 } |
| 38 }; |
| 39 |
| 40 } // namespace |
| 41 |
| 42 TEST_F(ScriptBadgeManifestTest, ScriptBadgeBasic) { |
| 31 scoped_refptr<Extension> extension( | 43 scoped_refptr<Extension> extension( |
| 32 ExtensionBuilder() | 44 ExtensionBuilder() |
| 33 .SetManifest(DictionaryBuilder() | 45 .SetManifest(DictionaryBuilder() |
| 34 .Set("manifest_version", 2) | 46 .Set("manifest_version", 2) |
| 35 .Set("name", "my extension") | 47 .Set("name", "my extension") |
| 36 .Set("version", "1.0.0.0") | 48 .Set("version", "1.0.0.0") |
| 37 .Set("description", | 49 .Set("description", |
| 38 "Check that a simple script_badge section parses") | 50 "Check that a simple script_badge section parses") |
| 39 .Set("icons", DictionaryBuilder() | 51 .Set("icons", DictionaryBuilder() |
| 40 .Set("16", "icon16.png") | 52 .Set("16", "icon16.png") |
| 41 .Set("32", "icon32.png") | 53 .Set("32", "icon32.png") |
| 42 .Set("19", "icon19.png") | 54 .Set("19", "icon19.png") |
| 43 .Set("48", "icon48.png")) | 55 .Set("48", "icon48.png")) |
| 44 .Set("script_badge", DictionaryBuilder() | 56 .Set("script_badge", DictionaryBuilder() |
| 45 .Set("default_popup", "popup.html"))) | 57 .Set("default_popup", "popup.html"))) |
| 46 .Build()); | 58 .Build()); |
| 47 ASSERT_TRUE(extension.get()); | 59 ASSERT_TRUE(extension.get()); |
| 48 ASSERT_TRUE(extension->script_badge_info()); | 60 const ActionInfo* script_badge_info = |
| 61 ActionInfo::GetScriptBadgeInfo(extension); |
| 62 ASSERT_TRUE(script_badge_info); |
| 49 EXPECT_THAT(StripMissingFlagWarning(extension->install_warnings()), | 63 EXPECT_THAT(StripMissingFlagWarning(extension->install_warnings()), |
| 50 testing::ElementsAre(/*empty*/)); | 64 testing::ElementsAre(/*empty*/)); |
| 51 | 65 |
| 52 const ExtensionIconSet& default_icon = | 66 const ExtensionIconSet& default_icon = |
| 53 extension->script_badge_info()->default_icon; | 67 script_badge_info->default_icon; |
| 54 // Should have a default icon set. | 68 // Should have a default icon set. |
| 55 ASSERT_FALSE(default_icon.empty()); | 69 ASSERT_FALSE(default_icon.empty()); |
| 56 | 70 |
| 57 // Verify that correct icon paths are registered in default_icon. | 71 // Verify that correct icon paths are registered in default_icon. |
| 58 EXPECT_EQ(2u, default_icon.map().size()); | 72 EXPECT_EQ(2u, default_icon.map().size()); |
| 59 EXPECT_EQ("icon16.png", | 73 EXPECT_EQ("icon16.png", |
| 60 default_icon.Get(extension_misc::EXTENSION_ICON_BITTY, | 74 default_icon.Get(extension_misc::EXTENSION_ICON_BITTY, |
| 61 ExtensionIconSet::MATCH_EXACTLY)); | 75 ExtensionIconSet::MATCH_EXACTLY)); |
| 62 EXPECT_EQ("icon32.png", | 76 EXPECT_EQ("icon32.png", |
| 63 default_icon.Get(2 * extension_misc::EXTENSION_ICON_BITTY, | 77 default_icon.Get(2 * extension_misc::EXTENSION_ICON_BITTY, |
| 64 ExtensionIconSet::MATCH_EXACTLY)); | 78 ExtensionIconSet::MATCH_EXACTLY)); |
| 65 | 79 |
| 66 EXPECT_EQ("my extension", extension->script_badge_info()->default_title); | 80 EXPECT_EQ("my extension", script_badge_info->default_title); |
| 67 EXPECT_FALSE(extension->script_badge_info()->default_popup_url.is_empty()); | 81 EXPECT_FALSE(script_badge_info->default_popup_url.is_empty()); |
| 68 } | 82 } |
| 69 | 83 |
| 70 TEST_F(ExtensionManifestTest, ScriptBadgeExplicitTitleAndIconsIgnored) { | 84 TEST_F(ScriptBadgeManifestTest, ScriptBadgeExplicitTitleAndIconsIgnored) { |
| 71 scoped_refptr<Extension> extension( | 85 scoped_refptr<Extension> extension( |
| 72 ExtensionBuilder() | 86 ExtensionBuilder() |
| 73 .SetManifest(DictionaryBuilder() | 87 .SetManifest(DictionaryBuilder() |
| 74 .Set("manifest_version", 2) | 88 .Set("manifest_version", 2) |
| 75 .Set("name", "my extension") | 89 .Set("name", "my extension") |
| 76 .Set("version", "1.0.0.0") | 90 .Set("version", "1.0.0.0") |
| 77 .Set("description", | 91 .Set("description", |
| 78 "Check that a simple script_badge section parses") | 92 "Check that a simple script_badge section parses") |
| 79 .Set("icons", DictionaryBuilder() | 93 .Set("icons", DictionaryBuilder() |
| 80 .Set("16", "icon16.png")) | 94 .Set("16", "icon16.png")) |
| 81 .Set("script_badge", DictionaryBuilder() | 95 .Set("script_badge", DictionaryBuilder() |
| 82 .Set("default_title", "Other Extension") | 96 .Set("default_title", "Other Extension") |
| 83 .Set("default_icon", "malicious.png"))) | 97 .Set("default_icon", "malicious.png"))) |
| 84 .Build()); | 98 .Build()); |
| 85 ASSERT_TRUE(extension.get()); | 99 ASSERT_TRUE(extension.get()); |
| 86 ASSERT_TRUE(extension->script_badge_info()); | 100 const ActionInfo* script_badge_info = |
| 101 ActionInfo::GetScriptBadgeInfo(extension); |
| 102 ASSERT_TRUE(script_badge_info); |
| 87 | 103 |
| 88 EXPECT_THAT(StripMissingFlagWarning(extension->install_warnings()), | 104 EXPECT_THAT(StripMissingFlagWarning(extension->install_warnings()), |
| 89 testing::ElementsAre( | 105 testing::ElementsAre( |
| 90 Extension::InstallWarning( | 106 Extension::InstallWarning( |
| 91 Extension::InstallWarning::FORMAT_TEXT, | 107 Extension::InstallWarning::FORMAT_TEXT, |
| 92 errors::kScriptBadgeTitleIgnored), | 108 errors::kScriptBadgeTitleIgnored), |
| 93 Extension::InstallWarning( | 109 Extension::InstallWarning( |
| 94 Extension::InstallWarning::FORMAT_TEXT, | 110 Extension::InstallWarning::FORMAT_TEXT, |
| 95 errors::kScriptBadgeIconIgnored))); | 111 errors::kScriptBadgeIconIgnored))); |
| 96 | 112 |
| 97 const ExtensionIconSet& default_icon = | 113 const ExtensionIconSet& default_icon = |
| 98 extension->script_badge_info()->default_icon; | 114 script_badge_info->default_icon; |
| 99 ASSERT_FALSE(default_icon.empty()); | 115 ASSERT_FALSE(default_icon.empty()); |
| 100 | 116 |
| 101 EXPECT_EQ(1u, default_icon.map().size()); | 117 EXPECT_EQ(1u, default_icon.map().size()); |
| 102 EXPECT_EQ("icon16.png", | 118 EXPECT_EQ("icon16.png", |
| 103 default_icon.Get(extension_misc::EXTENSION_ICON_BITTY, | 119 default_icon.Get(extension_misc::EXTENSION_ICON_BITTY, |
| 104 ExtensionIconSet::MATCH_EXACTLY)); | 120 ExtensionIconSet::MATCH_EXACTLY)); |
| 105 | 121 |
| 106 EXPECT_EQ("my extension", extension->script_badge_info()->default_title); | 122 EXPECT_EQ("my extension", script_badge_info->default_title); |
| 107 } | 123 } |
| 108 | 124 |
| 109 TEST_F(ExtensionManifestTest, ScriptBadgeIconFallsBackToPuzzlePiece) { | 125 TEST_F(ScriptBadgeManifestTest, ScriptBadgeIconFallsBackToPuzzlePiece) { |
| 110 scoped_refptr<Extension> extension( | 126 scoped_refptr<Extension> extension( |
| 111 ExtensionBuilder() | 127 ExtensionBuilder() |
| 112 .SetManifest(DictionaryBuilder() | 128 .SetManifest(DictionaryBuilder() |
| 113 .Set("manifest_version", 2) | 129 .Set("manifest_version", 2) |
| 114 .Set("name", "my extension") | 130 .Set("name", "my extension") |
| 115 .Set("version", "1.0.0.0") | 131 .Set("version", "1.0.0.0") |
| 116 .Set("description", | 132 .Set("description", |
| 117 "Check that a simple script_badge section parses") | 133 "Check that a simple script_badge section parses") |
| 118 .Set("icons", DictionaryBuilder() | 134 .Set("icons", DictionaryBuilder() |
| 119 .Set("128", "icon128.png"))) | 135 .Set("128", "icon128.png"))) |
| 120 .Build()); | 136 .Build()); |
| 121 ASSERT_TRUE(extension.get()); | 137 ASSERT_TRUE(extension.get()); |
| 122 ASSERT_TRUE(extension->script_badge_info()); | 138 const ActionInfo* script_badge_info = |
| 139 ActionInfo::GetScriptBadgeInfo(extension); |
| 140 ASSERT_TRUE(script_badge_info); |
| 123 EXPECT_THAT(extension->install_warnings(), | 141 EXPECT_THAT(extension->install_warnings(), |
| 124 testing::ElementsAre(/*empty*/)); | 142 testing::ElementsAre(/*empty*/)); |
| 125 | 143 |
| 126 const ExtensionIconSet& default_icon = | 144 const ExtensionIconSet& default_icon = |
| 127 extension->script_badge_info()->default_icon; | 145 script_badge_info->default_icon; |
| 128 ASSERT_FALSE(default_icon.empty()) << "Should fall back to the 128px icon."; | 146 ASSERT_FALSE(default_icon.empty()) << "Should fall back to the 128px icon."; |
| 129 EXPECT_EQ(2u, default_icon.map().size()); | 147 EXPECT_EQ(2u, default_icon.map().size()); |
| 130 EXPECT_EQ("icon128.png", | 148 EXPECT_EQ("icon128.png", |
| 131 default_icon.Get(extension_misc::EXTENSION_ICON_BITTY, | 149 default_icon.Get(extension_misc::EXTENSION_ICON_BITTY, |
| 132 ExtensionIconSet::MATCH_EXACTLY)); | 150 ExtensionIconSet::MATCH_EXACTLY)); |
| 133 EXPECT_EQ("icon128.png", | 151 EXPECT_EQ("icon128.png", |
| 134 default_icon.Get(2 * extension_misc::EXTENSION_ICON_BITTY, | 152 default_icon.Get(2 * extension_misc::EXTENSION_ICON_BITTY, |
| 135 ExtensionIconSet::MATCH_EXACTLY)); | 153 ExtensionIconSet::MATCH_EXACTLY)); |
| 136 } | 154 } |
| 137 | 155 |
| 138 } // namespace | 156 } // namespace extensions |
| OLD | NEW |