// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/views/bookmarks/bookmark_context_menu.h"

#include <stddef.h>

#include <memory>
#include <string>
#include <utility>
#include <vector>

#include "base/memory/raw_ptr.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/metrics/user_action_tester.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/with_feature_override.h"
#include "base/values.h"
#include "build/build_config.h"
#include "chrome/app/chrome_command_ids.h"
#include "chrome/browser/bookmarks/bookmark_merged_surface_service.h"
#include "chrome/browser/bookmarks/bookmark_merged_surface_service_factory.h"
#include "chrome/browser/bookmarks/bookmark_model_factory.h"
#include "chrome/browser/bookmarks/bookmark_test_helpers.h"
#include "chrome/browser/bookmarks/managed_bookmark_service_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/bookmarks/bookmark_context_menu_controller.h"
#include "chrome/browser/ui/bookmarks/bookmark_utils_desktop.h"
#include "chrome/browser/ui/bookmarks/test_bookmark_navigation_wrapper.h"
#include "chrome/common/webui_url_constants.h"
#include "chrome/test/base/testing_profile.h"
#include "components/bookmarks/browser/bookmark_model.h"
#include "components/bookmarks/browser/bookmark_node.h"
#include "components/bookmarks/common/bookmark_bar_visibility_state.h"
#include "components/bookmarks/common/bookmark_pref_names.h"
#include "components/bookmarks/managed/managed_bookmark_service.h"
#include "components/bookmarks/test/bookmark_test_helpers.h"
#include "components/prefs/pref_service.h"
#include "components/search/ntp_features.h"
#include "components/signin/public/base/signin_switches.h"
#include "components/sync_preferences/testing_pref_service_syncable.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/clipboard/clipboard.h"
#include "ui/base/clipboard/test/test_clipboard.h"
#include "ui/base/mojom/menu_source_type.mojom.h"
#include "ui/views/controls/image_view.h"
#include "ui/views/controls/menu/menu_item_view.h"
#include "ui/views/controls/menu/test_menu_item_view.h"
#include "ui/views/test/scoped_views_test_helper.h"
#include "ui/views/widget/widget.h"

using bookmarks::BookmarkModel;
using bookmarks::BookmarkNode;

namespace {

base::test::ScopedFeatureList CreateFeatureList(
    bool enable_ntp_simplification) {
  base::test::ScopedFeatureList feature_list;
  if (enable_ntp_simplification) {
    feature_list.InitAndEnableFeature(
        ntp_features::kNtpSimplificationBookmarkBar);
  }
  return feature_list;
}

}  // namespace

