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
18class AppApiTest : public ExtensionApiTest {
19};
20
21// Simulates a page calling window.open on an URL, and waits for the navigation.
22static 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.
47static 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
64IN_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
148IN_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