15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef CHROME_TEST_BASE_MENU_MODEL_TEST_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define CHROME_TEST_BASE_MENU_MODEL_TEST_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/accelerators/accelerator.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/models/simple_menu_model.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A mix-in class to be used in addition to something that derrives from 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// testing::Test to provide some extra functionality for testing menu models. 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MenuModelTest { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MenuModelTest() {} 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MenuModelTest() {} 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A menu delegate that counts the number of times certain things are called 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to make sure things are hooked up properly. 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) class Delegate : public ui::SimpleMenuModel::Delegate, 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public ui::AcceleratorProvider { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Delegate() : execute_count_(0), enable_count_(0) {} 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsCommandIdChecked(int command_id) const OVERRIDE; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool IsCommandIdEnabled(int command_id) const OVERRIDE; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual bool GetAcceleratorForCommandId( 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int command_id, 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::Accelerator* accelerator) OVERRIDE; 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void ExecuteCommand(int command_id, int event_flags) OVERRIDE; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int execute_count_; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mutable int enable_count_; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Recursively checks the enabled state and executes a command on every item 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // that's not a separator or a submenu parent item. The returned count should 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // match the number of times the delegate is called to ensure every item 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // works. 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void CountEnabledExecutable(ui::MenuModel* model, int* count); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Delegate delegate_; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // CHROME_TEST_BASE_MENU_MODEL_TEST_H_ 47