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
14class RenderViewContextMenu;
15
16class 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
35class 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_
65