| Index: chrome/browser/extensions/extension_message_bubble_controller_unittest.cc
|
| diff --git a/chrome/browser/extensions/extension_message_bubble_controller_unittest.cc b/chrome/browser/extensions/extension_message_bubble_controller_unittest.cc
|
| index 6eed0455145d64c8ed51b0a7231b4af14e26868b..b33c31ba1dbd74ea3648a66cbffc5bb3511d711c 100644
|
| --- a/chrome/browser/extensions/extension_message_bubble_controller_unittest.cc
|
| +++ b/chrome/browser/extensions/extension_message_bubble_controller_unittest.cc
|
| @@ -3,23 +3,31 @@
|
| // found in the LICENSE file.
|
|
|
| #include "base/command_line.h"
|
| +#include "base/strings/string_number_conversions.h"
|
| #include "base/strings/string_util.h"
|
| #include "base/strings/utf_string_conversions.h"
|
| +#include "base/values.h"
|
| #include "chrome/browser/extensions/dev_mode_bubble_controller.h"
|
| #include "chrome/browser/extensions/extension_function_test_utils.h"
|
| #include "chrome/browser/extensions/extension_message_bubble.h"
|
| #include "chrome/browser/extensions/extension_service.h"
|
| #include "chrome/browser/extensions/ntp_overridden_bubble_controller.h"
|
| +#include "chrome/browser/extensions/proxy_overridden_bubble_controller.h"
|
| #include "chrome/browser/extensions/settings_api_bubble_controller.h"
|
| #include "chrome/browser/extensions/suspicious_extension_bubble_controller.h"
|
| #include "chrome/browser/extensions/test_extension_system.h"
|
| #include "chrome/common/chrome_version_info.h"
|
| +#include "chrome/common/pref_names.h"
|
| #include "chrome/test/base/testing_profile.h"
|
| #include "content/public/test/test_browser_thread_bundle.h"
|
| +#include "extensions/browser/extension_pref_value_map.h"
|
| +#include "extensions/browser/extension_pref_value_map_factory.h"
|
| #include "extensions/browser/extension_prefs.h"
|
| +#include "extensions/browser/extension_registry.h"
|
| #include "extensions/common/extension.h"
|
| #include "extensions/common/extension_builder.h"
|
| #include "extensions/common/feature_switch.h"
|
| +#include "extensions/common/value_builder.h"
|
|
|
| namespace {
|
|
|
| @@ -159,6 +167,31 @@ class TestNtpOverriddenBubbleController
|
| }
|
| };
|
|
|
| +// A test class for the ProxyOverriddenBubbleController.
|
| +class TestProxyOverriddenBubbleController
|
| + : public ProxyOverriddenBubbleController,
|
| + public TestDelegate {
|
| + public:
|
| + explicit TestProxyOverriddenBubbleController(Profile* profile)
|
| + : ProxyOverriddenBubbleController(profile) {
|
| + }
|
| +
|
| + virtual void OnBubbleAction() OVERRIDE {
|
| + ++action_button_callback_count_;
|
| + ProxyOverriddenBubbleController::OnBubbleAction();
|
| + }
|
| +
|
| + virtual void OnBubbleDismiss() OVERRIDE {
|
| + ++dismiss_button_callback_count_;
|
| + ProxyOverriddenBubbleController::OnBubbleDismiss();
|
| + }
|
| +
|
| + virtual void OnLinkClicked() OVERRIDE {
|
| + ++link_click_callback_count_;
|
| + ProxyOverriddenBubbleController::OnLinkClicked();
|
| + }
|
| +};
|
| +
|
| // A fake bubble used for testing the controller. Takes an action that specifies
|
| // what should happen when the bubble is "shown" (the bubble is actually not
|
| // shown, the corresponding action is taken immediately).
|
| @@ -209,84 +242,146 @@ class ExtensionMessageBubbleTest : public testing::Test {
|
| public:
|
| ExtensionMessageBubbleTest() {}
|
|
|
| - void LoadGenericExtension(const std::string& index,
|
| - const std::string& id,
|
| - Manifest::Location location) {
|
| - extensions::ExtensionBuilder builder;
|
| - builder.SetManifest(extensions::DictionaryBuilder()
|
| + testing::AssertionResult LoadGenericExtension(const std::string& index,
|
| + const std::string& id,
|
| + Manifest::Location location) {
|
| + ExtensionBuilder builder;
|
| + builder.SetManifest(DictionaryBuilder()
|
| .Set("name", std::string("Extension " + index))
|
| .Set("version", "1.0")
|
| .Set("manifest_version", 2));
|
| builder.SetLocation(location);
|
| builder.SetID(id);
|
| service_->AddExtension(builder.Build().get());
|
| +
|
| + if (ExtensionRegistry::Get(profile())->enabled_extensions().GetByID(id))
|
| + return testing::AssertionSuccess();
|
| + return testing::AssertionFailure() << "Could not install extension: " << id;
|
| }
|
|
|
| - void LoadExtensionWithAction(const std::string& index,
|
| - const std::string& id,
|
| - Manifest::Location location) {
|
| - extensions::ExtensionBuilder builder;
|
| - builder.SetManifest(extensions::DictionaryBuilder()
|
| + testing::AssertionResult LoadExtensionWithAction(
|
| + const std::string& index,
|
| + const std::string& id,
|
| + Manifest::Location location) {
|
| + ExtensionBuilder builder;
|
| + builder.SetManifest(DictionaryBuilder()
|
| .Set("name", std::string("Extension " + index))
|
| .Set("version", "1.0")
|
| .Set("manifest_version", 2)
|
| .Set("browser_action",
|
| - extensions::DictionaryBuilder().Set(
|
| + DictionaryBuilder().Set(
|
| "default_title", "Default title")));
|
| builder.SetLocation(location);
|
| builder.SetID(id);
|
| service_->AddExtension(builder.Build().get());
|
| +
|
| + if (ExtensionRegistry::Get(profile())->enabled_extensions().GetByID(id))
|
| + return testing::AssertionSuccess();
|
| + return testing::AssertionFailure() << "Could not install extension: " << id;
|
| }
|
|
|
| - void LoadExtensionOverridingHome(const std::string& index,
|
| - const std::string& id,
|
| - Manifest::Location location) {
|
| - extensions::ExtensionBuilder builder;
|
| - builder.SetManifest(extensions::DictionaryBuilder()
|
| + testing::AssertionResult LoadExtensionOverridingHome(
|
| + const std::string& index,
|
| + const std::string& id,
|
| + Manifest::Location location) {
|
| + ExtensionBuilder builder;
|
| + builder.SetManifest(DictionaryBuilder()
|
| .Set("name", std::string("Extension " + index))
|
| .Set("version", "1.0")
|
| .Set("manifest_version", 2)
|
| .Set("chrome_settings_overrides",
|
| - extensions::DictionaryBuilder().Set(
|
| + DictionaryBuilder().Set(
|
| "homepage", "http://www.google.com")));
|
| builder.SetLocation(location);
|
| builder.SetID(id);
|
| service_->AddExtension(builder.Build().get());
|
| +
|
| + if (ExtensionRegistry::Get(profile())->enabled_extensions().GetByID(id))
|
| + return testing::AssertionSuccess();
|
| + return testing::AssertionFailure() << "Could not install extension: " << id;
|
| }
|
|
|
| - void LoadExtensionOverridingStart(const std::string& index,
|
| - const std::string& id,
|
| - Manifest::Location location) {
|
| - extensions::ExtensionBuilder builder;
|
| - builder.SetManifest(extensions::DictionaryBuilder()
|
| + testing::AssertionResult LoadExtensionOverridingStart(
|
| + const std::string& index,
|
| + const std::string& id,
|
| + Manifest::Location location) {
|
| + ExtensionBuilder builder;
|
| + builder.SetManifest(DictionaryBuilder()
|
| .Set("name", std::string("Extension " + index))
|
| .Set("version", "1.0")
|
| .Set("manifest_version", 2)
|
| .Set("chrome_settings_overrides",
|
| - extensions::DictionaryBuilder().Set(
|
| + DictionaryBuilder().Set(
|
| "startup_pages",
|
| - extensions::ListBuilder().Append(
|
| + ListBuilder().Append(
|
| "http://www.google.com"))));
|
| builder.SetLocation(location);
|
| builder.SetID(id);
|
| service_->AddExtension(builder.Build().get());
|
| +
|
| + if (ExtensionRegistry::Get(profile())->enabled_extensions().GetByID(id))
|
| + return testing::AssertionSuccess();
|
| + return testing::AssertionFailure() << "Could not install extension: " << id;
|
| }
|
|
|
| - void LoadExtensionOverridingNtp(const std::string& index,
|
| - const std::string& id,
|
| - Manifest::Location location) {
|
| - extensions::ExtensionBuilder builder;
|
| - builder.SetManifest(extensions::DictionaryBuilder()
|
| + testing::AssertionResult LoadExtensionOverridingNtp(
|
| + const std::string& index,
|
| + const std::string& id,
|
| + Manifest::Location location) {
|
| + ExtensionBuilder builder;
|
| + builder.SetManifest(DictionaryBuilder()
|
| .Set("name", std::string("Extension " + index))
|
| .Set("version", "1.0")
|
| .Set("manifest_version", 2)
|
| .Set("chrome_url_overrides",
|
| - extensions::DictionaryBuilder().Set(
|
| + DictionaryBuilder().Set(
|
| "newtab", "Default.html")));
|
|
|
| builder.SetLocation(location);
|
| builder.SetID(id);
|
| service_->AddExtension(builder.Build().get());
|
| +
|
| + if (ExtensionRegistry::Get(profile())->enabled_extensions().GetByID(id))
|
| + return testing::AssertionSuccess();
|
| + return testing::AssertionFailure() << "Could not install extension: " << id;
|
| + }
|
| +
|
| + testing::AssertionResult LoadExtensionOverridingProxy(
|
| + const std::string& index,
|
| + const std::string& id,
|
| + Manifest::Location location) {
|
| + ExtensionBuilder builder;
|
| + builder.SetManifest(DictionaryBuilder()
|
| + .Set("name", std::string("Extension " + index))
|
| + .Set("version", "1.0")
|
| + .Set("manifest_version", 2)
|
| + .Set("permissions",
|
| + ListBuilder().Append("proxy")));
|
| +
|
| + builder.SetLocation(location);
|
| + builder.SetID(id);
|
| + service_->AddExtension(builder.Build().get());
|
| +
|
| + // The proxy check relies on ExtensionPrefValueMap being up to date as to
|
| + // specifying which extension is controlling the proxy, but unfortunately
|
| + // that Map is not updated automatically for unit tests, so we simulate the
|
| + // update here to avoid test failures.
|
| + ExtensionPrefValueMap* extension_prefs_value_map =
|
| + ExtensionPrefValueMapFactory::GetForBrowserContext(profile());
|
| + extension_prefs_value_map->RegisterExtension(
|
| + id,
|
| + base::Time::Now(),
|
| + true, // is_enabled.
|
| + false); // is_incognito_enabled.
|
| + extension_prefs_value_map->SetExtensionPref(
|
| + id,
|
| + prefs::kProxy,
|
| + kExtensionPrefsScopeRegular,
|
| + base::Value::CreateStringValue(id));
|
| +
|
| + if (ExtensionRegistry::Get(profile())->enabled_extensions().GetByID(id))
|
| + return testing::AssertionSuccess();
|
| + return testing::AssertionFailure() << "Could not install extension: " << id;
|
| }
|
|
|
| void Init() {
|
| @@ -348,9 +443,9 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_WipeoutControllerTest) {
|
| Init();
|
| // Add three extensions, and control two of them in this test (extension 1
|
| // and 2).
|
| - LoadExtensionWithAction("1", kId1, Manifest::COMMAND_LINE);
|
| - LoadGenericExtension("2", kId2, Manifest::UNPACKED);
|
| - LoadGenericExtension("3", kId3, Manifest::EXTERNAL_POLICY);
|
| + ASSERT_TRUE(LoadExtensionWithAction("1", kId1, Manifest::COMMAND_LINE));
|
| + ASSERT_TRUE(LoadGenericExtension("2", kId2, Manifest::UNPACKED));
|
| + ASSERT_TRUE(LoadGenericExtension("3", kId3, Manifest::EXTERNAL_POLICY));
|
|
|
| scoped_ptr<TestSuspiciousExtensionBubbleController> controller(
|
| new TestSuspiciousExtensionBubbleController(profile()));
|
| @@ -425,9 +520,9 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_DevModeControllerTest) {
|
| // Add three extensions, and control two of them in this test (extension 1
|
| // and 2). Extension 1 is a regular extension, Extension 2 is UNPACKED so it
|
| // counts as a DevMode extension.
|
| - LoadExtensionWithAction("1", kId1, Manifest::COMMAND_LINE);
|
| - LoadGenericExtension("2", kId2, Manifest::UNPACKED);
|
| - LoadGenericExtension("3", kId3, Manifest::EXTERNAL_POLICY);
|
| + ASSERT_TRUE(LoadExtensionWithAction("1", kId1, Manifest::COMMAND_LINE));
|
| + ASSERT_TRUE(LoadGenericExtension("2", kId2, Manifest::UNPACKED));
|
| + ASSERT_TRUE(LoadGenericExtension("3", kId3, Manifest::EXTERNAL_POLICY));
|
|
|
| scoped_ptr<TestDevModeBubbleController> controller(
|
| new TestDevModeBubbleController(profile()));
|
| @@ -451,8 +546,9 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_DevModeControllerTest) {
|
| EXPECT_EQ(0U, controller->link_click_count());
|
| EXPECT_EQ(0U, controller->action_click_count());
|
| EXPECT_EQ(1U, controller->dismiss_click_count());
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) != NULL);
|
| + ExtensionRegistry* registry = ExtensionRegistry::Get(profile());
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId2) != NULL);
|
|
|
| // Do it again, but now press different button (Disable).
|
| bubble.set_action_on_show(
|
| @@ -467,8 +563,8 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_DevModeControllerTest) {
|
| EXPECT_EQ(0U, controller->link_click_count());
|
| EXPECT_EQ(1U, controller->action_click_count());
|
| EXPECT_EQ(0U, controller->dismiss_click_count());
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) == NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) == NULL);
|
| + EXPECT_TRUE(registry->disabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->disabled_extensions().GetByID(kId2) != NULL);
|
|
|
| // Re-enable the extensions (disabled by the action button above).
|
| service_->EnableExtension(kId1);
|
| @@ -487,8 +583,8 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_DevModeControllerTest) {
|
| EXPECT_EQ(1U, controller->link_click_count());
|
| EXPECT_EQ(0U, controller->action_click_count());
|
| EXPECT_EQ(0U, controller->dismiss_click_count());
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId2) != NULL);
|
|
|
| // Now disable the unpacked extension.
|
| service_->DisableExtension(kId1, Extension::DISABLE_USER_ACTION);
|
| @@ -511,8 +607,7 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_DevModeControllerTest) {
|
|
|
| TEST_F(ExtensionMessageBubbleTest, MAYBE_SettingsApiControllerTest) {
|
| Init();
|
| - extensions::ExtensionPrefs* prefs =
|
| - extensions::ExtensionPrefs::Get(profile());
|
| + ExtensionPrefs* prefs = ExtensionPrefs::Get(profile());
|
|
|
| for (int i = 0; i < 3; ++i) {
|
| switch (static_cast<SettingsApiOverrideType>(i)) {
|
| @@ -520,9 +615,10 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_SettingsApiControllerTest) {
|
| // Load two extensions overriding home page and one overriding something
|
| // unrelated (to check for interference). Extension 2 should still win
|
| // on the home page setting.
|
| - LoadExtensionOverridingHome("1", kId1, Manifest::UNPACKED);
|
| - LoadExtensionOverridingHome("2", kId2, Manifest::UNPACKED);
|
| - LoadExtensionOverridingStart("3", kId3, Manifest::UNPACKED);
|
| + ASSERT_TRUE(LoadExtensionOverridingHome("1", kId1, Manifest::UNPACKED));
|
| + ASSERT_TRUE(LoadExtensionOverridingHome("2", kId2, Manifest::UNPACKED));
|
| + ASSERT_TRUE(
|
| + LoadExtensionOverridingStart("3", kId3, Manifest::UNPACKED));
|
| break;
|
| case BUBBLE_TYPE_SEARCH_ENGINE:
|
| // We deliberately skip testing the search engine since it relies on
|
| @@ -535,9 +631,11 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_SettingsApiControllerTest) {
|
| // Load two extensions overriding start page and one overriding
|
| // something unrelated (to check for interference). Extension 2 should
|
| // still win on the startup page setting.
|
| - LoadExtensionOverridingStart("1", kId1, Manifest::UNPACKED);
|
| - LoadExtensionOverridingStart("2", kId2, Manifest::UNPACKED);
|
| - LoadExtensionOverridingHome("3", kId3, Manifest::UNPACKED);
|
| + ASSERT_TRUE(
|
| + LoadExtensionOverridingStart("1", kId1, Manifest::UNPACKED));
|
| + ASSERT_TRUE(
|
| + LoadExtensionOverridingStart("2", kId2, Manifest::UNPACKED));
|
| + ASSERT_TRUE(LoadExtensionOverridingHome("3", kId3, Manifest::UNPACKED));
|
| break;
|
| default:
|
| NOTREACHED();
|
| @@ -568,9 +666,10 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_SettingsApiControllerTest) {
|
| EXPECT_EQ(0U, controller->action_click_count());
|
| EXPECT_EQ(1U, controller->dismiss_click_count());
|
| // No extension should have become disabled.
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId3, false) != NULL);
|
| + ExtensionRegistry* registry = ExtensionRegistry::Get(profile());
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| // Only extension 2 should have been acknowledged.
|
| EXPECT_FALSE(prefs->HasSettingsApiBubbleBeenAcknowledged(kId1));
|
| EXPECT_TRUE(prefs->HasSettingsApiBubbleBeenAcknowledged(kId2));
|
| @@ -588,9 +687,9 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_SettingsApiControllerTest) {
|
| EXPECT_EQ(0U, controller->action_click_count());
|
| EXPECT_EQ(0U, controller->dismiss_click_count());
|
| // No extension should have become disabled.
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId3, false) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| // Only extension 2 should have been acknowledged.
|
| EXPECT_FALSE(prefs->HasSettingsApiBubbleBeenAcknowledged(kId1));
|
| EXPECT_TRUE(prefs->HasSettingsApiBubbleBeenAcknowledged(kId2));
|
| @@ -611,9 +710,9 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_SettingsApiControllerTest) {
|
| EXPECT_EQ(1U, controller->action_click_count());
|
| EXPECT_EQ(0U, controller->dismiss_click_count());
|
| // Only extension 2 should have become disabled.
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) == NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId3, false) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->disabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| // No extension should have been acknowledged (it got disabled).
|
| EXPECT_FALSE(prefs->HasSettingsApiBubbleBeenAcknowledged(kId1));
|
| EXPECT_FALSE(prefs->HasSettingsApiBubbleBeenAcknowledged(kId2));
|
| @@ -635,14 +734,13 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_SettingsApiControllerTest) {
|
|
|
| TEST_F(ExtensionMessageBubbleTest, MAYBE_NtpOverriddenControllerTest) {
|
| Init();
|
| - extensions::ExtensionPrefs* prefs =
|
| - extensions::ExtensionPrefs::Get(profile());
|
| + ExtensionPrefs* prefs = ExtensionPrefs::Get(profile());
|
| // Load two extensions overriding new tab page and one overriding something
|
| // unrelated (to check for interference). Extension 2 should still win
|
| // on the new tab page setting.
|
| - LoadExtensionOverridingNtp("1", kId1, Manifest::UNPACKED);
|
| - LoadExtensionOverridingNtp("2", kId2, Manifest::UNPACKED);
|
| - LoadExtensionOverridingStart("3", kId3, Manifest::UNPACKED);
|
| + ASSERT_TRUE(LoadExtensionOverridingNtp("1", kId1, Manifest::UNPACKED));
|
| + ASSERT_TRUE(LoadExtensionOverridingNtp("2", kId2, Manifest::UNPACKED));
|
| + ASSERT_TRUE(LoadExtensionOverridingStart("3", kId3, Manifest::UNPACKED));
|
|
|
| scoped_ptr<TestNtpOverriddenBubbleController> controller(
|
| new TestNtpOverriddenBubbleController(profile()));
|
| @@ -668,9 +766,10 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_NtpOverriddenControllerTest) {
|
| EXPECT_EQ(0U, controller->action_click_count());
|
| EXPECT_EQ(1U, controller->dismiss_click_count());
|
| // No extension should have become disabled.
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId3, false) != NULL);
|
| + ExtensionRegistry* registry = ExtensionRegistry::Get(profile());
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| // Only extension 2 should have been acknowledged.
|
| EXPECT_FALSE(prefs->HasNtpOverriddenBubbleBeenAcknowledged(kId1));
|
| EXPECT_TRUE(prefs->HasNtpOverriddenBubbleBeenAcknowledged(kId2));
|
| @@ -688,9 +787,9 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_NtpOverriddenControllerTest) {
|
| EXPECT_EQ(0U, controller->action_click_count());
|
| EXPECT_EQ(0U, controller->dismiss_click_count());
|
| // No extension should have become disabled.
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId3, false) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| // Only extension 2 should have been acknowledged.
|
| EXPECT_FALSE(prefs->HasNtpOverriddenBubbleBeenAcknowledged(kId1));
|
| EXPECT_TRUE(prefs->HasNtpOverriddenBubbleBeenAcknowledged(kId2));
|
| @@ -710,9 +809,9 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_NtpOverriddenControllerTest) {
|
| EXPECT_EQ(1U, controller->action_click_count());
|
| EXPECT_EQ(0U, controller->dismiss_click_count());
|
| // Only extension 2 should have become disabled.
|
| - EXPECT_TRUE(service_->GetExtensionById(kId1, false) != NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId2, false) == NULL);
|
| - EXPECT_TRUE(service_->GetExtensionById(kId3, false) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->disabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| // No extension should have been acknowledged (it got disabled).
|
| EXPECT_FALSE(prefs->HasNtpOverriddenBubbleBeenAcknowledged(kId1));
|
| EXPECT_FALSE(prefs->HasNtpOverriddenBubbleBeenAcknowledged(kId2));
|
| @@ -724,4 +823,124 @@ TEST_F(ExtensionMessageBubbleTest, MAYBE_NtpOverriddenControllerTest) {
|
| service_->UninstallExtension(kId3, false, NULL);
|
| }
|
|
|
| +void SetInstallTime(const std::string& extension_id,
|
| + const base::Time& time,
|
| + ExtensionPrefs* prefs) {
|
| + std::string time_str = base::Int64ToString(time.ToInternalValue());
|
| + prefs->UpdateExtensionPref(extension_id,
|
| + "install_time",
|
| + new base::StringValue(time_str));
|
| +}
|
| +
|
| +// The feature this is meant to test is only implemented on Windows.
|
| +#if defined(OS_WIN)
|
| +#define MAYBE_ProxyOverriddenControllerTest ProxyOverriddenControllerTest
|
| +#else
|
| +#define MAYBE_ProxyOverriddenControllerTest DISABLED_ProxyOverriddenControllerTest
|
| +#endif
|
| +
|
| +TEST_F(ExtensionMessageBubbleTest, MAYBE_ProxyOverriddenControllerTest) {
|
| + Init();
|
| + ExtensionPrefs* prefs = ExtensionPrefs::Get(profile());
|
| + // Load two extensions overriding proxy and one overriding something
|
| + // unrelated (to check for interference). Extension 2 should still win
|
| + // on the proxy setting.
|
| + ASSERT_TRUE(LoadExtensionOverridingProxy("1", kId1, Manifest::UNPACKED));
|
| + ASSERT_TRUE(LoadExtensionOverridingProxy("2", kId2, Manifest::UNPACKED));
|
| + ASSERT_TRUE(LoadExtensionOverridingStart("3", kId3, Manifest::UNPACKED));
|
| +
|
| + // The bubble will not show if the extension was installed in the last 7 days
|
| + // so we artificially set the install time to simulate an old install during
|
| + // testing.
|
| + base::Time old_enough = base::Time::Now() - base::TimeDelta::FromDays(8);
|
| + SetInstallTime(kId1, old_enough, prefs);
|
| + SetInstallTime(kId2, base::Time::Now(), prefs);
|
| + SetInstallTime(kId3, old_enough, prefs);
|
| +
|
| + scoped_ptr<TestProxyOverriddenBubbleController> controller(
|
| + new TestProxyOverriddenBubbleController(profile()));
|
| +
|
| + // The second extension is too new to warn about.
|
| + EXPECT_FALSE(controller->ShouldShow(kId1));
|
| + EXPECT_FALSE(controller->ShouldShow(kId2));
|
| + // Lets make it old enough.
|
| + SetInstallTime(kId2, old_enough, prefs);
|
| +
|
| + // The list will contain one enabled unpacked extension (ext 2).
|
| + EXPECT_TRUE(controller->ShouldShow(kId2));
|
| + EXPECT_FALSE(controller->ShouldShow(kId3));
|
| + std::vector<base::string16> override_extensions =
|
| + controller->GetExtensionList();
|
| + ASSERT_EQ(1U, override_extensions.size());
|
| + EXPECT_EQ(base::ASCIIToUTF16("Extension 2"), override_extensions[0]);
|
| + EXPECT_EQ(0U, controller->link_click_count());
|
| + EXPECT_EQ(0U, controller->dismiss_click_count());
|
| + EXPECT_EQ(0U, controller->action_click_count());
|
| +
|
| + // Simulate showing the bubble and dismissing it.
|
| + FakeExtensionMessageBubble bubble;
|
| + bubble.set_action_on_show(
|
| + FakeExtensionMessageBubble::BUBBLE_ACTION_CLICK_DISMISS_BUTTON);
|
| + controller->Show(&bubble);
|
| + EXPECT_EQ(0U, controller->link_click_count());
|
| + EXPECT_EQ(0U, controller->action_click_count());
|
| + EXPECT_EQ(1U, controller->dismiss_click_count());
|
| + // No extension should have become disabled.
|
| + ExtensionRegistry* registry = ExtensionRegistry::Get(profile());
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| + // Only extension 2 should have been acknowledged.
|
| + EXPECT_FALSE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId1));
|
| + EXPECT_TRUE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId2));
|
| + EXPECT_FALSE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId3));
|
| + // Clean up after ourselves.
|
| + prefs->SetProxyOverriddenBubbleBeenAcknowledged(kId2, false);
|
| +
|
| + // Simulate clicking the learn more link to dismiss it.
|
| + bubble.set_action_on_show(
|
| + FakeExtensionMessageBubble::BUBBLE_ACTION_CLICK_LINK);
|
| + controller.reset(new TestProxyOverriddenBubbleController(profile()));
|
| + EXPECT_TRUE(controller->ShouldShow(kId2));
|
| + controller->Show(&bubble);
|
| + EXPECT_EQ(1U, controller->link_click_count());
|
| + EXPECT_EQ(0U, controller->action_click_count());
|
| + EXPECT_EQ(0U, controller->dismiss_click_count());
|
| + // No extension should have become disabled.
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| + // Only extension 2 should have been acknowledged.
|
| + EXPECT_FALSE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId1));
|
| + EXPECT_TRUE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId2));
|
| + EXPECT_FALSE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId3));
|
| + // Clean up after ourselves.
|
| + prefs->SetProxyOverriddenBubbleBeenAcknowledged(kId2, false);
|
| +
|
| + // Do it again, but now opt to disable the extension.
|
| + bubble.set_action_on_show(
|
| + FakeExtensionMessageBubble::BUBBLE_ACTION_CLICK_ACTION_BUTTON);
|
| + controller.reset(new TestProxyOverriddenBubbleController(profile()));
|
| + EXPECT_TRUE(controller->ShouldShow(kId2));
|
| + override_extensions = controller->GetExtensionList();
|
| + EXPECT_EQ(1U, override_extensions.size());
|
| + controller->Show(&bubble); // Simulate showing the bubble.
|
| + EXPECT_EQ(0U, controller->link_click_count());
|
| + EXPECT_EQ(1U, controller->action_click_count());
|
| + EXPECT_EQ(0U, controller->dismiss_click_count());
|
| + // Only extension 2 should have become disabled.
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId1) != NULL);
|
| + EXPECT_TRUE(registry->disabled_extensions().GetByID(kId2) != NULL);
|
| + EXPECT_TRUE(registry->enabled_extensions().GetByID(kId3) != NULL);
|
| + // No extension should have been acknowledged (it got disabled).
|
| + EXPECT_FALSE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId1));
|
| + EXPECT_FALSE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId2));
|
| + EXPECT_FALSE(prefs->HasProxyOverriddenBubbleBeenAcknowledged(kId3));
|
| +
|
| + // Clean up after ourselves.
|
| + service_->UninstallExtension(kId1, false, NULL);
|
| + service_->UninstallExtension(kId2, false, NULL);
|
| + service_->UninstallExtension(kId3, false, NULL);
|
| +}
|
| +
|
| } // namespace extensions
|
|
|