• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 "content/public/test/content_browser_test.h"
6 
7 #include "base/command_line.h"
8 #include "base/files/file_path.h"
9 #include "base/logging.h"
10 #include "base/message_loop/message_loop.h"
11 #include "base/path_service.h"
12 #include "content/public/browser/render_process_host.h"
13 #include "content/public/common/content_paths.h"
14 #include "content/public/common/content_switches.h"
15 #include "content/public/common/url_constants.h"
16 #include "content/shell/browser/shell.h"
17 #include "content/shell/browser/shell_browser_context.h"
18 #include "content/shell/browser/shell_content_browser_client.h"
19 #include "content/shell/common/shell_switches.h"
20 #include "content/shell/renderer/shell_content_renderer_client.h"
21 #include "content/test/test_content_client.h"
22 #include "net/test/embedded_test_server/embedded_test_server.h"
23 
24 #if defined(OS_ANDROID)
25 #include "content/shell/app/shell_main_delegate.h"
26 #endif
27 
28 #if defined(OS_MACOSX)
29 #include "base/mac/scoped_nsautorelease_pool.h"
30 #endif
31 
32 #if !defined(OS_CHROMEOS) && defined(OS_LINUX)
33 #include "ui/base/ime/input_method_initializer.h"
34 #endif
35 
36 namespace content {
37 
ContentBrowserTest()38 ContentBrowserTest::ContentBrowserTest()
39     : setup_called_(false) {
40 #if defined(OS_MACOSX)
41   // See comment in InProcessBrowserTest::InProcessBrowserTest().
42   base::FilePath content_shell_path;
43   CHECK(PathService::Get(base::FILE_EXE, &content_shell_path));
44   content_shell_path = content_shell_path.DirName();
45   content_shell_path = content_shell_path.Append(
46       FILE_PATH_LITERAL("Content Shell.app/Contents/MacOS/Content Shell"));
47   CHECK(PathService::Override(base::FILE_EXE, content_shell_path));
48 #endif
49   base::FilePath content_test_data(FILE_PATH_LITERAL("content/test/data"));
50   CreateTestServer(content_test_data);
51   base::FilePath content_test_data_absolute;
52   CHECK(PathService::Get(base::DIR_SOURCE_ROOT, &content_test_data_absolute));
53   content_test_data_absolute =
54       content_test_data_absolute.Append(content_test_data);
55   embedded_test_server()->ServeFilesFromDirectory(content_test_data_absolute);
56 }
57 
~ContentBrowserTest()58 ContentBrowserTest::~ContentBrowserTest() {
59   CHECK(setup_called_) << "Overridden SetUp() did not call parent "
60                           "implementation, so test not run.";
61 }
62 
SetUp()63 void ContentBrowserTest::SetUp() {
64   CommandLine* command_line = CommandLine::ForCurrentProcess();
65   command_line->AppendSwitch(switches::kContentBrowserTest);
66 
67   SetUpCommandLine(command_line);
68 
69 #if defined(OS_ANDROID)
70   shell_main_delegate_.reset(new ShellMainDelegate);
71   shell_main_delegate_->PreSandboxStartup();
72   if (command_line->HasSwitch(switches::kSingleProcess)) {
73     // We explicitly leak the new ContentRendererClient as we're
74     // setting a global that may be used after ContentBrowserTest is
75     // destroyed.
76     ContentRendererClient* old_client =
77         SetRendererClientForTesting(new ShellContentRendererClient());
78     // No-one should have set this value before we did.
79     DCHECK(!old_client);
80   }
81 #elif defined(OS_MACOSX)
82   // See InProcessBrowserTest::PrepareTestCommandLine().
83   base::FilePath subprocess_path;
84   PathService::Get(base::FILE_EXE, &subprocess_path);
85   subprocess_path = subprocess_path.DirName().DirName();
86   DCHECK_EQ(subprocess_path.BaseName().value(), "Contents");
87   subprocess_path = subprocess_path.Append(
88       "Frameworks/Content Shell Helper.app/Contents/MacOS/Content Shell Helper");
89   command_line->AppendSwitchPath(switches::kBrowserSubprocessPath,
90                                  subprocess_path);
91 #endif
92 
93   // LinuxInputMethodContextFactory has to be initialized.
94 #if !defined(OS_CHROMEOS) && defined(OS_LINUX)
95   ui::InitializeInputMethodForTesting();
96 #endif
97 
98   setup_called_ = true;
99 
100   BrowserTestBase::SetUp();
101 }
102 
TearDown()103 void ContentBrowserTest::TearDown() {
104   BrowserTestBase::TearDown();
105 
106   // LinuxInputMethodContextFactory has to be shutdown.
107 #if !defined(OS_CHROMEOS) && defined(OS_LINUX)
108   ui::ShutdownInputMethodForTesting();
109 #endif
110 
111 #if defined(OS_ANDROID)
112   shell_main_delegate_.reset();
113 #endif
114 }
115 
RunTestOnMainThreadLoop()116 void ContentBrowserTest::RunTestOnMainThreadLoop() {
117   if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree)) {
118     CHECK_EQ(Shell::windows().size(), 1u);
119     shell_ = Shell::windows()[0];
120   }
121 
122 #if defined(OS_MACOSX)
123   // On Mac, without the following autorelease pool, code which is directly
124   // executed (as opposed to executed inside a message loop) would autorelease
125   // objects into a higher-level pool. This pool is not recycled in-sync with
126   // the message loops' pools and causes problems with code relying on
127   // deallocation via an autorelease pool (such as browser window closure and
128   // browser shutdown). To avoid this, the following pool is recycled after each
129   // time code is directly executed.
130   base::mac::ScopedNSAutoreleasePool pool;
131 #endif
132 
133   // Pump startup related events.
134   base::MessageLoopForUI::current()->RunUntilIdle();
135 
136 #if defined(OS_MACOSX)
137   pool.Recycle();
138 #endif
139 
140   SetUpOnMainThread();
141 
142   RunTestOnMainThread();
143 
144   TearDownOnMainThread();
145 #if defined(OS_MACOSX)
146   pool.Recycle();
147 #endif
148 
149   for (RenderProcessHost::iterator i(RenderProcessHost::AllHostsIterator());
150        !i.IsAtEnd(); i.Advance()) {
151     i.GetCurrentValue()->FastShutdownIfPossible();
152   }
153 
154   Shell::CloseAllWindows();
155 }
156 
CreateBrowser()157 Shell* ContentBrowserTest::CreateBrowser() {
158   return Shell::CreateNewWindow(
159       ShellContentBrowserClient::Get()->browser_context(),
160       GURL(url::kAboutBlankURL),
161       NULL,
162       MSG_ROUTING_NONE,
163       gfx::Size());
164 }
165 
CreateOffTheRecordBrowser()166 Shell* ContentBrowserTest::CreateOffTheRecordBrowser() {
167   return Shell::CreateNewWindow(
168       ShellContentBrowserClient::Get()->off_the_record_browser_context(),
169       GURL(url::kAboutBlankURL),
170       NULL,
171       MSG_ROUTING_NONE,
172       gfx::Size());
173 }
174 
175 }  // namespace content
176