chrome/browser/custom_handlers/protocol_handler_registry_browsertest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <string>
   6
   7 #include "base/memory/scoped_ptr.h"
   8 #include "base/strings/string16.h"
   9 #include "base/strings/utf_string_conversions.h"
  10 #include "chrome/app/chrome_command_ids.h"
  11 #include "chrome/browser/custom_handlers/protocol_handler_registry_factory.h"
  12 #include "chrome/browser/renderer_context_menu/render_view_context_menu_test_util.h"
  13 #include "chrome/browser/ui/browser.h"
  14 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  15 #include "chrome/test/base/in_process_browser_test.h"
  16 #include "chrome/test/base/ui_test_utils.h"
  17 #include "content/public/browser/navigation_controller.h"
  18 #include "content/public/browser/navigation_entry.h"
  19 #include "content/public/browser/web_contents.h"
  20 #include "third_party/WebKit/public/web/WebContextMenuData.h"
  21
  22 using content::WebContents;
  23
  24 class RegisterProtocolHandlerBrowserTest : public InProcessBrowserTest {
  25  public:
  26   RegisterProtocolHandlerBrowserTest() { }
  27
  28   TestRenderViewContextMenu* CreateContextMenu(GURL url) {
  29     content::ContextMenuParams params;
  30     params.media_type = blink::WebContextMenuData::MediaTypeNone;
  31     params.link_url = url;
  32     params.unfiltered_link_url = url;
  33     WebContents* web_contents =
  34         browser()->tab_strip_model()->GetActiveWebContents();
  35     params.page_url = web_contents->GetController().GetActiveEntry()->GetURL();
  36 #if defined(OS_MACOSX)
  37     params.writing_direction_default = 0;
  38     params.writing_direction_left_to_right = 0;
  39     params.writing_direction_right_to_left = 0;
  40 #endif  // OS_MACOSX
  41     TestRenderViewContextMenu* menu = new TestRenderViewContextMenu(
  42         browser()->tab_strip_model()->GetActiveWebContents()->GetMainFrame(),
  43         params);
  44     menu->Init();
  45     return menu;
  46   }
  47
  48   void AddProtocolHandler(const std::string& protocol,
  49                           const GURL& url) {
  50     ProtocolHandler handler = ProtocolHandler::CreateProtocolHandler(protocol,
  51                                                                      url);
  52     ProtocolHandlerRegistry* registry =
  53         ProtocolHandlerRegistryFactory::GetForProfile(browser()->profile());
  54     // Fake that this registration is happening on profile startup. Otherwise
  55     // it'll try to register with the OS, which causes DCHECKs on Windows when
  56     // running as admin on Windows 7.
  57     registry->is_loading_ = true;
  58     registry->OnAcceptRegisterProtocolHandler(handler);
  59     registry->is_loading_ = false;
  60     ASSERT_TRUE(registry->IsHandledProtocol(protocol));
  61   }
  62 };
  63
  64 IN_PROC_BROWSER_TEST_F(RegisterProtocolHandlerBrowserTest,
  65     ContextMenuEntryAppearsForHandledUrls) {
  66   scoped_ptr<TestRenderViewContextMenu> menu(
  67       CreateContextMenu(GURL("http://www.google.com/")));
  68   ASSERT_FALSE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKWITH));
  69
  70   AddProtocolHandler(std::string("web+search"),
  71                      GURL("http://www.google.com/%s"));
  72   GURL url("web+search:testing");
  73   ProtocolHandlerRegistry* registry =
  74       ProtocolHandlerRegistryFactory::GetForProfile(browser()->profile());
  75   ASSERT_EQ(1u, registry->GetHandlersFor(url.scheme()).size());
  76   menu.reset(CreateContextMenu(url));
  77   ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKWITH));
  78 }
  79
  80 IN_PROC_BROWSER_TEST_F(RegisterProtocolHandlerBrowserTest, CustomHandler) {
  81   ASSERT_TRUE(test_server()->Start());
  82   GURL handler_url = test_server()->GetURL("files/custom_handler_foo.html");
  83   AddProtocolHandler("foo", handler_url);
  84
  85   ui_test_utils::NavigateToURL(browser(), GURL("foo:test"));
  86
  87   ASSERT_EQ(handler_url,
  88             browser()->tab_strip_model()->GetActiveWebContents()->GetURL());
  89 }