chrome/browser/renderer_context_menu/render_view_context_menu_test_util.h

   1 // Copyright 2014 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 #ifndef CHROME_BROWSER_RENDERER_CONTEXT_MENU_RENDER_VIEW_CONTEXT_MENU_TEST_UTIL_H_
   6 #define CHROME_BROWSER_RENDERER_CONTEXT_MENU_RENDER_VIEW_CONTEXT_MENU_TEST_UTIL_H_
   7
   8 #include "base/basictypes.h"
   9 #include "chrome/browser/extensions/context_menu_matcher.h"
  10 #include "chrome/browser/renderer_context_menu/render_view_context_menu.h"
  11 #include "url/gurl.h"
  12
  13 class Browser;
  14
  15 namespace content {
  16 class WebContents;
  17 }
  18 namespace ui {
  19 class MenuModel;
  20 }
  21
  22 class TestRenderViewContextMenu : public RenderViewContextMenu {
  23  public:
  24   TestRenderViewContextMenu(content::RenderFrameHost* render_frame_host,
  25                             content::ContextMenuParams params);
  26   virtual ~TestRenderViewContextMenu();
  27
  28   // Factory.
  29   // This is a lightweight method to create a test RenderViewContextMenu
  30   // instance.
  31   // Use the constructor if you want to create menu with fine-grained params.
  32   static TestRenderViewContextMenu* Create(content::WebContents* web_contents,
  33                                            const GURL& page_url,
  34                                            const GURL& link_url,
  35                                            const GURL& frame_url);
  36
  37   // Implementation of pure virtuals in RenderViewContextMenu.
  38   virtual void PlatformInit() OVERRIDE;
  39   virtual void PlatformCancel() OVERRIDE;
  40   virtual bool GetAcceleratorForCommandId(
  41       int command_id,
  42       ui::Accelerator* accelerator) OVERRIDE;
  43
  44   // Returns true if command specified by |command_id| is present
  45   // in the menu.
  46   // List of command ids can be found in chrome/app/chrome_command_ids.h.
  47   bool IsItemPresent(int command_id);
  48
  49   // Searches for an menu item with |command_id|. If it's found, the return
  50   // value is true and the model and index where it appears in that model are
  51   // returned in |found_model| and |found_index|. Otherwise returns false.
  52   bool GetMenuModelAndItemIndex(int command_id,
  53                                 ui::MenuModel** found_model,
  54                                 int* found_index);
  55
  56   extensions::ContextMenuMatcher& extension_items() { return extension_items_; }
  57
  58  private:
  59   DISALLOW_COPY_AND_ASSIGN(TestRenderViewContextMenu);
  60 };
  61
  62 #endif  // CHROME_BROWSER_RENDERER_CONTEXT_MENU_RENDER_VIEW_CONTEXT_MENU_TEST_UTIL_H_