| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #ifndef CHROME_TEST_BASE_MENU_MODEL_TEST_H_ | 5 #ifndef CHROME_TEST_BASE_MENU_MODEL_TEST_H_ |
| 6 #define CHROME_TEST_BASE_MENU_MODEL_TEST_H_ | 6 #define CHROME_TEST_BASE_MENU_MODEL_TEST_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 #include "ui/base/models/accelerator.h" | 9 #include "ui/base/models/accelerator.h" |
| 10 #include "ui/base/models/simple_menu_model.h" | 10 #include "ui/base/models/simple_menu_model.h" |
| 11 | 11 |
| 12 // A mix-in class to be used in addition to something that derrives from | 12 // A mix-in class to be used in addition to something that derrives from |
| 13 // testing::Test to provide some extra functionality for testing menu models. | 13 // testing::Test to provide some extra functionality for testing menu models. |
| 14 class MenuModelTest { | 14 class MenuModelTest { |
| 15 public: | 15 public: |
| 16 MenuModelTest() { } | 16 MenuModelTest() { } |
| 17 virtual ~MenuModelTest() { } | 17 virtual ~MenuModelTest() { } |
| 18 | 18 |
| 19 protected: | 19 protected: |
| 20 // A menu delegate that counts the number of times certain things are called | 20 // A menu delegate that counts the number of times certain things are called |
| 21 // to make sure things are hooked up properly. | 21 // to make sure things are hooked up properly. |
| 22 class Delegate : public ui::SimpleMenuModel::Delegate, | 22 class Delegate : public ui::SimpleMenuModel::Delegate, |
| 23 public ui::AcceleratorProvider { | 23 public ui::AcceleratorProvider { |
| 24 public: | 24 public: |
| 25 Delegate() : execute_count_(0), enable_count_(0) { } | 25 Delegate() : execute_count_(0), enable_count_(0) { } |
| 26 | 26 |
| 27 virtual bool IsCommandIdChecked(int command_id) const { return false; } | 27 virtual bool IsCommandIdChecked(int command_id) const OVERRIDE { |
| 28 virtual bool IsCommandIdEnabled(int command_id) const { | 28 return false; |
| 29 } |
| 30 virtual bool IsCommandIdEnabled(int command_id) const OVERRIDE { |
| 29 ++enable_count_; | 31 ++enable_count_; |
| 30 return true; | 32 return true; |
| 31 } | 33 } |
| 32 virtual bool GetAcceleratorForCommandId( | 34 virtual bool GetAcceleratorForCommandId( |
| 33 int command_id, | 35 int command_id, |
| 34 ui::Accelerator* accelerator) { return false; } | 36 ui::Accelerator* accelerator) OVERRIDE { return false; } |
| 35 virtual void ExecuteCommand(int command_id) { ++execute_count_; } | 37 virtual void ExecuteCommand(int command_id) OVERRIDE { ++execute_count_; } |
| 36 | 38 |
| 37 int execute_count_; | 39 int execute_count_; |
| 38 mutable int enable_count_; | 40 mutable int enable_count_; |
| 39 }; | 41 }; |
| 40 | 42 |
| 41 // Recursively checks the enabled state and executes a command on every item | 43 // Recursively checks the enabled state and executes a command on every item |
| 42 // that's not a separator or a submenu parent item. The returned count should | 44 // that's not a separator or a submenu parent item. The returned count should |
| 43 // match the number of times the delegate is called to ensure every item | 45 // match the number of times the delegate is called to ensure every item |
| 44 // works. | 46 // works. |
| 45 void CountEnabledExecutable(ui::MenuModel* model, int* count); | 47 void CountEnabledExecutable(ui::MenuModel* model, int* count); |
| 46 | 48 |
| 47 Delegate delegate_; | 49 Delegate delegate_; |
| 48 }; | 50 }; |
| 49 | 51 |
| 50 #endif // CHROME_TEST_BASE_MENU_MODEL_TEST_H_ | 52 #endif // CHROME_TEST_BASE_MENU_MODEL_TEST_H_ |
| OLD | NEW |