class BookmarkContextMenuTest : public testing::Test,
                                public base::test::WithFeatureOverride {
 public:
  explicit BookmarkContextMenuTest(bool enable_ntp_simplification = false)
      : base::test::WithFeatureOverride(
            switches::kSyncEnableBookmarksInTransportMode),
        ntp_simplification_feature_list_(
            CreateFeatureList(enable_ntp_simplification)) {}

  void SetUp() override {
    TestingProfile::Builder profile_builder;
    profile_builder.AddTestingFactory(
        BookmarkModelFactory::GetInstance(),
        BookmarkModelFactory::GetDefaultFactory());
    profile_builder.AddTestingFactory(
        ManagedBookmarkServiceFactory::GetInstance(),
        ManagedBookmarkServiceFactory::GetDefaultFactory());
    profile_builder.AddTestingFactory(
        BookmarkMergedSurfaceServiceFactory::GetInstance(),
        BookmarkMergedSurfaceServiceFactory::GetDefaultFactory());
    profile_ = profile_builder.Build();
    WaitForBookmarkMergedSurfaceServiceToLoad(
        BookmarkMergedSurfaceServiceFactory::GetForProfile(profile_.get()));
    model_ = BookmarkMergedSurfaceServiceFactory::GetForProfile(profile_.get())
                 ->bookmark_model();
    AddTestData();

    if (SyncEnableBookmarksInTransportModeEnabled()) {
      model_->CreateAccountPermanentFolders();
      AddAccountTestData();
    }

    bookmarks::BookmarkNavigationWrapper::SetInstanceForTesting(&wrapper_);

    // CutCopyPasteNode executes IDC_COPY and IDC_CUT commands.
    ui::TestClipboard::CreateForCurrentThread();
  }

  void TearDown() override {
    if (SyncEnableBookmarksInTransportModeEnabled()) {
      model_->RemoveAccountPermanentFolders();
    }
    ui::Clipboard::DestroyClipboardForCurrentThread();
  }

 protected:
  bool SyncEnableBookmarksInTransportModeEnabled() {
    return IsParamFeatureEnabled();
  }

  // Must be declared before `task_environment_` so feature overrides are set
  // before background threads start.
  base::test::ScopedFeatureList ntp_simplification_feature_list_;
  content::BrowserTaskEnvironment task_environment_;
  views::ScopedViewsTestHelper views_test_helper_;
  std::unique_ptr<TestingProfile> profile_;
  raw_ptr<BookmarkModel> model_;
  TestingBookmarkNavigationWrapper wrapper_;

 private:
  // Creates the following structure:
  // a
  // F1
  //  f1a
  // -f1b as "chrome://settings"
  //  F11
  //   f11a
  // F2
  // F3
  // F4
  //   f4a
  void AddTestData() {
    const BookmarkNode* bb_node = model_->bookmark_bar_node();
    std::string test_base = "file:///c:/tmp/";
    model_->AddURL(bb_node, 0, u"a", GURL(test_base + "a"));
    const BookmarkNode* f1 = model_->AddFolder(bb_node, 1, u"F1");
    model_->AddURL(f1, 0, u"f1a", GURL(test_base + "f1a"));
    model_->AddURL(f1, 1, u"f1b", GURL(chrome::kChromeUISettingsURL));
    const BookmarkNode* f11 = model_->AddFolder(f1, 2, u"F11");
    model_->AddURL(f11, 0, u"f11a", GURL(test_base + "f11a"));
    model_->AddFolder(bb_node, 2, u"F2");
    model_->AddFolder(bb_node, 3, u"F3");
    const BookmarkNode* f4 = model_->AddFolder(bb_node, 4, u"F4");
    model_->AddURL(f4, 0, u"f4a", GURL(test_base + "f4a"));
  }

  // Creates the following structure:
  // acc_a
  // acc_F1
  //  acc_f1a
  //  acc_F11
  //   acc_f11a
  // acc_F2
  void AddAccountTestData() {
    CHECK(SyncEnableBookmarksInTransportModeEnabled());
    const BookmarkNode* bb_node = model_->account_bookmark_bar_node();
    std::string test_base = "file:///c:/tmp/";
    model_->AddURL(bb_node, 0, u"acc_a", GURL(test_base + "acc_a"));
    const BookmarkNode* f1 = model_->AddFolder(bb_node, 1, u"acc_F1");
    model_->AddURL(f1, 0, u"acc_f1a", GURL(test_base + "acc_f1a"));
    const BookmarkNode* f11 = model_->AddFolder(f1, 1, u"acc_F11");
    model_->AddURL(f11, 0, u"acc_f11a", GURL(test_base + "acc_f11a"));
    model_->AddFolder(bb_node, 2, u"acc_F2");
  }
};

// Tests deleting from the menu.
TEST_P(BookmarkContextMenuTest, DeleteURL) {
  const BookmarkNode* node_to_delete =
      SyncEnableBookmarksInTransportModeEnabled()
          ? model_->account_bookmark_bar_node()->children().front().get()
          : model_->bookmark_bar_node()->children().front().get();
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      node_to_delete};
  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);
  GURL url = node_to_delete->url();
  ASSERT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_REMOVE));
  // Delete the URL.
  controller.ExecuteCommand(IDC_BOOKMARK_BAR_REMOVE, 0);
  // Model shouldn't have URL anymore.
  ASSERT_FALSE(model_->IsBookmarked(url));
}

