src/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/tab_contents/render_view_context_menu.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 namespace {
  25
  26 class TestRenderViewContextMenu : public RenderViewContextMenu {
  27  public:
  28   TestRenderViewContextMenu(WebContents* web_contents,
  29                             content::ContextMenuParams params)
  30       : RenderViewContextMenu(web_contents, params) { }
  31
  32   virtual void PlatformInit() OVERRIDE { }
  33   virtual void PlatformCancel() OVERRIDE { }
  34   virtual bool GetAcceleratorForCommandId(
  35       int command_id,
  36       ui::Accelerator* accelerator) OVERRIDE {
  37     return false;
  38   }
  39
  40   bool IsItemPresent(int command_id) {
  41     return menu_model_.GetIndexOfCommandId(command_id) != -1;
  42   }
  43 };
  44
  45 }  // namespace
  46
  47 class RegisterProtocolHandlerBrowserTest : public InProcessBrowserTest {
  48  public:
  49   RegisterProtocolHandlerBrowserTest() { }
  50
  51   TestRenderViewContextMenu* CreateContextMenu(GURL url) {
  52     content::ContextMenuParams params;
  53     params.media_type = blink::WebContextMenuData::MediaTypeNone;
  54     params.link_url = url;
  55     params.unfiltered_link_url = url;
  56     WebContents* web_contents =
  57         browser()->tab_strip_model()->GetActiveWebContents();
  58     params.page_url = web_contents->GetController().GetActiveEntry()->GetURL();
  59 #if defined(OS_MACOSX)
  60     params.writing_direction_default = 0;
  61     params.writing_direction_left_to_right = 0;
  62     params.writing_direction_right_to_left = 0;
  63 #endif  // OS_MACOSX
  64     TestRenderViewContextMenu* menu = new TestRenderViewContextMenu(
  65         browser()->tab_strip_model()->GetActiveWebContents(), params);
  66     menu->Init();
  67     return menu;
  68   }
  69
  70   void AddProtocolHandler(const std::string& protocol,
  71                           const GURL& url,
  72                           const base::string16& title) {
  73     ProtocolHandler handler = ProtocolHandler::CreateProtocolHandler(
  74           protocol, url, title);
  75     ProtocolHandlerRegistry* registry =
  76         ProtocolHandlerRegistryFactory::GetForProfile(browser()->profile());
  77     // Fake that this registration is happening on profile startup. Otherwise
  78     // it'll try to register with the OS, which causes DCHECKs on Windows when
  79     // running as admin on Windows 7.
  80     registry->is_loading_ = true;
  81     registry->OnAcceptRegisterProtocolHandler(handler);
  82     registry->is_loading_ = false;
  83     ASSERT_TRUE(registry->IsHandledProtocol(protocol));
  84   }
  85 };
  86
  87 IN_PROC_BROWSER_TEST_F(RegisterProtocolHandlerBrowserTest,
  88     ContextMenuEntryAppearsForHandledUrls) {
  89   scoped_ptr<TestRenderViewContextMenu> menu(
  90       CreateContextMenu(GURL("http://www.google.com/")));
  91   ASSERT_FALSE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKWITH));
  92
  93   AddProtocolHandler(std::string("web+search"),
  94                      GURL("http://www.google.com/%s"),
  95                      base::UTF8ToUTF16(std::string("Test handler")));
  96   GURL url("web+search:testing");
  97   ProtocolHandlerRegistry* registry =
  98       ProtocolHandlerRegistryFactory::GetForProfile(browser()->profile());
  99   ASSERT_EQ(1u, registry->GetHandlersFor(url.scheme()).size());
 100   menu.reset(CreateContextMenu(url));
 101   ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKWITH));
 102 }
 103
 104 IN_PROC_BROWSER_TEST_F(RegisterProtocolHandlerBrowserTest, CustomHandler) {
 105   ASSERT_TRUE(test_server()->Start());
 106   GURL handler_url = test_server()->GetURL("files/custom_handler_foo.html");
 107   AddProtocolHandler("foo", handler_url,
 108                      base::UTF8ToUTF16(std::string("Test foo Handler")));
 109
 110   ui_test_utils::NavigateToURL(browser(), GURL("foo:test"));
 111
 112   ASSERT_EQ(handler_url,
 113             browser()->tab_strip_model()->GetActiveWebContents()->GetURL());
 114 }