Testing contextmenu attribute which enables webpage to add custom menu items to the platform contex…

chrome/browser/renderer_context_menu/render_view_context_menu_browsertest.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 <string>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/string16.h"
 #include "base/strings/utf_string_conversions.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 "chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.h"
 #include "chrome/browser/renderer_context_menu/render_view_context_menu_test_util.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/browser_test_utils.h"
+#include "net/test/embedded_test_server/embedded_test_server.h"
 #include "third_party/WebKit/public/web/WebContextMenuData.h"
 #include "third_party/WebKit/public/web/WebInputEvent.h"
 
 using content::WebContents;
 
 namespace {
 
 class ContextMenuBrowserTest : public InProcessBrowserTest {
  public:
   ContextMenuBrowserTest() { }
@@ skipping 10 matching lines @@
     params.writing_direction_default = 0;
     params.writing_direction_left_to_right = 0;
     params.writing_direction_right_to_left = 0;
 #endif  // OS_MACOSX
     TestRenderViewContextMenu* menu = new TestRenderViewContextMenu(
         browser()->tab_strip_model()->GetActiveWebContents()->GetMainFrame(),
         params);
     menu->Init();
     return menu;
   }
+
+  void OpenContextMenuAtPoint(int x, int y) {
+    content::WebContents* tab =
+        browser()->tab_strip_model()->GetActiveWebContents();
+    // Open context menu
+    blink::WebMouseEvent mouse_event;
+    mouse_event.button = blink::WebMouseEvent::ButtonRight;
+    mouse_event.x = x;
+    mouse_event.y = y;
+    gfx::Rect offset = tab->GetContainerBounds();
+    mouse_event.globalX = x + offset.x();
+    mouse_event.globalY = y + offset.y();
+    mouse_event.clickCount = 1;
+    mouse_event.type = blink::WebInputEvent::MouseDown;
+    tab->GetRenderViewHost()->ForwardMouseEvent(mouse_event);
+    mouse_event.type = blink::WebInputEvent::MouseUp;
+    tab->GetRenderViewHost()->ForwardMouseEvent(mouse_event);
+  }
+
+  void ExecuteCustomMenuCommandAndVerifyTabTitle(unsigned command_id,
+                                                 const std::string& expected) {
+    // menu_observer executes the menu item command with |command_id|.
+    ContextMenuNotificationObserver menu_observer(command_id);
+    // Right click on the center (50, 50) of the div with dimension (100, 100).
+    OpenContextMenuAtPoint(50, 50);
+
+    // The menu_observer will execute the action corresponding to |command_id|,
+    // wait for the title to change.
+    base::string16 title = base::ASCIIToUTF16(expected);
+    content::WebContents* tab =
+        browser()->tab_strip_model()->GetActiveWebContents();
+    content::TitleWatcher title_watcher(tab, title);
+    ASSERT_EQ(title, title_watcher.WaitAndGetTitle());
+
+    // Verify that it's the correct tab.
+    EXPECT_EQ(base::ASCIIToUTF16(expected), tab->GetTitle());
+  }
 };
 
 IN_PROC_BROWSER_TEST_F(ContextMenuBrowserTest,
                        OpenEntryPresentForNormalURLs) {
   scoped_ptr<TestRenderViewContextMenu> menu(
       CreateContextMenu(GURL("http://www.google.com/"),
                         GURL("http://www.google.com/")));
 
   ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKNEWTAB));
   ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKNEWWINDOW));
@@ skipping 30 matching lines @@
   ContextMenuNotificationObserver menu_observer(
       IDC_CONTENT_CONTEXT_OPENLINKNEWTAB);
   ui_test_utils::WindowedTabAddedNotificationObserver tab_observer(
       content::NotificationService::AllSources());
 
   // Go to a page with a link
   ui_test_utils::NavigateToURL(
       browser(), GURL("data:text/html,<a href='about:blank'>link</a>"));
 
   // Open a context menu.
-  blink::WebMouseEvent mouse_event;
-  mouse_event.type = blink::WebInputEvent::MouseDown;
-  mouse_event.button = blink::WebMouseEvent::ButtonRight;
-  mouse_event.x = 15;
-  mouse_event.y = 15;
+  OpenContextMenuAtPoint(15, 15);
   content::WebContents* tab =
       browser()->tab_strip_model()->GetActiveWebContents();