// Tests counting tabs for 'open all' on a folder with a couple of bookmarks.
TEST_P(BookmarkContextMenuTest, OpenCount) {
  const BookmarkNode* folder = model_->bookmark_bar_node()->children()[1].get();
  // Should count F1's child but not F11's child, as that's what OpenAll would
  // open.
  EXPECT_EQ(2, bookmarks::OpenCount(folder));

  if (SyncEnableBookmarksInTransportModeEnabled()) {
    folder = model_->account_bookmark_bar_node()->children()[1].get();
    // Should count acc_F1's child but not acc_F11's child.
    EXPECT_EQ(1, bookmarks::OpenCount(folder));
  }
}

// Same as above, but for counting bookmarks that would be opened in an
// incognito window.
TEST_P(BookmarkContextMenuTest, OpenCountIncognito) {
  const BookmarkNode* folder = model_->bookmark_bar_node()->children()[1].get();

  // Should count f1a but not f1b, as that's what OpenAll would open.
  EXPECT_EQ(1, bookmarks::OpenCount(folder, profile_.get()));
}

// Tests the enabled state of the menus when supplied a vector with a single
// url.
TEST_P(BookmarkContextMenuTest, SingleURL) {
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      model_->bookmark_bar_node()->children().front().get(),
  };
  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL));
  EXPECT_TRUE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_NEW_WINDOW));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_INCOGNITO));
  // Due to no active browser.
  EXPECT_FALSE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_SPLIT_VIEW));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_REMOVE));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_ADD_NEW_BOOKMARK));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_NEW_FOLDER));
}

// Tests the enabled state of the menus when supplied a vector with multiple
// urls.
TEST_P(BookmarkContextMenuTest, MultipleURLs) {
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      model_->bookmark_bar_node()->children()[0].get(),
      model_->bookmark_bar_node()->children()[1]->children()[0].get(),
  };
  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL));
  EXPECT_TRUE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_NEW_WINDOW));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_INCOGNITO));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_REMOVE));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_ADD_NEW_BOOKMARK));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_NEW_FOLDER));
}

// Tests the enabled state of the menus when supplied an vector with a single
// folder.
TEST_P(BookmarkContextMenuTest, SingleFolder) {
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      model_->bookmark_bar_node()->children()[2].get(),
  };
  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);
  EXPECT_FALSE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL));
  EXPECT_FALSE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_NEW_WINDOW));
  EXPECT_FALSE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_INCOGNITO));
  EXPECT_FALSE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_SPLIT_VIEW));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_REMOVE));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_ADD_NEW_BOOKMARK));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_NEW_FOLDER));
}

// Tests the enabled state of the menus when supplied a vector with multiple
// folders, all of which are empty.
TEST_P(BookmarkContextMenuTest, MultipleEmptyFolders) {
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      model_->bookmark_bar_node()->children()[2].get(),
      model_->bookmark_bar_node()->children()[3].get(),
  };
  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);
  EXPECT_FALSE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL));
  EXPECT_FALSE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_NEW_WINDOW));
  EXPECT_FALSE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_INCOGNITO));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_REMOVE));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_ADD_NEW_BOOKMARK));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_NEW_FOLDER));
}

// Tests the enabled state of the menus when supplied a vector with multiple
// folders, some of which contain URLs.
TEST_P(BookmarkContextMenuTest, MultipleFoldersWithURLs) {
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      model_->bookmark_bar_node()->children()[3].get(),
      model_->bookmark_bar_node()->children()[4].get(),
  };
  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL));
  EXPECT_TRUE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_NEW_WINDOW));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_INCOGNITO));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_REMOVE));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_ADD_NEW_BOOKMARK));
  EXPECT_TRUE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_NEW_FOLDER));
}

// Tests the enabled state of open incognito.
TEST_P(BookmarkContextMenuTest, DisableIncognito) {
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      model_->bookmark_bar_node()->children().front().get(),
  };
  Profile* incognito =
      profile_->GetPrimaryOTRProfile(/*create_if_needed=*/true);
  BookmarkContextMenu controller(nullptr, nullptr, incognito,
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);
  EXPECT_FALSE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_INCOGNITO));
  EXPECT_FALSE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_OPEN_ALL_INCOGNITO));
}

