Home | History | Annotate | Download | only in custom_handlers 
      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 = WebKit::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 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                      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                      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 }
    115