Add frameId to contextMenus.onClicked / onclick.

chrome/browser/extensions/extension_context_menu_browsertest.cc

 // Copyright (c) 2012 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 <stddef.h>
 
+#include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "chrome/browser/extensions/extension_browsertest.h"
 #include "chrome/browser/extensions/lazy_background_page_test_util.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/renderer_context_menu/render_view_context_menu.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/ui_test_utils.h"
+#include "content/public/browser/render_frame_host.h"
 #include "content/public/common/context_menu_params.h"
+#include "content/public/test/browser_test_utils.h"
+#include "extensions/browser/extension_api_frame_id_map.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/browser/process_manager.h"
 #include "extensions/browser/test_management_policy.h"
 #include "extensions/common/extension_set.h"
 #include "extensions/test/extension_test_message_listener.h"
 #include "net/dns/mock_host_resolver.h"
 #include "ui/base/models/menu_model.h"
 
 using content::WebContents;
 using extensions::ContextMenuMatcher;
@@ skipping 73 matching lines @@
   bool MenuHasItemWithLabel(const GURL& page_url,
                             const GURL& link_url,
                             const GURL& frame_url,
                             const std::string& label) {
     scoped_ptr<TestRenderViewContextMenu> menu(
         TestRenderViewContextMenu::Create(
             GetWebContents(), page_url, link_url, frame_url));
     return MenuHasExtensionItemWithLabel(menu.get(), label);
   }
 
+  // Click on a context menu identified by |target_menu_item_id|, and returns
+  // the result of chrome.test.sendMessage. The .js test file that sets up the
+  // context menu should call chrome.test.sendMessage in its onclick event.
+  std::string ClickMenuInFrame(content::RenderFrameHost* frame,
+                               const std::string& target_menu_item_id) {
+    content::ContextMenuParams params;
+    if (frame->GetParent())
+      params.frame_url = frame->GetLastCommittedURL();
+    else
+      params.page_url = frame->GetLastCommittedURL();
+
+    TestRenderViewContextMenu menu(frame, params);
+    menu.Init();
+
+    MenuItem::Id menu_item_id;
+    menu_item_id.string_uid = target_menu_item_id;
+    int command_id = -1;
+    if (!FindCommandId(&menu, menu_item_id, &command_id))
+      return "Menu item not found: " + target_menu_item_id;
+
+    ExtensionTestMessageListener listener(false);
+    menu.ExecuteCommand(command_id, 0);
+    if (!listener.WaitUntilSatisfied())
+      return "Onclick never fired for menu item: " + target_menu_item_id;
+
+    return listener.message();
+  }
+
   // This creates an extension that starts |enabled| and then switches to
   // |!enabled|.
   void TestEnabledContextMenu(bool enabled) {
     ExtensionTestMessageListener begin("begin", true);
     ExtensionTestMessageListener create("create", true);
     ExtensionTestMessageListener update("update", false);
     ASSERT_TRUE(LoadContextMenuExtension("enabled"));
 
     ASSERT_TRUE(begin.WaitUntilSatisfied());
 
@@ skipping 467 matching lines @@
       page_url, GURL(), no_frame_url, std::string("Page item")));
   ASSERT_FALSE(MenuHasItemWithLabel(
       page_url, GURL(), no_frame_url, std::string("Frame item")));
 
   ASSERT_TRUE(MenuHasItemWithLabel(
       page_url, GURL(), frame_url, std::string("Page item")));
   ASSERT_TRUE(MenuHasItemWithLabel(
       page_url, GURL(), frame_url, std::string("Frame item")));
 }
 
+// Tests that info.frameId is correctly set when the context menu is invoked.
+IN_PROC_BROWSER_TEST_F(ExtensionContextMenuBrowserTest, ClickInFrame) {
+  ExtensionTestMessageListener listener("created items", false);
+  ASSERT_TRUE(LoadContextMenuExtension("frames"));
+  GURL url_with_frame("data:text/html,<iframe name='child'>");
+  ui_test_utils::NavigateToURL(browser(), url_with_frame);
+  ASSERT_TRUE(listener.WaitUntilSatisfied());
+
+  // Click on a menu item in the main frame.
+  EXPECT_EQ(
+      "pageUrl=" + url_with_frame.spec() + ", frameUrl=undefined, frameId=0",
+      ClickMenuInFrame(GetWebContents()->GetMainFrame(), "item1"));
+
+  // Click on a menu item in the child frame.
+  content::RenderFrameHost* child_frame = content::FrameMatchingPredicate(
+      GetWebContents(), base::Bind(&content::FrameMatchesName, "child"));
+  ASSERT_TRUE(child_frame);
+  int extension_api_frame_id =
+      extensions::ExtensionApiFrameIdMap::GetFrameId(child_frame);
+  EXPECT_EQ(
+      base::StringPrintf("pageUrl=undefined, frameUrl=about:blank, frameId=%d",
+                         extension_api_frame_id),
+      ClickMenuInFrame(child_frame, "item1"));
+}
+
 // Tests enabling and disabling a context menu item.
 IN_PROC_BROWSER_TEST_F(ExtensionContextMenuBrowserTest, Enabled) {
   TestEnabledContextMenu(true);
   TestEnabledContextMenu(false);
 }
 
 class ExtensionContextMenuBrowserLazyTest :
     public ExtensionContextMenuBrowserTest {
   void SetUpOnMainThread() override {
     ExtensionContextMenuBrowserTest::SetUpOnMainThread();
@@ skipping 48 matching lines @@
   ASSERT_TRUE(LoadContextMenuExtensionIncognito("incognito"));
 
   // Wait for the extension's processes to tell us they've created an item.
   ASSERT_TRUE(created.WaitUntilSatisfied());
   ASSERT_TRUE(created_incognito.WaitUntilSatisfied());
   ASSERT_EQ(2u, GetItems().size());
 
   browser()->profile()->DestroyOffTheRecordProfile();
   ASSERT_EQ(1u, GetItems().size());
 }