// Tests that you can't remove/edit when showing the other node.
TEST_P(BookmarkContextMenuTest, DisabledItemsWithOtherNode) {
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      model_->other_node()};
  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);
  EXPECT_FALSE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_EDIT));
  EXPECT_FALSE(controller.IsCommandEnabled(IDC_BOOKMARK_BAR_REMOVE));
}

// Tests the Cut/Copy/Paste commands.
TEST_P(BookmarkContextMenuTest, CutCopyPasteNodeWithoutAccountNodes) {
  // Skip this test if account folders are created.
  if (SyncEnableBookmarksInTransportModeEnabled()) {
    GTEST_SKIP();
  }

  const BookmarkNode* bb_node = model_->bookmark_bar_node();
  size_t original_count = bb_node->children().size();
  ASSERT_EQ(original_count, 5u);

  // Select the first bookmark (f1a).
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> selected_nodes =
      {bb_node->children()[0].get()};

  // Test the Copy command.
  {
    BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                   BookmarkLaunchLocation::kNone,
                                   selected_nodes, false, false);

    ASSERT_TRUE(controller.IsCommandEnabled(IDC_COPY));
    controller.ExecuteCommand(IDC_COPY, 0);

    // No change after copy.
    EXPECT_EQ(original_count, bb_node->children().size());
  }

  // Test the Paste command.
  {
    BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                   BookmarkLaunchLocation::kNone,
                                   selected_nodes, false, /*can_paste=*/true);

    ASSERT_TRUE(controller.IsCommandEnabled(IDC_PASTE));
    controller.ExecuteCommand(IDC_PASTE, 0);

    // The newly pasted bookmark should be inserted after the copied bookmark.
    EXPECT_EQ(original_count + 1, bb_node->children().size());
    EXPECT_TRUE(bb_node->children()[1]->is_url());
    EXPECT_EQ(bb_node->children()[0]->url(), bb_node->children()[1]->url());
  }

  // Test the Cut command.
  {
    BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                   BookmarkLaunchLocation::kNone,
                                   selected_nodes, false, false);

    ASSERT_TRUE(controller.IsCommandEnabled(IDC_CUT));
    controller.ExecuteCommand(IDC_CUT, 0);

    EXPECT_TRUE(bb_node->children()[0]->is_url());
    EXPECT_TRUE(bb_node->children()[1]->is_folder());
    EXPECT_EQ(original_count, bb_node->children().size());
  }
}

// Tests the Cut/Copy/Paste commands when account nodes are created.
TEST_P(BookmarkContextMenuTest, CutCopyPasteNodeWithAccountNodes) {
  // This test requires account nodes to be created.
  if (!SyncEnableBookmarksInTransportModeEnabled()) {
    GTEST_SKIP();
  }

  const BookmarkNode* bb_node = model_->bookmark_bar_node();
  const BookmarkNode* account_bb_node = model_->account_bookmark_bar_node();
  size_t original_local_count = bb_node->children().size();
  size_t original_account_count = account_bb_node->children().size();
  ASSERT_EQ(original_local_count, 5u);
  ASSERT_EQ(original_account_count, 3u);

  // Test the Copy command.
  {
    // Select the first local bookmark (f1a).
    std::vector<raw_ptr<const BookmarkNode, VectorExperimental>>
        selected_nodes = {bb_node->children()[0].get()};
    BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                   BookmarkLaunchLocation::kNone,
                                   selected_nodes, false, false);

    ASSERT_TRUE(controller.IsCommandEnabled(IDC_COPY));
    controller.ExecuteCommand(IDC_COPY, 0);

    // No change in local count after copy.
    EXPECT_EQ(original_local_count, bb_node->children().size());
    EXPECT_EQ(original_account_count, account_bb_node->children().size());
  }

  // Test the Paste command.
  {
    // Select the first local bookmark (f1a).
    const BookmarkNode* local_bookmark_to_copy = bb_node->children()[0].get();
    std::vector<raw_ptr<const BookmarkNode, VectorExperimental>>
        selected_nodes = {local_bookmark_to_copy};

    BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                   BookmarkLaunchLocation::kNone,
                                   selected_nodes, false, /*can_paste=*/true);

    ASSERT_TRUE(controller.IsCommandEnabled(IDC_PASTE));
    controller.ExecuteCommand(IDC_PASTE, 0);

    // The newly pasted bookmark should be the last child of the account
    // bookmark bar node.
    ASSERT_EQ(original_account_count + 1, account_bb_node->children().size());
    EXPECT_EQ(original_local_count, bb_node->children().size());

    // Verify the pasted bookmark.
    EXPECT_TRUE(account_bb_node->children().back()->is_url());
    EXPECT_EQ(local_bookmark_to_copy->url(),
              account_bb_node->children().back()->url());
  }

  // Test the Cut command.
  {
    // Select the first account bookmark (acc_f1a).
    std::vector<raw_ptr<const BookmarkNode, VectorExperimental>>
        selected_nodes = {account_bb_node->children()[0].get()};
    BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                   BookmarkLaunchLocation::kNone,
                                   selected_nodes, false, false);

    ASSERT_TRUE(controller.IsCommandEnabled(IDC_CUT));
    controller.ExecuteCommand(IDC_CUT, 0);

    EXPECT_EQ(original_account_count, account_bb_node->children().size());
    EXPECT_TRUE(account_bb_node->children()[0]->is_folder());
    EXPECT_EQ(original_local_count, bb_node->children().size());
  }
}

