[email protected] | fb8e3a3 | 2012-05-10 21:03:52 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
[email protected] | 3268d7b7 | 2013-03-28 17:41:43 | [diff] [blame] | 5 | #include "base/strings/sys_string_conversions.h" |
[email protected] | 112158af | 2013-06-07 23:46:18 | [diff] [blame] | 6 | #include "base/strings/utf_string_conversions.h" |
[email protected] | 8c484b74 | 2012-11-29 06:05:36 | [diff] [blame] | 7 | #include "chrome/browser/extensions/component_loader.h" |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 8 | #include "chrome/browser/extensions/extension_browsertest.h" |
[email protected] | 27658f3 | 2013-11-14 03:20:37 | [diff] [blame] | 9 | #include "chrome/browser/search/search.h" |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 10 | #include "chrome/browser/ui/browser.h" |
| 11 | #include "chrome/browser/ui/browser_window.h" |
[email protected] | d57ce6a | 2014-07-03 15:39:23 | [diff] [blame] | 12 | #include "chrome/browser/ui/location_bar/location_bar.h" |
Conley Owens | aafcf12 | 2018-02-26 18:53:51 | [diff] [blame] | 13 | #include "chrome/browser/ui/search/local_ntp_test_utils.h" |
[email protected] | 47ae2337 | 2013-01-29 01:50:48 | [diff] [blame] | 14 | #include "chrome/browser/ui/tabs/tab_strip_model.h" |
calamity | ae7fed4 | 2017-06-22 04:58:22 | [diff] [blame] | 15 | #include "chrome/browser/ui/webui/md_bookmarks/md_bookmarks_ui.h" |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 16 | #include "chrome/common/url_constants.h" |
[email protected] | af44e7fb | 2011-07-29 18:32:32 | [diff] [blame] | 17 | #include "chrome/test/base/in_process_browser_test.h" |
[email protected] | a4ff9eae | 2011-08-01 19:58:16 | [diff] [blame] | 18 | #include "chrome/test/base/testing_profile.h" |
[email protected] | af44e7fb | 2011-07-29 18:32:32 | [diff] [blame] | 19 | #include "chrome/test/base/ui_test_utils.h" |
blundell | 7dbd379 | 2015-08-05 15:14:19 | [diff] [blame] | 20 | #include "components/omnibox/browser/omnibox_view.h" |
[email protected] | cdcb1dee | 2012-01-04 00:46:20 | [diff] [blame] | 21 | #include "content/public/browser/navigation_controller.h" |
[email protected] | ad23a09 | 2011-12-28 07:02:04 | [diff] [blame] | 22 | #include "content/public/browser/navigation_entry.h" |
[email protected] | 4ca1530 | 2012-01-03 05:53:20 | [diff] [blame] | 23 | #include "content/public/browser/web_contents.h" |
[email protected] | 885c0e9 | 2012-11-13 20:27:42 | [diff] [blame] | 24 | #include "extensions/common/constants.h" |
[email protected] | a6483d2 | 2013-07-03 22:11:00 | [diff] [blame] | 25 | #include "url/gurl.h" |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 26 | |
[email protected] | 10f417c5 | 2011-12-28 21:04:23 | [diff] [blame] | 27 | using content::NavigationEntry; |
| 28 | |
Devlin Cronin | 836f545d | 2018-05-09 00:25:05 | [diff] [blame] | 29 | class ExtensionURLRewriteBrowserTest : public extensions::ExtensionBrowserTest { |
[email protected] | 8c484b74 | 2012-11-29 06:05:36 | [diff] [blame] | 30 | public: |
dcheng | 7219181 | 2014-10-28 20:49:56 | [diff] [blame] | 31 | void SetUp() override { |
[email protected] | 8c484b74 | 2012-11-29 06:05:36 | [diff] [blame] | 32 | extensions::ComponentLoader::EnableBackgroundExtensionsForTesting(); |
Devlin Cronin | 836f545d | 2018-05-09 00:25:05 | [diff] [blame] | 33 | extensions::ExtensionBrowserTest::SetUp(); |
[email protected] | 8c484b74 | 2012-11-29 06:05:36 | [diff] [blame] | 34 | } |
| 35 | |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 36 | protected: |
| 37 | std::string GetLocationBarText() const { |
[email protected] | 0433872 | 2013-12-24 23:18:05 | [diff] [blame] | 38 | return base::UTF16ToUTF8( |
[email protected] | e9ee7ea | 2013-11-20 20:38:02 | [diff] [blame] | 39 | browser()->window()->GetLocationBar()->GetOmniboxView()->GetText()); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 40 | } |
| 41 | |
| 42 | GURL GetLocationBarTextAsURL() const { |
| 43 | return GURL(GetLocationBarText()); |
| 44 | } |
| 45 | |
[email protected] | cdcb1dee | 2012-01-04 00:46:20 | [diff] [blame] | 46 | content::NavigationController* GetNavigationController() const { |
Conley Owens | aafcf12 | 2018-02-26 18:53:51 | [diff] [blame] | 47 | return &browser() |
| 48 | ->tab_strip_model() |
| 49 | ->GetActiveWebContents() |
| 50 | ->GetController(); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 51 | } |
| 52 | |
[email protected] | 10f417c5 | 2011-12-28 21:04:23 | [diff] [blame] | 53 | NavigationEntry* GetNavigationEntry() const { |
[email protected] | afe9aba | 2013-08-16 20:31:34 | [diff] [blame] | 54 | return GetNavigationController()->GetVisibleEntry(); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 55 | } |
| 56 | |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 57 | base::FilePath GetTestExtensionPath(const char* extension_name) const { |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 58 | 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] | 36fc039 | 2011-12-25 03:59:51 | [diff] [blame] | 68 | EXPECT_EQ(url, GetNavigationEntry()->GetVirtualURL()); |
| 69 | EXPECT_TRUE( |
[email protected] | 885c0e9 | 2012-11-13 20:27:42 | [diff] [blame] | 70 | GetNavigationEntry()->GetURL().SchemeIs(extensions::kExtensionScheme)); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 71 | } |
| 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 Weber | 01bea8ca | 2018-02-22 18:15:24 | [diff] [blame] | 78 | EXPECT_EQ(url, GetNavigationEntry()->GetVirtualURL()); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 79 | } |
| 80 | }; |
| 81 | |
| 82 | IN_PROC_BROWSER_TEST_F(ExtensionURLRewriteBrowserTest, NewTabPageURL) { |
| 83 | // Navigate to chrome://newtab and check that the location bar text is blank. |
Conley Owens | aafcf12 | 2018-02-26 18:53:51 | [diff] [blame] | 84 | // 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()); |
sdefresne | 51bbec7b | 2015-08-03 14:18:13 | [diff] [blame] | 92 | EXPECT_TRUE(search::IsNTPURL(GetNavigationEntry()->GetURL(), profile())); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 93 | } |
| 94 | |
| 95 | IN_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] | 2746e2a | 2013-02-25 22:11:48 | [diff] [blame] | 98 | ASSERT_TRUE(LoadExtension(GetTestExtensionPath("newtab"))); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 99 | TestURLNotShown(GURL(chrome::kChromeUINewTabURL)); |
| 100 | // Check that the internal URL uses the chrome-extension:// scheme. |
[email protected] | 885c0e9 | 2012-11-13 20:27:42 | [diff] [blame] | 101 | EXPECT_TRUE(GetNavigationEntry()->GetURL().SchemeIs( |
| 102 | extensions::kExtensionScheme)); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 103 | } |
| 104 | |
[email protected] | 2746e2a | 2013-02-25 22:11:48 | [diff] [blame] | 105 | IN_PROC_BROWSER_TEST_F(ExtensionURLRewriteBrowserTest, BookmarksURLOverride) { |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 106 | // Load an extension that overrides chrome://bookmarks. |
[email protected] | 2746e2a | 2013-02-25 22:11:48 | [diff] [blame] | 107 | ASSERT_TRUE(LoadExtension(GetTestExtensionPath("bookmarks"))); |
[email protected] | f1eb87a | 2011-05-06 17:49:41 | [diff] [blame] | 108 | // 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 | } |