1 // Copyright (c) 2011 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 "base/utf_string_conversions.h" 6 #include "chrome/browser/extensions/extension_apitest.h" 7 #include "chrome/browser/extensions/extension_host.h" 8 #include "chrome/browser/extensions/extension_process_manager.h" 9 #include "chrome/browser/profiles/profile.h" 10 #include "chrome/browser/ui/browser.h" 11 #include "chrome/browser/ui/browser_list.h" 12 #include "chrome/common/chrome_switches.h" 13 #include "chrome/test/ui_test_utils.h" 14 #include "content/browser/renderer_host/render_view_host.h" 15 #include "content/browser/tab_contents/tab_contents.h" 16 #include "net/base/mock_host_resolver.h" 17 18 class AppApiTest : public ExtensionApiTest { 19 }; 20 21 // Simulates a page calling window.open on an URL, and waits for the navigation. 22 static void WindowOpenHelper(Browser* browser, 23 RenderViewHost* opener_host, 24 const GURL& url, 25 bool newtab_process_should_equal_opener) { 26 ASSERT_TRUE(ui_test_utils::ExecuteJavaScript( 27 opener_host, L"", L"window.open('" + UTF8ToWide(url.spec()) + L"');")); 28 29 // The above window.open call is not user-initiated, it will create 30 // a popup window instead of a new tab in current window. 31 // Now the active tab in last active window should be the new tab. 32 Browser* last_active_browser = BrowserList::GetLastActive(); 33 EXPECT_TRUE(last_active_browser); 34 TabContents* newtab = last_active_browser->GetSelectedTabContents(); 35 EXPECT_TRUE(newtab); 36 if (!newtab->controller().GetLastCommittedEntry() || 37 newtab->controller().GetLastCommittedEntry()->url() != url) 38 ui_test_utils::WaitForNavigation(&newtab->controller()); 39 EXPECT_EQ(url, newtab->controller().GetLastCommittedEntry()->url()); 40 if (newtab_process_should_equal_opener) 41 EXPECT_EQ(opener_host->process(), newtab->render_view_host()->process()); 42 else 43 EXPECT_NE(opener_host->process(), newtab->render_view_host()->process()); 44 } 45 46 // Simulates a page navigating itself to an URL, and waits for the navigation. 47 static void NavigateTabHelper(TabContents* contents, const GURL& url) { 48 bool result = false; 49 ASSERT_TRUE(ui_test_utils::ExecuteJavaScriptAndExtractBool( 50 contents->render_view_host(), L"", 51 L"window.addEventListener('unload', function() {" 52 L" window.domAutomationController.send(true);" 53 L"}, false);" 54 L"window.location = '" + UTF8ToWide(url.spec()) + L"';", 55 &result)); 56 ASSERT_TRUE(result); 57 58 if (!contents->controller().GetLastCommittedEntry() || 59 contents->controller().GetLastCommittedEntry()->url() != url) 60 ui_test_utils::WaitForNavigation(&contents->controller()); 61 EXPECT_EQ(url, contents->controller().GetLastCommittedEntry()->url()); 62 } 63 64 IN_PROC_BROWSER_TEST_F(AppApiTest, AppProcess) { 65 CommandLine::ForCurrentProcess()->AppendSwitch( 66 switches::kDisablePopupBlocking); 67 68 host_resolver()->AddRule("*", "127.0.0.1"); 69 ASSERT_TRUE(test_server()->Start()); 70 71 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("app_process"))); 72 73 // Open two tabs in the app, one outside it. 74 GURL base_url = test_server()->GetURL( 75 "files/extensions/api_test/app_process/"); 76 77 // The app under test acts on URLs whose host is "localhost", 78 // so the URLs we navigate to must have host "localhost". 79 GURL::Replacements replace_host; 80 std::string host_str("localhost"); // must stay in scope with replace_host 81 replace_host.SetHostStr(host_str); 82 base_url = base_url.ReplaceComponents(replace_host); 83 84 browser()->NewTab(); 85 ui_test_utils::NavigateToURL(browser(), base_url.Resolve("path1/empty.html")); 86 browser()->NewTab(); 87 ui_test_utils::NavigateToURL(browser(), base_url.Resolve("path2/empty.html")); 88 browser()->NewTab(); 89 ui_test_utils::NavigateToURL(browser(), base_url.Resolve("path3/empty.html")); 90 91 // The extension should have opened 3 new tabs. Including the original blank 92 // tab, we now have 4 tabs. Two should be part of the extension app, and 93 // grouped in the same process. 94 ASSERT_EQ(4, browser()->tab_count()); 95 RenderViewHost* host = browser()->GetTabContentsAt(1)->render_view_host(); 96 97 EXPECT_EQ(host->process(), 98 browser()->GetTabContentsAt(2)->render_view_host()->process()); 99 EXPECT_NE(host->process(), 100 browser()->GetTabContentsAt(3)->render_view_host()->process()); 101 102 // Now let's do the same using window.open. The same should happen. 103 ASSERT_EQ(1u, BrowserList::GetBrowserCount(browser()->profile())); 104 WindowOpenHelper(browser(), host, 105 base_url.Resolve("path1/empty.html"), true); 106 WindowOpenHelper(browser(), host, 107 base_url.Resolve("path2/empty.html"), true); 108 // TODO(creis): This should open in a new process (i.e., false for the last 109 // argument), but we temporarily avoid swapping processes away from an app 110 // until we're able to restore window.opener if the page later returns to an 111 // in-app URL. See crbug.com/65953. 112 WindowOpenHelper(browser(), host, 113 base_url.Resolve("path3/empty.html"), true); 114 115 // Now let's have these pages navigate, into or out of the extension web 116 // extent. They should switch processes. 117 const GURL& app_url(base_url.Resolve("path1/empty.html")); 118 const GURL& non_app_url(base_url.Resolve("path3/empty.html")); 119 NavigateTabHelper(browser()->GetTabContentsAt(2), non_app_url); 120 NavigateTabHelper(browser()->GetTabContentsAt(3), app_url); 121 // TODO(creis): This should swap out of the app's process (i.e., EXPECT_NE), 122 // but we temporarily avoid swapping away from an app in case it needs to 123 // communicate with window.opener later. See crbug.com/65953. 124 EXPECT_EQ(host->process(), 125 browser()->GetTabContentsAt(2)->render_view_host()->process()); 126 EXPECT_EQ(host->process(), 127 browser()->GetTabContentsAt(3)->render_view_host()->process()); 128 129 // If one of the popup tabs navigates back to the app, window.opener should 130 // be valid. 131 NavigateTabHelper(browser()->GetTabContentsAt(6), app_url); 132 EXPECT_EQ(host->process(), 133 browser()->GetTabContentsAt(6)->render_view_host()->process()); 134 bool windowOpenerValid = false; 135 ASSERT_TRUE(ui_test_utils::ExecuteJavaScriptAndExtractBool( 136 browser()->GetTabContentsAt(6)->render_view_host(), L"", 137 L"window.domAutomationController.send(window.opener != null)", 138 &windowOpenerValid)); 139 ASSERT_TRUE(windowOpenerValid); 140 } 141 142 // Tests that app process switching works properly in the following scenario: 143 // 1. navigate to a page1 in the app 144 // 2. page1 redirects to a page2 outside the app extent (ie, "/server-redirect") 145 // 3. page2 redirects back to a page in the app 146 // The final navigation should end up in the app process. 147 // See http://crbug.com/61757 148 IN_PROC_BROWSER_TEST_F(AppApiTest, AppProcessRedirectBack) { 149 CommandLine::ForCurrentProcess()->AppendSwitch( 150 switches::kDisablePopupBlocking); 151 152 host_resolver()->AddRule("*", "127.0.0.1"); 153 ASSERT_TRUE(test_server()->Start()); 154 155 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("app_process"))); 156 157 // Open two tabs in the app. 158 GURL base_url = test_server()->GetURL( 159 "files/extensions/api_test/app_process/"); 160 161 // The app under test acts on URLs whose host is "localhost", 162 // so the URLs we navigate to must have host "localhost". 163 GURL::Replacements replace_host; 164 std::string host_str("localhost"); // must stay in scope with replace_host 165 replace_host.SetHostStr(host_str); 166 base_url = base_url.ReplaceComponents(replace_host); 167 168 browser()->NewTab(); 169 ui_test_utils::NavigateToURL(browser(), base_url.Resolve("path1/empty.html")); 170 browser()->NewTab(); 171 // Wait until the second tab finishes its redirect train (2 hops). 172 ui_test_utils::NavigateToURLBlockUntilNavigationsComplete( 173 browser(), base_url.Resolve("path1/redirect.html"), 2); 174 175 // 3 tabs, including the initial about:blank. The last 2 should be the same 176 // process. 177 ASSERT_EQ(3, browser()->tab_count()); 178 EXPECT_EQ("/files/extensions/api_test/app_process/path1/empty.html", 179 browser()->GetTabContentsAt(2)->controller(). 180 GetLastCommittedEntry()->url().path()); 181 RenderViewHost* host = browser()->GetTabContentsAt(1)->render_view_host(); 182 EXPECT_EQ(host->process(), 183 browser()->GetTabContentsAt(2)->render_view_host()->process()); 184 } 185