// Tests that the "Show managed bookmarks" option in the context menu is only
// visible if the policy is set.
TEST_P(BookmarkContextMenuTest, ShowManagedBookmarks) {
  // Create a BookmarkContextMenu for the bookmarks bar.
  const BookmarkNode* bb_node = model_->bookmark_bar_node();
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      bb_node->children().front().get(),
  };
  std::unique_ptr<BookmarkContextMenu> controller(new BookmarkContextMenu(
      nullptr, nullptr, profile_.get(), BookmarkLaunchLocation::kNone, nodes,
      false, false));

  // Verify that there are no managed nodes yet.
  bookmarks::ManagedBookmarkService* managed =
      ManagedBookmarkServiceFactory::GetForProfile(profile_.get());
  EXPECT_TRUE(managed->managed_node()->children().empty());

  // The context menu should not show the option to "Show managed bookmarks".
  EXPECT_FALSE(
      controller->IsCommandVisible(IDC_BOOKMARK_BAR_SHOW_MANAGED_BOOKMARKS));
  views::MenuItemView* menu = controller->menu();
  EXPECT_FALSE(menu->GetMenuItemByID(IDC_BOOKMARK_BAR_SHOW_MANAGED_BOOKMARKS)
                   ->GetVisible());

  // Other options are not affected.
  EXPECT_TRUE(controller->IsCommandVisible(IDC_BOOKMARK_BAR_NEW_FOLDER));
  EXPECT_TRUE(menu->GetMenuItemByID(IDC_BOOKMARK_BAR_NEW_FOLDER)->GetVisible());

  // Now set the managed bookmarks policy.
  base::DictValue dict;
  dict.Set("name", "Google");
  dict.Set("url", "http://google.com");
  base::ListValue list;
  list.Append(std::move(dict));
  EXPECT_TRUE(managed->managed_node()->children().empty());
  profile_->GetPrefs()->Set(bookmarks::prefs::kManagedBookmarks,
                            base::Value(std::move(list)));
  EXPECT_FALSE(managed->managed_node()->children().empty());

  // New context menus now show the "Show managed bookmarks" option.
  controller = std::make_unique<BookmarkContextMenu>(
      nullptr, nullptr, profile_.get(), BookmarkLaunchLocation::kAttachedBar,
      nodes, false, false);

  EXPECT_TRUE(controller->IsCommandVisible(IDC_BOOKMARK_BAR_NEW_FOLDER));
  EXPECT_TRUE(
      controller->IsCommandVisible(IDC_BOOKMARK_BAR_SHOW_MANAGED_BOOKMARKS));
  menu = controller->menu();
  EXPECT_TRUE(menu->GetMenuItemByID(IDC_BOOKMARK_BAR_NEW_FOLDER)->GetVisible());
  EXPECT_TRUE(menu->GetMenuItemByID(IDC_BOOKMARK_BAR_SHOW_MANAGED_BOOKMARKS)
                  ->GetVisible());
}

