OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/wrench_menu_model.h" | 5 #include "chrome/browser/wrench_menu_model.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "chrome/app/chrome_dll_resource.h" | 8 #include "chrome/app/chrome_dll_resource.h" |
9 #include "chrome/test/browser_with_test_window_test.h" | 9 #include "chrome/test/browser_with_test_window_test.h" |
10 #include "chrome/test/menu_model_test.h" | 10 #include "chrome/test/menu_model_test.h" |
11 #include "grit/generated_resources.h" | 11 #include "grit/generated_resources.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
13 | 13 |
14 class WrenchMenuModelTest : public BrowserWithTestWindowTest, | 14 class WrenchMenuModelTest : public BrowserWithTestWindowTest, |
15 public menus::AcceleratorProvider { | 15 public MenuModelTest { |
16 public: | |
17 // Don't handle accelerators. | |
18 virtual bool GetAcceleratorForCommandId( | |
19 int command_id, | |
20 menus::Accelerator* accelerator) { return false; } | |
21 }; | |
22 | |
23 // Copies parts of MenuModelTest::Delegate and combines them with the | |
24 // WrenchMenuModel since WrenchMenuModel is now a SimpleMenuModel::Delegate and | |
25 // not derived from SimpleMenuModel. | |
26 class TestWrenchMenuModel : public WrenchMenuModel { | |
27 public: | |
28 TestWrenchMenuModel(menus::AcceleratorProvider* provider, | |
29 Browser* browser) | |
30 : WrenchMenuModel(provider, browser), | |
31 execute_count_(0), | |
32 checked_count_(0), | |
33 enable_count_(0) { | |
34 } | |
35 | |
36 // Testing overrides to menus::SimpleMenuModel::Delegate: | |
37 virtual bool IsCommandIdChecked(int command_id) const { | |
38 bool val = WrenchMenuModel::IsCommandIdChecked(command_id); | |
39 if (val) | |
40 checked_count_++; | |
41 return val; | |
42 } | |
43 | |
44 virtual bool IsCommandIdEnabled(int command_id) const { | |
45 ++enable_count_; | |
46 return true; | |
47 } | |
48 | |
49 virtual void ExecuteCommand(int command_id) { ++execute_count_; } | |
50 | |
51 int execute_count_; | |
52 mutable int checked_count_; | |
53 mutable int enable_count_; | |
54 }; | 16 }; |
55 | 17 |
56 TEST_F(WrenchMenuModelTest, Basics) { | 18 TEST_F(WrenchMenuModelTest, Basics) { |
57 TestWrenchMenuModel wrench(this, browser()); | 19 WrenchMenuModel model(&delegate_, browser()); |
58 menus::SimpleMenuModel* model = wrench.menu_model(); | 20 int itemCount = model.GetItemCount(); |
59 int itemCount = model->GetItemCount(); | |
60 | 21 |
61 // Verify it has items. The number varies by platform, so we don't check | 22 // Verify it has items. The number varies by platform, so we don't check |
62 // the exact number. | 23 // the exact number. |
63 EXPECT_GT(itemCount, 10); | 24 EXPECT_GT(itemCount, 10); |
64 | 25 |
65 // Execute a couple of the items and make sure it gets back to our delegate. | 26 // Execute a couple of the items and make sure it gets back to our delegate. |
66 // We can't use CountEnabledExecutable() here because the encoding menu's | 27 // We can't use CountEnabledExecutable() here because the encoding menu's |
67 // delegate is internal, it doesn't use the one we pass in. | 28 // delegate is internal, it doesn't use the one we pass in. |
68 model->ActivatedAt(0); | 29 model.ActivatedAt(0); |
69 EXPECT_TRUE(model->IsEnabledAt(0)); | 30 EXPECT_TRUE(model.IsEnabledAt(0)); |
70 // Make sure to use the index that is not separator in all configurations. | 31 // Make sure to use the index that is not separator in all configurations. |
71 model->ActivatedAt(2); | 32 model.ActivatedAt(2); |
72 EXPECT_TRUE(model->IsEnabledAt(2)); | 33 EXPECT_TRUE(model.IsEnabledAt(2)); |
73 EXPECT_EQ(wrench.execute_count_, 2); | 34 EXPECT_EQ(delegate_.execute_count_, 2); |
74 EXPECT_EQ(wrench.enable_count_, 2); | 35 EXPECT_EQ(delegate_.enable_count_, 2); |
75 | 36 |
76 wrench.execute_count_ = 0; | 37 delegate_.execute_count_ = 0; |
77 wrench.enable_count_ = 0; | 38 delegate_.enable_count_ = 0; |
78 | 39 |
79 // Choose something from the tools submenu and make sure it makes it back to | 40 // Choose something from the tools submenu and make sure it makes it back to |
80 // the delegate as well. Use the first submenu as the tools one. | 41 // the delegate as well. Use the first submenu as the tools one. |
81 int toolsModelIndex = -1; | 42 int toolsModelIndex = -1; |
82 for (int i = 0; i < itemCount; ++i) { | 43 for (int i = 0; i < itemCount; ++i) { |
83 if (model->GetTypeAt(i) == menus::MenuModel::TYPE_SUBMENU) { | 44 if (model.GetTypeAt(i) == menus::MenuModel::TYPE_SUBMENU) { |
84 toolsModelIndex = i; | 45 toolsModelIndex = i; |
85 break; | 46 break; |
86 } | 47 } |
87 } | 48 } |
88 EXPECT_GT(toolsModelIndex, -1); | 49 EXPECT_GT(toolsModelIndex, -1); |
89 menus::MenuModel* toolsModel = model->GetSubmenuModelAt(toolsModelIndex); | 50 menus::MenuModel* toolsModel = model.GetSubmenuModelAt(toolsModelIndex); |
90 EXPECT_TRUE(toolsModel); | 51 EXPECT_TRUE(toolsModel); |
91 EXPECT_GT(toolsModel->GetItemCount(), 2); | 52 EXPECT_GT(toolsModel->GetItemCount(), 2); |
92 toolsModel->ActivatedAt(2); | 53 toolsModel->ActivatedAt(2); |
93 EXPECT_TRUE(toolsModel->IsEnabledAt(2)); | 54 EXPECT_TRUE(toolsModel->IsEnabledAt(2)); |
94 EXPECT_EQ(wrench.execute_count_, 1); | 55 EXPECT_EQ(delegate_.execute_count_, 1); |
95 EXPECT_EQ(wrench.enable_count_, 1); | 56 EXPECT_EQ(delegate_.enable_count_, 1); |
96 } | 57 } |
97 | 58 |
98 class EncodingMenuModelTest : public BrowserWithTestWindowTest, | 59 class EncodingMenuModelTest : public BrowserWithTestWindowTest, |
99 public MenuModelTest { | 60 public MenuModelTest { |
100 }; | 61 }; |
101 | 62 |
102 TEST_F(EncodingMenuModelTest, IsCommandIdCheckedWithNoTabs) { | 63 TEST_F(EncodingMenuModelTest, IsCommandIdCheckedWithNoTabs) { |
103 EncodingMenuModel model(browser()); | 64 EncodingMenuModel model(browser()); |
104 ASSERT_EQ(NULL, browser()->GetSelectedTabContents()); | 65 ASSERT_EQ(NULL, browser()->GetSelectedTabContents()); |
105 EXPECT_FALSE(model.IsCommandIdChecked(IDC_ENCODING_ISO88591)); | 66 EXPECT_FALSE(model.IsCommandIdChecked(IDC_ENCODING_ISO88591)); |
106 } | 67 } |
OLD | NEW |