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
3 // found in the LICENSE file.
4
5 #include "chrome/browser/ui/tabs/tab_menu_model.h"
6
7 #include "base/test/scoped_feature_list.h"
8 #include "chrome/browser/ui/browser_commands.h"
9 #include "chrome/browser/ui/tabs/existing_base_sub_menu_model.h"
10 #include "chrome/browser/ui/ui_features.h"
11 #include "chrome/test/base/browser_with_test_window_test.h"
12 #include "chrome/test/base/menu_model_test.h"
13 #include "testing/gtest/include/gtest/gtest.h"
14 #include "ui/base/models/menu_model.h"
15
16 class TabMenuModelTest : public MenuModelTest,
17 public BrowserWithTestWindowTest {
18 };
19
TEST_F(TabMenuModelTest,Basics)20 TEST_F(TabMenuModelTest, Basics) {
21 chrome::NewTab(browser());
22 TabMenuModel model(&delegate_, browser()->tab_strip_model(), 0);
23
24 // Verify it has items. The number varies by platform, so we don't check
25 // the exact number.
26 EXPECT_GT(model.GetItemCount(), 5);
27
28 int item_count = 0;
29 CountEnabledExecutable(&model, &item_count);
30 EXPECT_GT(item_count, 0);
31 EXPECT_EQ(item_count, delegate_.execute_count_);
32 EXPECT_EQ(item_count, delegate_.enable_count_);
33 }
34
TEST_F(TabMenuModelTest,MoveToNewWindow)35 TEST_F(TabMenuModelTest, MoveToNewWindow) {
36 chrome::NewTab(browser());
37 TabMenuModel model(&delegate_, browser()->tab_strip_model(), 0);
38
39 // Verify that CommandMoveTabsToNewWindow is in the menu.
40 EXPECT_GT(
41 model.GetIndexOfCommandId(TabStripModel::CommandMoveTabsToNewWindow), -1);
42 }
43
TEST_F(TabMenuModelTest,AddToExistingGroupSubmenu)44 TEST_F(TabMenuModelTest, AddToExistingGroupSubmenu) {
45 chrome::NewTab(browser());
46 chrome::NewTab(browser());
47 chrome::NewTab(browser());
48 chrome::NewTab(browser());
49
50 TabStripModel* tab_strip_model = browser()->tab_strip_model();
51
52 tab_strip_model->AddToNewGroup({0});
53 tab_strip_model->AddToNewGroup({1});
54 tab_strip_model->AddToNewGroup({2});
55
56 TabMenuModel menu(&delegate_, tab_strip_model, 3);
57
58 int submenu_index =
59 menu.GetIndexOfCommandId(TabStripModel::CommandAddToExistingGroup);
60 ui::MenuModel* submenu = menu.GetSubmenuModelAt(submenu_index);
61
62 EXPECT_TRUE(submenu->HasIcons());
63 EXPECT_EQ(submenu->GetItemCount(), 5);
64 EXPECT_EQ(submenu->GetCommandIdAt(0),
65 ExistingBaseSubMenuModel::kMinExistingTabGroupCommandId);
66 EXPECT_EQ(submenu->GetTypeAt(1), ui::MenuModel::TYPE_SEPARATOR);
67 EXPECT_EQ(submenu->GetCommandIdAt(2),
68 ExistingBaseSubMenuModel::kMinExistingTabGroupCommandId + 1);
69 EXPECT_EQ(submenu->GetCommandIdAt(3),
70 ExistingBaseSubMenuModel::kMinExistingTabGroupCommandId + 2);
71 EXPECT_EQ(submenu->GetCommandIdAt(4),
72 ExistingBaseSubMenuModel::kMinExistingTabGroupCommandId + 3);
73 }
74
TEST_F(TabMenuModelTest,AddToExistingGroupSubmenu_DoesNotIncludeCurrentGroup)75 TEST_F(TabMenuModelTest, AddToExistingGroupSubmenu_DoesNotIncludeCurrentGroup) {
76 chrome::NewTab(browser());
77 chrome::NewTab(browser());
78 chrome::NewTab(browser());
79 chrome::NewTab(browser());
80
81 TabStripModel* tab_strip_model = browser()->tab_strip_model();
82
83 tab_strip_model->AddToNewGroup({0});
84 tab_strip_model->AddToNewGroup({1});
85 tab_strip_model->AddToNewGroup({2});
86
87 TabMenuModel menu(&delegate_, tab_strip_model, 1);
88
89 int submenu_index =
90 menu.GetIndexOfCommandId(TabStripModel::CommandAddToExistingGroup);
91 ui::MenuModel* submenu = menu.GetSubmenuModelAt(submenu_index);
92
93 EXPECT_TRUE(submenu->HasIcons());
94 EXPECT_EQ(submenu->GetItemCount(), 4);
95 EXPECT_EQ(submenu->GetCommandIdAt(0),
96 ExistingBaseSubMenuModel::kMinExistingTabGroupCommandId);
97 EXPECT_EQ(submenu->GetTypeAt(1), ui::MenuModel::TYPE_SEPARATOR);
98 EXPECT_EQ(submenu->GetCommandIdAt(2),
99 ExistingBaseSubMenuModel::kMinExistingTabGroupCommandId + 1);
100 EXPECT_EQ(submenu->GetCommandIdAt(3),
101 ExistingBaseSubMenuModel::kMinExistingTabGroupCommandId + 2);
102 }
103