// Regression test: parent widget destroyed before BookmarkContextMenu should
// not cause a dangling pointer crash.
TEST_P(BookmarkContextMenuTest, ParentWidgetDestroyedBeforeMenu) {
  views::Widget::InitParams params(
      views::Widget::InitParams::CLIENT_OWNS_WIDGET,
      views::Widget::InitParams::TYPE_WINDOW);
  params.context = views_test_helper_.GetContext();
  auto widget = std::make_unique<views::Widget>();
  widget->Init(std::move(params));

  const BookmarkNode* bb_node = model_->bookmark_bar_node();
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      bb_node->children().front().get(),
  };

  auto controller = std::make_unique<BookmarkContextMenu>(
      widget.get(), nullptr, profile_.get(),
      BookmarkLaunchLocation::kAttachedBar, nodes, false, false);

  // Destroy the widget first, then verify RunMenuAt and destruction are safe.
  widget->CloseNow();
  widget.reset();

  controller->RunMenuAt(gfx::Point(), ui::mojom::MenuSourceType::kNone);
  controller.reset();
}

class BookmarkContextMenuUpdateSubMenuStateTest
    : public BookmarkContextMenuTest {
 public:
  BookmarkContextMenuUpdateSubMenuStateTest()
      : BookmarkContextMenuTest(/*enable_ntp_simplification=*/true) {}
};

// Tests that the submenu items checked state is updated.
TEST_P(BookmarkContextMenuUpdateSubMenuStateTest, UpdateSubMenuState) {
  const BookmarkNode* bb_node = model_->bookmark_bar_node();
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      bb_node->children().front().get(),
  };

  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);

  views::MenuItemView* menu = controller.menu();
  views::MenuItemView* always_show_item =
      menu->GetMenuItemByID(IDC_BOOKMARK_BAR_SUBMENU_ALWAYS_SHOW);
  views::MenuItemView* always_hide_item =
      menu->GetMenuItemByID(IDC_BOOKMARK_BAR_SUBMENU_ALWAYS_HIDE);
  views::MenuItemView* only_on_ntp_item =
      menu->GetMenuItemByID(IDC_BOOKMARK_BAR_SUBMENU_ONLY_ON_NTP);

  ASSERT_TRUE(always_show_item);
  ASSERT_TRUE(always_hide_item);
  ASSERT_TRUE(only_on_ntp_item);

  // Helper function to check checkmark visibility.
  auto is_checked = [](views::MenuItemView* item) {
    views::ImageView* check_icon =
        views::TestMenuItemView::radio_check_image_view(item);
    return check_icon && check_icon->GetVisible();
  };

  profile_->GetPrefs()->SetInteger(
      bookmarks::prefs::kBookmarkBarVisibilityState,
      static_cast<int>(bookmarks::BookmarkBarVisibilityState::kAlwaysShow));
  controller.UpdateSubMenuState();

  EXPECT_TRUE(is_checked(always_show_item));
  EXPECT_FALSE(is_checked(always_hide_item));
  EXPECT_FALSE(is_checked(only_on_ntp_item));

  profile_->GetPrefs()->SetInteger(
      bookmarks::prefs::kBookmarkBarVisibilityState,
      static_cast<int>(bookmarks::BookmarkBarVisibilityState::kAlwaysHide));
  controller.UpdateSubMenuState();

  EXPECT_FALSE(is_checked(always_show_item));
  EXPECT_TRUE(is_checked(always_hide_item));
  EXPECT_FALSE(is_checked(only_on_ntp_item));

  profile_->GetPrefs()->SetInteger(
      bookmarks::prefs::kBookmarkBarVisibilityState,
      static_cast<int>(bookmarks::BookmarkBarVisibilityState::kOnlyShowOnNtp));
  controller.UpdateSubMenuState();

  EXPECT_FALSE(is_checked(always_show_item));
  EXPECT_FALSE(is_checked(always_hide_item));
  EXPECT_TRUE(is_checked(only_on_ntp_item));
}

