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