chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_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_BROWSERTEST_UTIL_H_
   6 #define CHROME_BROWSER_RENDERER_CONTEXT_MENU_RENDER_VIEW_CONTEXT_MENU_BROWSERTEST_UTIL_H_
   7
   8 #include "base/basictypes.h"
   9 #include "base/strings/string16.h"
  10 #include "content/public/browser/notification_observer.h"
  11 #include "content/public/browser/notification_registrar.h"
  12 #include "content/public/common/context_menu_params.h"
  13
  14 class RenderViewContextMenu;
  15
  16 class ContextMenuNotificationObserver : public content::NotificationObserver {
  17  public:
  18   // Wait for a context menu to be shown, and then execute |command_to_execute|.
  19   explicit ContextMenuNotificationObserver(int command_to_execute);
  20   virtual ~ContextMenuNotificationObserver();
  21
  22  private:
  23   virtual void Observe(int type,
  24                        const content::NotificationSource& source,
  25                        const content::NotificationDetails& details) OVERRIDE;
  26
  27   void ExecuteCommand(RenderViewContextMenu* context_menu);
  28
  29   content::NotificationRegistrar registrar_;
  30   int command_to_execute_;
  31
  32   DISALLOW_COPY_AND_ASSIGN(ContextMenuNotificationObserver);
  33 };
  34
  35 class SaveLinkAsContextMenuObserver : public ContextMenuNotificationObserver {
  36  public:
  37   // Register to listen for notifications of
  38   // NOTIFICATION_RENDER_VIEW_CONTEXT_MENU_SHOWN from either
  39   // a specific source, or from all sources if |source| is
  40   // NotificationService::AllSources().
  41   explicit SaveLinkAsContextMenuObserver(
  42       const content::NotificationSource& source);
  43   virtual ~SaveLinkAsContextMenuObserver();
  44
  45   // Suggested filename for file downloaded through "Save Link As" option.
  46   base::string16 GetSuggestedFilename();
  47
  48   // Wait for context menu to be visible.
  49   void WaitForMenu();
  50
  51  private:
  52   virtual void Observe(int type,
  53                        const content::NotificationSource& source,
  54                        const content::NotificationDetails& details) OVERRIDE;
  55
  56   void Cancel(RenderViewContextMenu* context_menu);
  57
  58   bool menu_visible_;
  59   content::ContextMenuParams params_;
  60
  61   DISALLOW_COPY_AND_ASSIGN(SaveLinkAsContextMenuObserver);
  62 };
  63
  64 #endif  // CHROME_BROWSER_RENDERER_CONTEXT_MENU_RENDER_VIEW_CONTEXT_MENU_BROWSERTEST_UTIL_H_