Use suggested filename for "Save Link As"

chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.h"
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
+#include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/renderer_context_menu/render_view_context_menu.h"
 #include "content/public/browser/notification_service.h"
+#include "content/public/test/test_utils.h"
 
 ContextMenuNotificationObserver::ContextMenuNotificationObserver(
     int command_to_execute)
     : command_to_execute_(command_to_execute) {
   registrar_.Add(this,
                  chrome::NOTIFICATION_RENDER_VIEW_CONTEXT_MENU_SHOWN,
                  content::NotificationService::AllSources());
 }
 
 ContextMenuNotificationObserver::~ContextMenuNotificationObserver() {
 }
 
 void ContextMenuNotificationObserver::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   switch (type) {
     case chrome::NOTIFICATION_RENDER_VIEW_CONTEXT_MENU_SHOWN: {
       RenderViewContextMenu* context_menu =
           content::Source<RenderViewContextMenu>(source).ptr();
       base::MessageLoop::current()->PostTask(
           FROM_HERE,
           base::Bind(&ContextMenuNotificationObserver::ExecuteCommand,
-                     base::Unretained(this), context_menu));
+                     base::Unretained(this),
+                     context_menu));
       break;
     }
 
     default:
       NOTREACHED();
   }
 }
 
 void ContextMenuNotificationObserver::ExecuteCommand(
     RenderViewContextMenu* context_menu) {
   context_menu->ExecuteCommand(command_to_execute_, 0);
   context_menu->Cancel();
 }
+
+SaveLinkAsContextMenuObserver::SaveLinkAsContextMenuObserver(
+    const content::NotificationSource& source)
+    : ContextMenuNotificationObserver(IDC_CONTENT_CONTEXT_SAVELINKAS),
+      menu_visible_(false) {
+}
+
+SaveLinkAsContextMenuObserver::~SaveLinkAsContextMenuObserver() {
+}
+
+void SaveLinkAsContextMenuObserver::Observe(
+    int type,
+    const content::NotificationSource& source,
+    const content::NotificationDetails& details) {
+  switch (type) {
+    case chrome::NOTIFICATION_RENDER_VIEW_CONTEXT_MENU_SHOWN: {
+      menu_visible_ = true;
+      RenderViewContextMenu* context_menu =
+          content::Source<RenderViewContextMenu>(source).ptr();
+      base::MessageLoop::current()->PostTask(
+          FROM_HERE,
+          base::Bind(&SaveLinkAsContextMenuObserver::Cancel,
+                     base::Unretained(this),
+                     context_menu));
+      break;
+    }
+
+    default:
+      NOTREACHED();
+  }
+}
+
+void SaveLinkAsContextMenuObserver::WaitForMenu() {
+  content::WindowedNotificationObserver menu_observer(
+      chrome::NOTIFICATION_RENDER_VIEW_CONTEXT_MENU_SHOWN,
+      content::NotificationService::AllSources());
+  if (!menu_visible_)
+    menu_observer.Wait();
+  menu_visible_ = false;
+}
+
+base::string16 SaveLinkAsContextMenuObserver::GetSuggestedFilename() {
+  return params_.suggested_filename;
+}
+
+void SaveLinkAsContextMenuObserver::Cancel(
+    RenderViewContextMenu* context_menu) {
+  params_ = context_menu->params();
+  context_menu->Cancel();
+}