-  gfx::Rect offset = tab->GetContainerBounds();
-  mouse_event.globalX = 15 + offset.x();
-  mouse_event.globalY = 15 + offset.y();
-  mouse_event.clickCount = 1;
-  tab->GetRenderViewHost()->ForwardMouseEvent(mouse_event);
-  mouse_event.type = blink::WebInputEvent::MouseUp;
-  tab->GetRenderViewHost()->ForwardMouseEvent(mouse_event);
 
   // The menu_observer will select "Open in new tab", wait for the new tab to
   // be added.
   tab_observer.Wait();
   tab = tab_observer.GetTab();
   content::WaitForLoadStop(tab);
 
   // Verify that it's the correct tab.
   EXPECT_EQ(GURL("about:blank"), tab->GetURL());
 }
 
+IN_PROC_BROWSER_TEST_F(ContextMenuBrowserTest, CustomMenu) {
+  ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
+  // Go to a test page
+  GURL url = embedded_test_server()->GetURL(
+      "/context_menu/custom_context_menu.html");
+  ui_test_utils::NavigateToURL(browser(), url);
+
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST, "Title 1");
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST + 1, "Title 2");
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST + 2, "Title 3");
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST + 3, "Title 4");
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST + 4, "Title 5");
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST + 5, "Title 6");
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST + 6, "Title 7");
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST + 7, "Title 8");
+  ExecuteCustomMenuCommandAndVerifyTabTitle(
+      IDC_CONTENT_CONTEXT_CUSTOM_FIRST + 8, "Title 9");
+}
+
 // Verify that "Open Link in New Tab" doesn't send URL fragment as referrer.
 IN_PROC_BROWSER_TEST_F(ContextMenuBrowserTest, OpenInNewTabReferrer) {
   ui_test_utils::WindowedTabAddedNotificationObserver tab_observer(
       content::NotificationService::AllSources());
 
   ASSERT_TRUE(test_server()->Start());
   GURL echoheader(test_server()->GetURL("echoheader?Referer"));
 
   // Go to a |page| with a link to echoheader URL.
   GURL page("data:text/html,<a href='" + echoheader.spec() + "'>link</a>");
@@ skipping 97 matching lines @@
       content::NotificationService::AllSources());
 
   // Go to a page with a link having download attribute.
   const std::string kSuggestedFilename("test_filename.png");
   ui_test_utils::NavigateToURL(
       browser(),
       GURL("data:text/html,<a href='about:blank' download='" +
            kSuggestedFilename + "'>link</a>"));
 
   // Open a context menu.
-  blink::WebMouseEvent mouse_event;
-  mouse_event.type = blink::WebInputEvent::MouseDown;
-  mouse_event.button = blink::WebMouseEvent::ButtonRight;
-  mouse_event.x = 15;
-  mouse_event.y = 15;
-  content::WebContents* tab =
-      browser()->tab_strip_model()->GetActiveWebContents();
-  tab->GetRenderViewHost()->ForwardMouseEvent(mouse_event);
-  mouse_event.type = blink::WebInputEvent::MouseUp;
-  tab->GetRenderViewHost()->ForwardMouseEvent(mouse_event);
+  OpenContextMenuAtPoint(15, 15);
 
   // Wait for context menu to be visible.
   menu_observer.WaitForMenu();
 
   // Compare filename.
   base::string16 suggested_filename = menu_observer.GetSuggestedFilename();
   ASSERT_EQ(kSuggestedFilename, base::UTF16ToUTF8(suggested_filename).c_str());
 }
 
 // Ensure that View Page Info won't crash if there is no visible entry.
@@ skipping 15 matching lines @@
   menu.Init();
 
   // The item shouldn't be enabled in the menu.
   EXPECT_FALSE(menu.IsCommandIdEnabled(IDC_CONTENT_CONTEXT_VIEWPAGEINFO));
 
   // Ensure that viewing page info doesn't crash even if you can get to it.
   menu.ExecuteCommand(IDC_CONTENT_CONTEXT_VIEWPAGEINFO, 0);
 }
 
 }  // namespace