// Tests that the submenu items are disabled when kBookmarkBarVisibilityState is
// managed.
TEST_P(BookmarkContextMenuUpdateSubMenuStateTest, ManagedVisibilityState) {
  const BookmarkNode* bb_node = model_->bookmark_bar_node();
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      bb_node->children().front().get(),
  };

  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);

  EXPECT_TRUE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_SUBMENU_ALWAYS_SHOW));
  EXPECT_TRUE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_SUBMENU_ALWAYS_HIDE));
  EXPECT_TRUE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_SUBMENU_ONLY_ON_NTP));

  profile_->GetTestingPrefService()->SetManagedPref(
      bookmarks::prefs::kBookmarkBarVisibilityState,
      base::Value(static_cast<int>(
          bookmarks::BookmarkBarVisibilityState::kAlwaysShow)));

  EXPECT_FALSE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_SUBMENU_ALWAYS_SHOW));
  EXPECT_FALSE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_SUBMENU_ALWAYS_HIDE));
  EXPECT_FALSE(
      controller.IsCommandEnabled(IDC_BOOKMARK_BAR_SUBMENU_ONLY_ON_NTP));
}

TEST_P(BookmarkContextMenuUpdateSubMenuStateTest,
       UpdateBookmarkBarVisibilityOnUserAction) {
  const BookmarkNode* bb_node = model_->bookmark_bar_node();
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      bb_node->children().front().get(),
  };

  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kAttachedBar, nodes,
                                 false, false);

  EXPECT_TRUE(
      profile_->GetPrefs()
          ->FindPreference(bookmarks::prefs::kBookmarkBarVisibilityState)
          ->IsDefaultValue());

  controller.ExecuteCommand(IDC_BOOKMARK_BAR_REMOVE, 0);

  EXPECT_FALSE(
      profile_->GetPrefs()
          ->FindPreference(bookmarks::prefs::kBookmarkBarVisibilityState)
          ->IsDefaultValue());
}

// Tests that changing bookmark bar visibility from the context menu logs the
// correct user action metrics.
TEST_P(BookmarkContextMenuUpdateSubMenuStateTest, LogsUserActions) {
  const BookmarkNode* bb_node = model_->bookmark_bar_node();
  std::vector<raw_ptr<const BookmarkNode, VectorExperimental>> nodes = {
      bb_node->children().front().get(),
  };

  BookmarkContextMenu controller(nullptr, nullptr, profile_.get(),
                                 BookmarkLaunchLocation::kNone, nodes, false,
                                 false);

  base::UserActionTester user_action_tester;

  EXPECT_EQ(0, user_action_tester.GetActionCount(
                   "BookmarkBar_ContextMenu_AlwaysShow"));
  controller.ExecuteCommand(IDC_BOOKMARK_BAR_SUBMENU_ALWAYS_SHOW, 0);
  EXPECT_EQ(1, user_action_tester.GetActionCount(
                   "BookmarkBar_ContextMenu_AlwaysShow"));

  EXPECT_EQ(0, user_action_tester.GetActionCount(
                   "BookmarkBar_ContextMenu_AlwaysHide"));
  controller.ExecuteCommand(IDC_BOOKMARK_BAR_SUBMENU_ALWAYS_HIDE, 0);
  EXPECT_EQ(1, user_action_tester.GetActionCount(
                   "BookmarkBar_ContextMenu_AlwaysHide"));

  EXPECT_EQ(0, user_action_tester.GetActionCount(
                   "BookmarkBar_ContextMenu_OnlyShowOnNtp"));
  controller.ExecuteCommand(IDC_BOOKMARK_BAR_SUBMENU_ONLY_ON_NTP, 0);
  EXPECT_EQ(1, user_action_tester.GetActionCount(
                   "BookmarkBar_ContextMenu_OnlyShowOnNtp"));
}

INSTANTIATE_FEATURE_OVERRIDE_TEST_SUITE(BookmarkContextMenuTest);
INSTANTIATE_FEATURE_OVERRIDE_TEST_SUITE(
    BookmarkContextMenuUpdateSubMenuStateTest);