| // Copyright (c) 2006-2008 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/bookmarks/bookmark_context_menu.h" |
| #include "chrome/browser/bookmarks/bookmark_model.h" |
| #include "chrome/browser/bookmarks/bookmark_utils.h" |
| #include "chrome/browser/profile.h" |
| #include "chrome/browser/views/bookmark_bar_view.h" |
| #include "chrome/common/pref_names.h" |
| #include "chrome/common/pref_service.h" |
| #include "chrome/browser/tab_contents/page_navigator.h" |
| #include "chrome/test/testing_profile.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| #include "generated_resources.h" |
| |
| namespace { |
| |
| // PageNavigator implementation that records the URL. |
| class TestingPageNavigator : public PageNavigator { |
| public: |
| virtual void OpenURL(const GURL& url, |
| const GURL& referrer, |
| WindowOpenDisposition disposition, |
| PageTransition::Type transition) { |
| urls_.push_back(url); |
| } |
| |
| std::vector<GURL> urls_; |
| }; |
| |
| } // namespace |
| |
| class BookmarkContextMenuTest : public testing::Test { |
| public: |
| BookmarkContextMenuTest() |
| : model_(NULL) { |
| } |
| |
| virtual void SetUp() { |
| BookmarkBarView::testing_ = true; |
| |
| profile_.reset(new TestingProfile()); |
| profile_->set_has_history_service(true); |
| profile_->CreateBookmarkModel(true); |
| profile_->BlockUntilBookmarkModelLoaded(); |
| |
| model_ = profile_->GetBookmarkModel(); |
| |
| AddTestData(); |
| } |
| |
| virtual void TearDown() { |
| BookmarkBarView::testing_ = false; |
| |
| // Flush the message loop to make Purify happy. |
| message_loop_.RunAllPending(); |
| } |
| |
| protected: |
| MessageLoopForUI message_loop_; |
| scoped_ptr<TestingProfile> profile_; |
| BookmarkModel* model_; |
| TestingPageNavigator navigator_; |
| |
| private: |
| // Creates the following structure: |
| // a |
| // F1 |
| // f1a |
| // F11 |
| // f11a |
| // F2 |
| // F3 |
| // F4 |
| // f4a |
| void AddTestData() { |
| std::string test_base = "file:///c:/tmp/"; |
| |
| model_->AddURL(model_->GetBookmarkBarNode(), 0, L"a", |
| GURL(test_base + "a")); |
| BookmarkNode* f1 = |
| model_->AddGroup(model_->GetBookmarkBarNode(), 1, L"F1"); |
| model_->AddURL(f1, 0, L"f1a", GURL(test_base + "f1a")); |
| BookmarkNode* f11 = model_->AddGroup(f1, 1, L"F11"); |
| model_->AddURL(f11, 0, L"f11a", GURL(test_base + "f11a")); |
| model_->AddGroup(model_->GetBookmarkBarNode(), 2, L"F2"); |
| model_->AddGroup(model_->GetBookmarkBarNode(), 3, L"F3"); |
| BookmarkNode* f4 = |
| model_->AddGroup(model_->GetBookmarkBarNode(), 4, L"F4"); |
| model_->AddURL(f4, 0, L"f4a", GURL(test_base + "f4a")); |
| } |
| }; |
| |
| // Tests Deleting from the menu. |
| TEST_F(BookmarkContextMenuTest, DeleteURL) { |
| std::vector<BookmarkNode*> nodes; |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(0)); |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, nodes[0]->GetParent(), nodes, |
| BookmarkContextMenu::BOOKMARK_BAR); |
| GURL url = model_->GetBookmarkBarNode()->GetChild(0)->GetURL(); |
| ASSERT_TRUE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| // Delete the URL. |
| controller.ExecuteCommand(IDS_BOOKMARK_BAR_REMOVE); |
| // Model shouldn't have URL anymore. |
| ASSERT_FALSE(model_->IsBookmarked(url)); |
| } |
| |
| // Tests open all on a folder with a couple of bookmarks. |
| TEST_F(BookmarkContextMenuTest, OpenAll) { |
| BookmarkNode* folder = model_->GetBookmarkBarNode()->GetChild(1); |
| bookmark_utils::OpenAll( |
| NULL, profile_.get(), &navigator_, folder, NEW_FOREGROUND_TAB); |
| |
| // Should have navigated to F1's children. |
| ASSERT_EQ(2, navigator_.urls_.size()); |
| ASSERT_TRUE(folder->GetChild(0)->GetURL() == navigator_.urls_[0]); |
| ASSERT_TRUE(folder->GetChild(1)->GetChild(0)->GetURL() == |
| navigator_.urls_[1]); |
| } |
| |
| // Tests the enabled state of the menus when supplied an empty vector. |
| TEST_F(BookmarkContextMenuTest, EmptyNodes) { |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, model_->other_node(), |
| std::vector<BookmarkNode*>(), BookmarkContextMenu::BOOKMARK_BAR); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_NEW_WINDOW)); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_INCOGNITO)); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOKMARK_MANAGER_SHOW_IN_FOLDER)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_ADD_NEW_BOOKMARK)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_NEW_FOLDER)); |
| } |
| |
| // Tests the enabled state of the menus when supplied a vector with a single |
| // url. |
| TEST_F(BookmarkContextMenuTest, SingleURL) { |
| std::vector<BookmarkNode*> nodes; |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(0)); |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, nodes[0]->GetParent(), |
| nodes, BookmarkContextMenu::BOOKMARK_BAR); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_NEW_WINDOW)); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_INCOGNITO)); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOKMARK_MANAGER_SHOW_IN_FOLDER)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_ADD_NEW_BOOKMARK)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_NEW_FOLDER)); |
| } |
| |
| // Tests the enabled state of the menus when supplied a vector with multiple |
| // urls. |
| TEST_F(BookmarkContextMenuTest, MultipleURLs) { |
| std::vector<BookmarkNode*> nodes; |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(0)); |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(1)->GetChild(0)); |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, nodes[0]->GetParent(), |
| nodes, BookmarkContextMenu::BOOKMARK_BAR); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_NEW_WINDOW)); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_INCOGNITO)); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOKMARK_MANAGER_SHOW_IN_FOLDER)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_ADD_NEW_BOOKMARK)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_NEW_FOLDER)); |
| } |
| |
| // Tests the enabled state of the menus when supplied an vector with a single |
| // folder. |
| TEST_F(BookmarkContextMenuTest, SingleFolder) { |
| std::vector<BookmarkNode*> nodes; |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(2)); |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, nodes[0]->GetParent(), |
| nodes, BookmarkContextMenu::BOOKMARK_BAR); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_NEW_WINDOW)); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_INCOGNITO)); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOKMARK_MANAGER_SHOW_IN_FOLDER)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_ADD_NEW_BOOKMARK)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_NEW_FOLDER)); |
| } |
| |
| // Tests the enabled state of the menus when supplied a vector with multiple |
| // folders, all of which are empty. |
| TEST_F(BookmarkContextMenuTest, MultipleEmptyFolders) { |
| std::vector<BookmarkNode*> nodes; |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(2)); |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(3)); |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, nodes[0]->GetParent(), |
| nodes, BookmarkContextMenu::BOOKMARK_BAR); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_NEW_WINDOW)); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_INCOGNITO)); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOKMARK_MANAGER_SHOW_IN_FOLDER)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_ADD_NEW_BOOKMARK)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_NEW_FOLDER)); |
| } |
| |
| // Tests the enabled state of the menus when supplied a vector with multiple |
| // folders, some of which contain URLs. |
| TEST_F(BookmarkContextMenuTest, MultipleFoldersWithURLs) { |
| std::vector<BookmarkNode*> nodes; |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(3)); |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(4)); |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, nodes[0]->GetParent(), |
| nodes, BookmarkContextMenu::BOOKMARK_BAR); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_NEW_WINDOW)); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_INCOGNITO)); |
| EXPECT_TRUE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOKMARK_MANAGER_SHOW_IN_FOLDER)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_ADD_NEW_BOOKMARK)); |
| EXPECT_TRUE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_NEW_FOLDER)); |
| } |
| |
| // Tests the enabled state of open incognito. |
| TEST_F(BookmarkContextMenuTest, DisableIncognito) { |
| std::vector<BookmarkNode*> nodes; |
| nodes.push_back(model_->GetBookmarkBarNode()->GetChild(0)); |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, nodes[0]->GetParent(), |
| nodes, BookmarkContextMenu::BOOKMARK_BAR); |
| profile_->set_off_the_record(true); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_INCOGNITO)); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_INCOGNITO)); |
| } |
| |
| // Tests that you can't remove/edit when showing the other node. |
| TEST_F(BookmarkContextMenuTest, DisabledItemsWithOtherNode) { |
| std::vector<BookmarkNode*> nodes; |
| nodes.push_back(model_->other_node()); |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, nodes[0], nodes, |
| BookmarkContextMenu::BOOKMARK_BAR); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_EDIT)); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| } |
| |
| // Tests the enabled state of the menus when supplied an empty vector and null |
| // parent. |
| TEST_F(BookmarkContextMenuTest, EmptyNodesNullParent) { |
| BookmarkContextMenu controller( |
| NULL, profile_.get(), NULL, NULL, NULL, std::vector<BookmarkNode*>(), |
| BookmarkContextMenu::BOOKMARK_MANAGER_ORGANIZE_MENU); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_NEW_WINDOW)); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOMARK_BAR_OPEN_ALL_INCOGNITO)); |
| EXPECT_FALSE(controller.IsCommandEnabled(IDS_BOOKMARK_BAR_REMOVE)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOKMARK_MANAGER_SHOW_IN_FOLDER)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_ADD_NEW_BOOKMARK)); |
| EXPECT_FALSE( |
| controller.IsCommandEnabled(IDS_BOOMARK_BAR_NEW_FOLDER)); |
| } |