Index: chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.h |
diff --git a/chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.h b/chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.h |
index cb581748c5b1884161608e7fbeeb38b77af93b51..8fdeef3746d6b1fdf1ed9a981ac6842a9b912e76 100644 |
--- a/chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.h |
+++ b/chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.h |
@@ -6,8 +6,10 @@ |
#define CHROME_BROWSER_RENDERER_CONTEXT_MENU_RENDER_VIEW_CONTEXT_MENU_BROWSERTEST_UTIL_H_ |
#include "base/basictypes.h" |
+#include "base/strings/string16.h" |
#include "content/public/browser/notification_observer.h" |
#include "content/public/browser/notification_registrar.h" |
+#include "content/public/common/context_menu_params.h" |
class RenderViewContextMenu; |
@@ -30,4 +32,33 @@ class ContextMenuNotificationObserver : public content::NotificationObserver { |
DISALLOW_COPY_AND_ASSIGN(ContextMenuNotificationObserver); |
}; |
+class SaveLinkAsContextMenuObserver : public ContextMenuNotificationObserver { |
+ public: |
+ // Register to listen for notifications of |
+ // NOTIFICATION_RENDER_VIEW_CONTEXT_MENU_SHOWN from either |
+ // a specific source, or from all sources if |source| is |
+ // NotificationService::AllSources(). |
+ explicit SaveLinkAsContextMenuObserver( |
+ const content::NotificationSource& source); |
+ virtual ~SaveLinkAsContextMenuObserver(); |
+ |
+ // Suggested filename for file downloaded through "Save Link As" option. |
+ base::string16 GetSuggestedFilename(); |
+ |
+ // Wait for context menu to be visible. |
+ void WaitForMenu(); |
+ |
+ private: |
+ virtual void Observe(int type, |
+ const content::NotificationSource& source, |
+ const content::NotificationDetails& details) OVERRIDE; |
+ |
+ void Cancel(RenderViewContextMenu* context_menu); |
+ |
+ bool menu_visible_; |
+ content::ContextMenuParams params_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(SaveLinkAsContextMenuObserver); |
+}; |
+ |
#endif // CHROME_BROWSER_RENDERER_CONTEXT_MENU_RENDER_VIEW_CONTEXT_MENU_BROWSERTEST_UTIL_H_ |