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.
WindowOpenHelper(Browser * browser,RenderViewHost * opener_host,const GURL & url,bool newtab_process_should_equal_opener)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.
NavigateTabHelper(TabContents * contents,const GURL & url)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
IN_PROC_BROWSER_TEST_F(AppApiTest,AppProcess)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
IN_PROC_BROWSER_TEST_F(AppApiTest,AppProcessRedirectBack)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