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 executeCustomMenuCommandAndVerifyTabTitle(unsigned command_id,

[brettw, 2014/09/02 23:50:29]

The formatting of this is all messed up. Also: function names should be
uppercase.

[pals, 2014/09/03 17:10:57]

Done.

+      const std::string& expected) {
+  content::WebContents* tab =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  // Open context menu and execute |command_id|

[brettw, 2014/09/02 23:50:29]

This context menu code is almost an exact copy of some code below. Is it
reasonable to share this code somehow?

[pals, 2014/09/03 17:10:57]

Done.

+  blink::WebMouseEvent mouse_event;
+  mouse_event.button = blink::WebMouseEvent::ButtonRight;
+  mouse_event.x = 50;
+  mouse_event.y = 50;
+  gfx::Rect offset = tab->GetContainerBounds();
+  mouse_event.globalX = 50 + offset.x();
+  mouse_event.globalY = 50 + offset.y();
+  mouse_event.clickCount = 1;
+  // menu_observer executes the menu item command with |command_id|.
+  ContextMenuNotificationObserver menu_observer(
+      command_id);
+  mouse_event.type = blink::WebInputEvent::MouseDown;
+  tab->GetRenderViewHost()->ForwardMouseEvent(mouse_event);
+  mouse_event.type = blink::WebInputEvent::MouseUp;
+  tab->GetRenderViewHost()->ForwardMouseEvent(mouse_event);
+
+  // The menu_observer will execute the action corresponding to |command_id|,
+  // wait for the title to change.
+  base::string16 title = base::ASCIIToUTF16(expected);
+  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 55 matching lines @@
   // 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 142 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