blob: 67fd950d3b6e5b8d961e2716cb661efbecb921e0 [file] [log] [blame]
[email protected]fb8e3a32012-05-10 21:03:521// Copyright (c) 2012 The Chromium Authors. All rights reserved.
[email protected]f1eb87a2011-05-06 17:49:412// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
[email protected]3268d7b72013-03-28 17:41:435#include "base/strings/sys_string_conversions.h"
[email protected]112158af2013-06-07 23:46:186#include "base/strings/utf_string_conversions.h"
[email protected]8c484b742012-11-29 06:05:367#include "chrome/browser/extensions/component_loader.h"
[email protected]f1eb87a2011-05-06 17:49:418#include "chrome/browser/extensions/extension_browsertest.h"
[email protected]27658f32013-11-14 03:20:379#include "chrome/browser/search/search.h"
[email protected]f1eb87a2011-05-06 17:49:4110#include "chrome/browser/ui/browser.h"
11#include "chrome/browser/ui/browser_window.h"
[email protected]d57ce6a2014-07-03 15:39:2312#include "chrome/browser/ui/location_bar/location_bar.h"
Conley Owensaafcf122018-02-26 18:53:5113#include "chrome/browser/ui/search/local_ntp_test_utils.h"
[email protected]47ae23372013-01-29 01:50:4814#include "chrome/browser/ui/tabs/tab_strip_model.h"
calamityae7fed42017-06-22 04:58:2215#include "chrome/browser/ui/webui/md_bookmarks/md_bookmarks_ui.h"
[email protected]f1eb87a2011-05-06 17:49:4116#include "chrome/common/url_constants.h"
[email protected]af44e7fb2011-07-29 18:32:3217#include "chrome/test/base/in_process_browser_test.h"
[email protected]a4ff9eae2011-08-01 19:58:1618#include "chrome/test/base/testing_profile.h"
[email protected]af44e7fb2011-07-29 18:32:3219#include "chrome/test/base/ui_test_utils.h"
blundell7dbd3792015-08-05 15:14:1920#include "components/omnibox/browser/omnibox_view.h"
[email protected]cdcb1dee2012-01-04 00:46:2021#include "content/public/browser/navigation_controller.h"
[email protected]ad23a092011-12-28 07:02:0422#include "content/public/browser/navigation_entry.h"
[email protected]4ca15302012-01-03 05:53:2023#include "content/public/browser/web_contents.h"
[email protected]885c0e92012-11-13 20:27:4224#include "extensions/common/constants.h"
[email protected]a6483d22013-07-03 22:11:0025#include "url/gurl.h"
[email protected]f1eb87a2011-05-06 17:49:4126
[email protected]10f417c52011-12-28 21:04:2327using content::NavigationEntry;
28
Devlin Cronin836f545d2018-05-09 00:25:0529class ExtensionURLRewriteBrowserTest : public extensions::ExtensionBrowserTest {
[email protected]8c484b742012-11-29 06:05:3630 public:
dcheng72191812014-10-28 20:49:5631 void SetUp() override {
[email protected]8c484b742012-11-29 06:05:3632 extensions::ComponentLoader::EnableBackgroundExtensionsForTesting();
Devlin Cronin836f545d2018-05-09 00:25:0533 extensions::ExtensionBrowserTest::SetUp();
[email protected]8c484b742012-11-29 06:05:3634 }
35
[email protected]f1eb87a2011-05-06 17:49:4136 protected:
37 std::string GetLocationBarText() const {
[email protected]04338722013-12-24 23:18:0538 return base::UTF16ToUTF8(
[email protected]e9ee7ea2013-11-20 20:38:0239 browser()->window()->GetLocationBar()->GetOmniboxView()->GetText());
[email protected]f1eb87a2011-05-06 17:49:4140 }
41
42 GURL GetLocationBarTextAsURL() const {
43 return GURL(GetLocationBarText());
44 }
45
[email protected]cdcb1dee2012-01-04 00:46:2046 content::NavigationController* GetNavigationController() const {
Conley Owensaafcf122018-02-26 18:53:5147 return &browser()
48 ->tab_strip_model()
49 ->GetActiveWebContents()
50 ->GetController();
[email protected]f1eb87a2011-05-06 17:49:4151 }
52
[email protected]10f417c52011-12-28 21:04:2353 NavigationEntry* GetNavigationEntry() const {
[email protected]afe9aba2013-08-16 20:31:3454 return GetNavigationController()->GetVisibleEntry();
[email protected]f1eb87a2011-05-06 17:49:4155 }
56
[email protected]650b2d52013-02-10 03:41:4557 base::FilePath GetTestExtensionPath(const char* extension_name) const {
[email protected]f1eb87a2011-05-06 17:49:4158 return test_data_dir_.AppendASCII("browsertest/url_rewrite/").
59 AppendASCII(extension_name);
60 }
61
62 // Navigates to |url| and tests that the location bar and the |virtual_url|
63 // correspond to |url|, while the real URL of the navigation entry uses the
64 // chrome-extension:// scheme.
65 void TestExtensionURLOverride(const GURL& url) {
66 ui_test_utils::NavigateToURL(browser(), url);
67 EXPECT_EQ(url, GetLocationBarTextAsURL());
[email protected]36fc0392011-12-25 03:59:5168 EXPECT_EQ(url, GetNavigationEntry()->GetVirtualURL());
69 EXPECT_TRUE(
[email protected]885c0e92012-11-13 20:27:4270 GetNavigationEntry()->GetURL().SchemeIs(extensions::kExtensionScheme));
[email protected]f1eb87a2011-05-06 17:49:4171 }
72
73 // Navigates to |url| and tests that the location bar is empty while the
74 // |virtual_url| is the same as |url|.
75 void TestURLNotShown(const GURL& url) {
76 ui_test_utils::NavigateToURL(browser(), url);
77 EXPECT_EQ("", GetLocationBarText());
Nico Weber01bea8ca2018-02-22 18:15:2478 EXPECT_EQ(url, GetNavigationEntry()->GetVirtualURL());
[email protected]f1eb87a2011-05-06 17:49:4179 }
80};
81
82IN_PROC_BROWSER_TEST_F(ExtensionURLRewriteBrowserTest, NewTabPageURL) {
83 // Navigate to chrome://newtab and check that the location bar text is blank.
Conley Owensaafcf122018-02-26 18:53:5184 // We do not use TestURLNotShown here because the virtual URL may be
85 // updated to the local NTP since we do not have a network connection to
86 // reach the remote NTP.
87 ui_test_utils::NavigateToURL(browser(), GURL(chrome::kChromeUINewTabURL));
88 EXPECT_EQ("", GetLocationBarText());
89 // Check that the actual and virtual URL corresponds to the new tab URL.
90 EXPECT_EQ(local_ntp_test_utils::GetFinalNtpUrl(browser()->profile()),
91 GetNavigationEntry()->GetVirtualURL());
sdefresne51bbec7b2015-08-03 14:18:1392 EXPECT_TRUE(search::IsNTPURL(GetNavigationEntry()->GetURL(), profile()));
[email protected]f1eb87a2011-05-06 17:49:4193}
94
95IN_PROC_BROWSER_TEST_F(ExtensionURLRewriteBrowserTest, NewTabPageURLOverride) {
96 // Load an extension to override the NTP and check that the location bar text
97 // is blank after navigating to chrome://newtab.
[email protected]2746e2a2013-02-25 22:11:4898 ASSERT_TRUE(LoadExtension(GetTestExtensionPath("newtab")));
[email protected]f1eb87a2011-05-06 17:49:4199 TestURLNotShown(GURL(chrome::kChromeUINewTabURL));
100 // Check that the internal URL uses the chrome-extension:// scheme.
[email protected]885c0e92012-11-13 20:27:42101 EXPECT_TRUE(GetNavigationEntry()->GetURL().SchemeIs(
102 extensions::kExtensionScheme));
[email protected]f1eb87a2011-05-06 17:49:41103}
104
[email protected]2746e2a2013-02-25 22:11:48105IN_PROC_BROWSER_TEST_F(ExtensionURLRewriteBrowserTest, BookmarksURLOverride) {
[email protected]f1eb87a2011-05-06 17:49:41106 // Load an extension that overrides chrome://bookmarks.
[email protected]2746e2a2013-02-25 22:11:48107 ASSERT_TRUE(LoadExtension(GetTestExtensionPath("bookmarks")));
[email protected]f1eb87a2011-05-06 17:49:41108 // Navigate to chrome://bookmarks and check that the location bar URL is what
109 // was entered and the internal URL uses the chrome-extension:// scheme.
110 TestExtensionURLOverride(GURL(chrome::kChromeUIBookmarksURL));
111}