| // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/extensions/extension_browsertest.h" |
| #include "chrome/browser/extensions/extension_service.h" |
| #include "chrome/browser/extensions/extension_system.h" |
| #include "chrome/browser/extensions/extension_toolbar_model.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/test/base/in_process_browser_test.h" |
| |
| using extensions::Extension; |
| |
| // An InProcessBrowserTest for testing the ExtensionToolbarModel. |
| // TODO(erikkay) It's unfortunate that this needs to be an in-proc browser test. |
| // It would be nice to refactor things so that ExtensionService could run |
| // without so much of the browser in place. |
| class ExtensionToolbarModelTest : public ExtensionBrowserTest, |
| public ExtensionToolbarModel::Observer { |
| public: |
| virtual void SetUp() { |
| inserted_count_ = 0; |
| removed_count_ = 0; |
| moved_count_ = 0; |
| |
| ExtensionBrowserTest::SetUp(); |
| } |
| |
| virtual void SetUpOnMainThread() OVERRIDE { |
| ExtensionService* service = extensions::ExtensionSystem::Get( |
| browser()->profile())->extension_service(); |
| model_ = service->toolbar_model(); |
| model_->AddObserver(this); |
| } |
| |
| virtual void CleanUpOnMainThread() OVERRIDE { |
| model_->RemoveObserver(this); |
| } |
| |
| virtual void BrowserActionAdded(const Extension* extension, |
| int index) OVERRIDE { |
| inserted_count_++; |
| } |
| |
| virtual void BrowserActionRemoved(const Extension* extension) OVERRIDE { |
| removed_count_++; |
| } |
| |
| virtual void BrowserActionMoved(const Extension* extension, |
| int index) OVERRIDE { |
| moved_count_++; |
| } |
| |
| const Extension* ExtensionAt(int index) { |
| const extensions::ExtensionList& toolbar_items = model_->toolbar_items(); |
| for (extensions::ExtensionList::const_iterator i = toolbar_items.begin(); |
| i < toolbar_items.end(); ++i) { |
| if (index-- == 0) |
| return i->get(); |
| } |
| return NULL; |
| } |
| |
| protected: |
| ExtensionToolbarModel* model_; |
| |
| int inserted_count_; |
| int removed_count_; |
| int moved_count_; |
| }; |
| |
| IN_PROC_BROWSER_TEST_F(ExtensionToolbarModelTest, Basic) { |
| // Load an extension with no browser action. |
| ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("none"))); |
| |
| // This extension should not be in the model (has no browser action). |
| EXPECT_EQ(0, inserted_count_); |
| EXPECT_EQ(0u, model_->toolbar_items().size()); |
| ASSERT_EQ(NULL, ExtensionAt(0)); |
| |
| // Load an extension with a browser action. |
| ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("basics"))); |
| |
| // We should now find our extension in the model. |
| EXPECT_EQ(1, inserted_count_); |
| EXPECT_EQ(1u, model_->toolbar_items().size()); |
| const Extension* extension = ExtensionAt(0); |
| ASSERT_TRUE(NULL != extension); |
| EXPECT_STREQ("A browser action with no icon that makes the page red", |
| extension->name().c_str()); |
| |
| // Should be a no-op, but still fires the events. |
| model_->MoveBrowserAction(extension, 0); |
| EXPECT_EQ(1, moved_count_); |
| EXPECT_EQ(1u, model_->toolbar_items().size()); |
| const Extension* extension2 = ExtensionAt(0); |
| EXPECT_EQ(extension, extension2); |
| |
| UnloadExtension(extension->id()); |
| EXPECT_EQ(1, removed_count_); |
| EXPECT_EQ(0u, model_->toolbar_items().size()); |
| EXPECT_EQ(NULL, ExtensionAt(0)); |
| } |
| |
| #if defined(OS_MACOSX) |
| // Flaky on Mac 10.8 Blink canary bots: http://crbug.com/166580 |
| #define MAYBE_ReorderAndReinsert DISABLED_ReorderAndReinsert |
| #else |
| #define MAYBE_ReorderAndReinsert ReorderAndReinsert |
| #endif |
| IN_PROC_BROWSER_TEST_F(ExtensionToolbarModelTest, MAYBE_ReorderAndReinsert) { |
| // Load an extension with a browser action. |
| base::FilePath extension_a_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("basics")); |
| ASSERT_TRUE(LoadExtension(extension_a_path)); |
| |
| // First extension loaded. |
| EXPECT_EQ(1, inserted_count_); |
| EXPECT_EQ(1u, model_->toolbar_items().size()); |
| const Extension* extensionA = ExtensionAt(0); |
| ASSERT_TRUE(NULL != extensionA); |
| EXPECT_STREQ("A browser action with no icon that makes the page red", |
| extensionA->name().c_str()); |
| |
| // Load another extension with a browser action. |
| base::FilePath extension_b_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("popup")); |
| ASSERT_TRUE(LoadExtension(extension_b_path)); |
| |
| // Second extension loaded. |
| EXPECT_EQ(2, inserted_count_); |
| EXPECT_EQ(2u, model_->toolbar_items().size()); |
| const Extension* extensionB = ExtensionAt(1); |
| ASSERT_TRUE(NULL != extensionB); |
| EXPECT_STREQ("Popup tester", extensionB->name().c_str()); |
| |
| // Load yet another extension with a browser action. |
| base::FilePath extension_c_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("remove_popup")); |
| ASSERT_TRUE(LoadExtension(extension_c_path)); |
| |
| // Third extension loaded. |
| EXPECT_EQ(3, inserted_count_); |
| EXPECT_EQ(3u, model_->toolbar_items().size()); |
| const Extension* extensionC = ExtensionAt(2); |
| ASSERT_TRUE(NULL != extensionC); |
| EXPECT_STREQ("A page action which removes a popup.", |
| extensionC->name().c_str()); |
| |
| // Order is now A, B, C. Let's put C first. |
| model_->MoveBrowserAction(extensionC, 0); |
| EXPECT_EQ(1, moved_count_); |
| EXPECT_EQ(3u, model_->toolbar_items().size()); |
| EXPECT_EQ(extensionC, ExtensionAt(0)); |
| EXPECT_EQ(extensionA, ExtensionAt(1)); |
| EXPECT_EQ(extensionB, ExtensionAt(2)); |
| EXPECT_EQ(NULL, ExtensionAt(3)); |
| |
| // Order is now C, A, B. Let's put A last. |
| model_->MoveBrowserAction(extensionA, 2); |
| EXPECT_EQ(2, moved_count_); |
| EXPECT_EQ(3u, model_->toolbar_items().size()); |
| EXPECT_EQ(extensionC, ExtensionAt(0)); |
| EXPECT_EQ(extensionB, ExtensionAt(1)); |
| EXPECT_EQ(extensionA, ExtensionAt(2)); |
| EXPECT_EQ(NULL, ExtensionAt(3)); |
| |
| // Order is now C, B, A. Let's remove B. |
| std::string idB = extensionB->id(); |
| UnloadExtension(idB); |
| EXPECT_EQ(1, removed_count_); |
| EXPECT_EQ(2u, model_->toolbar_items().size()); |
| EXPECT_EQ(extensionC, ExtensionAt(0)); |
| EXPECT_EQ(extensionA, ExtensionAt(1)); |
| EXPECT_EQ(NULL, ExtensionAt(2)); |
| |
| // Load extension B again. |
| ASSERT_TRUE(LoadExtension(extension_b_path)); |
| |
| // Extension B loaded again. |
| EXPECT_EQ(4, inserted_count_); |
| EXPECT_EQ(3u, model_->toolbar_items().size()); |
| // Make sure it gets its old spot in the list. We should get the same |
| // extension again, otherwise the order has changed. |
| ASSERT_STREQ(idB.c_str(), ExtensionAt(1)->id().c_str()); |
| |
| // Unload B again. |
| UnloadExtension(idB); |
| EXPECT_EQ(2, removed_count_); |
| EXPECT_EQ(2u, model_->toolbar_items().size()); |
| EXPECT_EQ(extensionC, ExtensionAt(0)); |
| EXPECT_EQ(extensionA, ExtensionAt(1)); |
| EXPECT_EQ(NULL, ExtensionAt(2)); |
| |
| // Order is now C, A. Flip it. |
| model_->MoveBrowserAction(extensionA, 0); |
| EXPECT_EQ(3, moved_count_); |
| EXPECT_EQ(2u, model_->toolbar_items().size()); |
| EXPECT_EQ(extensionA, ExtensionAt(0)); |
| EXPECT_EQ(extensionC, ExtensionAt(1)); |
| EXPECT_EQ(NULL, ExtensionAt(2)); |
| |
| // Move A to the location it already occupies. |
| model_->MoveBrowserAction(extensionA, 0); |
| EXPECT_EQ(4, moved_count_); |
| EXPECT_EQ(2u, model_->toolbar_items().size()); |
| EXPECT_EQ(extensionA, ExtensionAt(0)); |
| EXPECT_EQ(extensionC, ExtensionAt(1)); |
| EXPECT_EQ(NULL, ExtensionAt(2)); |
| |
| // Order is now A, C. Remove C. |
| std::string idC = extensionC->id(); |
| UnloadExtension(idC); |
| EXPECT_EQ(3, removed_count_); |
| EXPECT_EQ(1u, model_->toolbar_items().size()); |
| EXPECT_EQ(extensionA, ExtensionAt(0)); |
| EXPECT_EQ(NULL, ExtensionAt(1)); |
| |
| // Load extension C again. |
| ASSERT_TRUE(LoadExtension(extension_c_path)); |
| |
| // Extension C loaded again. |
| EXPECT_EQ(5, inserted_count_); |
| EXPECT_EQ(2u, model_->toolbar_items().size()); |
| // Make sure it gets its old spot in the list (at the very end). |
| ASSERT_STREQ(idC.c_str(), ExtensionAt(1)->id().c_str()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(ExtensionToolbarModelTest, UnloadAndDisableMultiple) { |
| // Load three extensions with browser action. |
| base::FilePath extension_a_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("basics")); |
| ASSERT_TRUE(LoadExtension(extension_a_path)); |
| base::FilePath extension_b_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("popup")); |
| ASSERT_TRUE(LoadExtension(extension_b_path)); |
| base::FilePath extension_c_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("remove_popup")); |
| ASSERT_TRUE(LoadExtension(extension_c_path)); |
| |
| // Verify we got the three we asked for and that they are ordered as: A, B, C. |
| const Extension* extensionA = ExtensionAt(0); |
| const Extension* extensionB = ExtensionAt(1); |
| const Extension* extensionC = ExtensionAt(2); |
| std::string idA = extensionA->id(); |
| std::string idB = extensionB->id(); |
| std::string idC = extensionC->id(); |
| EXPECT_STREQ("A browser action with no icon that makes the page red", |
| extensionA->name().c_str()); |
| EXPECT_STREQ("Popup tester", extensionB->name().c_str()); |
| EXPECT_STREQ("A page action which removes a popup.", |
| extensionC->name().c_str()); |
| |
| // Unload B, then C, then A. |
| UnloadExtension(idB); |
| UnloadExtension(idC); |
| UnloadExtension(idA); |
| |
| // Load C, then A, then B. |
| ASSERT_TRUE(LoadExtension(extension_c_path)); |
| ASSERT_TRUE(LoadExtension(extension_a_path)); |
| ASSERT_TRUE(LoadExtension(extension_b_path)); |
| EXPECT_EQ(0, moved_count_); |
| |
| extensionA = ExtensionAt(0); |
| extensionB = ExtensionAt(1); |
| extensionC = ExtensionAt(2); |
| |
| // Make sure we get the order we started with (A, B, C). |
| EXPECT_STREQ(idA.c_str(), extensionA->id().c_str()); |
| EXPECT_STREQ(idB.c_str(), extensionB->id().c_str()); |
| EXPECT_STREQ(idC.c_str(), extensionC->id().c_str()); |
| |
| // Put C in the middle and A to the end. |
| model_->MoveBrowserAction(extensionC, 1); |
| model_->MoveBrowserAction(extensionA, 2); |
| |
| // Make sure we get this order (C, B, A). |
| EXPECT_STREQ(idC.c_str(), ExtensionAt(0)->id().c_str()); |
| EXPECT_STREQ(idB.c_str(), ExtensionAt(1)->id().c_str()); |
| EXPECT_STREQ(idA.c_str(), ExtensionAt(2)->id().c_str()); |
| |
| // Disable B, then C, then A. |
| DisableExtension(idB); |
| DisableExtension(idC); |
| DisableExtension(idA); |
| |
| // Enable C, then A, then B. |
| EnableExtension(idA); |
| EnableExtension(idB); |
| EnableExtension(idC); |
| |
| // Make sure we get the order we started with. |
| EXPECT_STREQ(idC.c_str(), ExtensionAt(0)->id().c_str()); |
| EXPECT_STREQ(idB.c_str(), ExtensionAt(1)->id().c_str()); |
| EXPECT_STREQ(idA.c_str(), ExtensionAt(2)->id().c_str()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(ExtensionToolbarModelTest, Uninstall) { |
| // Load two extensions with browser action. |
| base::FilePath extension_a_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("basics")); |
| ASSERT_TRUE(LoadExtension(extension_a_path)); |
| base::FilePath extension_b_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("popup")); |
| ASSERT_TRUE(LoadExtension(extension_b_path)); |
| |
| // Verify we got what we came for. |
| const Extension* extensionA = ExtensionAt(0); |
| const Extension* extensionB = ExtensionAt(1); |
| std::string idA = extensionA->id(); |
| std::string idB = extensionB->id(); |
| EXPECT_STREQ("A browser action with no icon that makes the page red", |
| extensionA->name().c_str()); |
| EXPECT_STREQ("Popup tester", extensionB->name().c_str()); |
| |
| // Order is now A, B. Make B first. |
| model_->MoveBrowserAction(extensionB, 0); |
| |
| // Order is now B, A. Uninstall Extension B. |
| UninstallExtension(idB); |
| |
| // List contains only A now. Validate that. |
| EXPECT_STREQ(idA.c_str(), ExtensionAt(0)->id().c_str()); |
| EXPECT_EQ(1u, model_->toolbar_items().size()); |
| |
| // Load Extension B again. |
| ASSERT_TRUE(LoadExtension(extension_b_path)); |
| EXPECT_EQ(2u, model_->toolbar_items().size()); |
| |
| // Make sure Extension B is _not_ first (should have been forgotten at |
| // uninstall time). |
| EXPECT_STREQ(idA.c_str(), ExtensionAt(0)->id().c_str()); |
| EXPECT_STREQ(idB.c_str(), ExtensionAt(1)->id().c_str()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(ExtensionToolbarModelTest, ReorderOnPrefChange) { |
| // Load three extensions with browser action. |
| base::FilePath extension_a_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("basics")); |
| ASSERT_TRUE(LoadExtension(extension_a_path)); |
| base::FilePath extension_b_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("popup")); |
| ASSERT_TRUE(LoadExtension(extension_b_path)); |
| base::FilePath extension_c_path(test_data_dir_.AppendASCII("api_test") |
| .AppendASCII("browser_action") |
| .AppendASCII("remove_popup")); |
| ASSERT_TRUE(LoadExtension(extension_c_path)); |
| std::string id_a = ExtensionAt(0)->id(); |
| std::string id_b = ExtensionAt(1)->id(); |
| std::string id_c = ExtensionAt(2)->id(); |
| |
| // Change value of toolbar preference. |
| extensions::ExtensionIdList new_order; |
| new_order.push_back(id_c); |
| new_order.push_back(id_b); |
| extensions::ExtensionPrefs::Get(browser()->profile())->SetToolbarOrder( |
| new_order); |
| |
| // Verify order is changed. |
| EXPECT_EQ(id_c, ExtensionAt(0)->id()); |
| EXPECT_EQ(id_b, ExtensionAt(1)->id()); |
| EXPECT_EQ(id_a, ExtensionAt(2)->id()); |
| } |