1 // Copyright (c) 2012 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.
5 #include "chrome/test/base/testing_browser_process.h"
7 #include "base/prefs/pref_service.h"
8 #include "base/strings/string_util.h"
9 #include "base/time/default_tick_clock.h"
10 #include "build/build_config.h"
11 #include "chrome/browser/browser_process.h"
12 #include "chrome/browser/browser_process_impl.h"
13 #include "chrome/browser/printing/print_job_manager.h"
14 #include "chrome/browser/profiles/profile_manager.h"
15 #include "chrome/test/base/testing_browser_process_platform_part.h"
16 #include "components/network_time/network_time_tracker.h"
17 #include "content/public/browser/notification_service.h"
18 #include "net/url_request/url_request_context_getter.h"
19 #include "testing/gtest/include/gtest/gtest.h"
20 #include "ui/message_center/message_center.h"
23 #include "chrome/browser/notifications/notification_ui_manager.h"
24 #include "chrome/browser/prerender/prerender_tracker.h"
25 #include "chrome/browser/safe_browsing/safe_browsing_service.h"
28 #if defined(ENABLE_BACKGROUND)
29 #include "chrome/browser/background/background_mode_manager.h"
32 #if defined(ENABLE_CONFIGURATION_POLICY)
33 #include "components/policy/core/browser/browser_policy_connector.h"
35 #include "components/policy/core/common/policy_service_stub.h"
36 #endif // defined(ENABLE_CONFIGURATION_POLICY)
38 #if defined(ENABLE_EXTENSIONS)
39 #include "chrome/browser/extensions/chrome_extensions_browser_client.h"
40 #include "chrome/browser/media_galleries/media_file_system_registry.h"
41 #include "chrome/browser/ui/apps/chrome_app_window_client.h"
42 #include "components/storage_monitor/storage_monitor.h"
43 #include "components/storage_monitor/test_storage_monitor.h"
46 #if defined(ENABLE_PRINT_PREVIEW)
47 #include "chrome/browser/printing/background_printing_manager.h"
48 #include "chrome/browser/printing/print_preview_dialog_controller.h"
52 TestingBrowserProcess
* TestingBrowserProcess::GetGlobal() {
53 return static_cast<TestingBrowserProcess
*>(g_browser_process
);
57 void TestingBrowserProcess::CreateInstance() {
58 DCHECK(!g_browser_process
);
59 g_browser_process
= new TestingBrowserProcess
;
63 void TestingBrowserProcess::DeleteInstance() {
64 // g_browser_process must be NULL during its own destruction.
65 BrowserProcess
* browser_process
= g_browser_process
;
66 g_browser_process
= NULL
;
67 delete browser_process
;
70 TestingBrowserProcess::TestingBrowserProcess()
71 : notification_service_(content::NotificationService::Create()),
76 system_request_context_(NULL
),
77 platform_part_(new TestingBrowserProcessPlatformPart()) {
78 #if defined(ENABLE_EXTENSIONS)
79 extensions_browser_client_
.reset(
80 new extensions::ChromeExtensionsBrowserClient
);
81 extensions::AppWindowClient::Set(ChromeAppWindowClient::GetInstance());
82 extensions::ExtensionsBrowserClient::Set(extensions_browser_client_
.get());
86 TestingBrowserProcess::~TestingBrowserProcess() {
87 EXPECT_FALSE(local_state_
);
88 #if defined(ENABLE_CONFIGURATION_POLICY)
89 SetBrowserPolicyConnector(NULL
);
91 #if defined(ENABLE_EXTENSIONS)
92 extensions::ExtensionsBrowserClient::Set(NULL
);
95 // Destructors for some objects owned by TestingBrowserProcess will use
96 // g_browser_process if it is not NULL, so it must be NULL before proceeding.
97 DCHECK_EQ(static_cast<BrowserProcess
*>(NULL
), g_browser_process
);
100 void TestingBrowserProcess::ResourceDispatcherHostCreated() {
103 void TestingBrowserProcess::EndSession() {
106 MetricsServicesManager
* TestingBrowserProcess::GetMetricsServicesManager() {
110 metrics::MetricsService
* TestingBrowserProcess::metrics_service() {
114 rappor::RapporService
* TestingBrowserProcess::rappor_service() {
118 IOThread
* TestingBrowserProcess::io_thread() {
122 WatchDogThread
* TestingBrowserProcess::watchdog_thread() {
126 ProfileManager
* TestingBrowserProcess::profile_manager() {
131 return profile_manager_
.get();
135 void TestingBrowserProcess::SetProfileManager(ProfileManager
* profile_manager
) {
137 // NotificationUIManager can contain references to elements in the current
138 // ProfileManager (for example, the MessageCenterSettingsController maintains
139 // a pointer to the ProfileInfoCache). So when we change the ProfileManager
140 // (typically during test shutdown) make sure to reset any objects that might
141 // maintain references to it. See SetLocalState() for a description of a
142 // similar situation.
143 notification_ui_manager_
.reset();
144 profile_manager_
.reset(profile_manager
);
148 PrefService
* TestingBrowserProcess::local_state() {
152 chrome_variations::VariationsService
*
153 TestingBrowserProcess::variations_service() {
157 policy::BrowserPolicyConnector
*
158 TestingBrowserProcess::browser_policy_connector() {
159 #if defined(ENABLE_CONFIGURATION_POLICY)
160 if (!browser_policy_connector_
)
161 browser_policy_connector_
= platform_part_
->CreateBrowserPolicyConnector();
162 return browser_policy_connector_
.get();
168 policy::PolicyService
* TestingBrowserProcess::policy_service() {
172 #elif defined(ENABLE_CONFIGURATION_POLICY)
173 return browser_policy_connector()->GetPolicyService();
175 if (!policy_service_
)
176 policy_service_
.reset(new policy::PolicyServiceStub());
177 return policy_service_
.get();
181 IconManager
* TestingBrowserProcess::icon_manager() {
185 GLStringManager
* TestingBrowserProcess::gl_string_manager() {
189 GpuModeManager
* TestingBrowserProcess::gpu_mode_manager() {
193 BackgroundModeManager
* TestingBrowserProcess::background_mode_manager() {
197 void TestingBrowserProcess::set_background_mode_manager_for_test(
198 scoped_ptr
<BackgroundModeManager
> manager
) {
202 StatusTray
* TestingBrowserProcess::status_tray() {
206 SafeBrowsingService
* TestingBrowserProcess::safe_browsing_service() {
211 return sb_service_
.get();
215 safe_browsing::ClientSideDetectionService
*
216 TestingBrowserProcess::safe_browsing_detection_service() {
220 net::URLRequestContextGetter
* TestingBrowserProcess::system_request_context() {
221 return system_request_context_
;
224 BrowserProcessPlatformPart
* TestingBrowserProcess::platform_part() {
225 return platform_part_
.get();
228 extensions::EventRouterForwarder
*
229 TestingBrowserProcess::extension_event_router_forwarder() {
233 NotificationUIManager
* TestingBrowserProcess::notification_ui_manager() {
234 #if defined(ENABLE_NOTIFICATIONS)
235 if (!notification_ui_manager_
.get())
236 notification_ui_manager_
.reset(
237 NotificationUIManager::Create(local_state()));
238 return notification_ui_manager_
.get();
245 message_center::MessageCenter
* TestingBrowserProcess::message_center() {
246 return message_center::MessageCenter::Get();
249 IntranetRedirectDetector
* TestingBrowserProcess::intranet_redirect_detector() {
252 void TestingBrowserProcess::CreateDevToolsHttpProtocolHandler(
253 chrome::HostDesktopType host_desktop_type
,
254 const std::string
& ip
,
258 unsigned int TestingBrowserProcess::AddRefModule() {
259 return ++module_ref_count_
;
262 unsigned int TestingBrowserProcess::ReleaseModule() {
263 DCHECK_GT(module_ref_count_
, 0U);
264 return --module_ref_count_
;
267 bool TestingBrowserProcess::IsShuttingDown() {
271 printing::PrintJobManager
* TestingBrowserProcess::print_job_manager() {
272 #if defined(ENABLE_PRINTING)
273 if (!print_job_manager_
.get())
274 print_job_manager_
.reset(new printing::PrintJobManager());
275 return print_job_manager_
.get();
282 printing::PrintPreviewDialogController
*
283 TestingBrowserProcess::print_preview_dialog_controller() {
284 #if defined(ENABLE_PRINT_PREVIEW)
285 if (!print_preview_dialog_controller_
.get())
286 print_preview_dialog_controller_
=
287 new printing::PrintPreviewDialogController();
288 return print_preview_dialog_controller_
.get();
295 printing::BackgroundPrintingManager
*
296 TestingBrowserProcess::background_printing_manager() {
297 #if defined(ENABLE_PRINT_PREVIEW)
298 if (!background_printing_manager_
.get()) {
299 background_printing_manager_
.reset(
300 new printing::BackgroundPrintingManager());
302 return background_printing_manager_
.get();
309 const std::string
& TestingBrowserProcess::GetApplicationLocale() {
313 void TestingBrowserProcess::SetApplicationLocale(
314 const std::string
& app_locale
) {
315 app_locale_
= app_locale
;
318 DownloadStatusUpdater
* TestingBrowserProcess::download_status_updater() {
322 DownloadRequestLimiter
* TestingBrowserProcess::download_request_limiter() {
326 ChromeNetLog
* TestingBrowserProcess::net_log() {
330 prerender::PrerenderTracker
* TestingBrowserProcess::prerender_tracker() {
335 if (!prerender_tracker_
.get())
336 prerender_tracker_
.reset(new prerender::PrerenderTracker());
337 return prerender_tracker_
.get();
341 component_updater::ComponentUpdateService
*
342 TestingBrowserProcess::component_updater() {
346 CRLSetFetcher
* TestingBrowserProcess::crl_set_fetcher() {
350 component_updater::PnaclComponentInstaller
*
351 TestingBrowserProcess::pnacl_component_installer() {
355 MediaFileSystemRegistry
* TestingBrowserProcess::media_file_system_registry() {
356 #if defined(OS_IOS) || defined(OS_ANDROID)
360 if (!media_file_system_registry_
)
361 media_file_system_registry_
.reset(new MediaFileSystemRegistry());
362 return media_file_system_registry_
.get();
366 bool TestingBrowserProcess::created_local_state() const {
367 return (local_state_
!= NULL
);
370 #if defined(ENABLE_WEBRTC)
371 WebRtcLogUploader
* TestingBrowserProcess::webrtc_log_uploader() {
376 network_time::NetworkTimeTracker
*
377 TestingBrowserProcess::network_time_tracker() {
378 if (!network_time_tracker_
) {
379 DCHECK(local_state_
);
380 network_time_tracker_
.reset(new network_time::NetworkTimeTracker(
381 scoped_ptr
<base::TickClock
>(new base::DefaultTickClock()),
384 return network_time_tracker_
.get();
387 gcm::GCMDriver
* TestingBrowserProcess::gcm_driver() {
391 void TestingBrowserProcess::SetSystemRequestContext(
392 net::URLRequestContextGetter
* context_getter
) {
393 system_request_context_
= context_getter
;
396 void TestingBrowserProcess::SetLocalState(PrefService
* local_state
) {
398 // The local_state_ PrefService is owned outside of TestingBrowserProcess,
399 // but some of the members of TestingBrowserProcess hold references to it
400 // (for example, via PrefNotifier members). But given our test
401 // infrastructure which tears down individual tests before freeing the
402 // TestingBrowserProcess, there's not a good way to make local_state outlive
403 // these dependencies. As a workaround, whenever local_state_ is cleared
404 // (assumedly as part of exiting the test and freeing TestingBrowserProcess)
405 // any components owned by TestingBrowserProcess that depend on local_state
407 network_time_tracker_
.reset();
409 notification_ui_manager_
.reset();
411 #if defined(ENABLE_CONFIGURATION_POLICY)
412 SetBrowserPolicyConnector(NULL
);
415 local_state_
= local_state
;
418 void TestingBrowserProcess::SetIOThread(IOThread
* io_thread
) {
419 io_thread_
= io_thread
;
422 void TestingBrowserProcess::SetBrowserPolicyConnector(
423 policy::BrowserPolicyConnector
* connector
) {
424 #if defined(ENABLE_CONFIGURATION_POLICY)
425 if (browser_policy_connector_
) {
426 browser_policy_connector_
->Shutdown();
428 browser_policy_connector_
.reset(connector
);
434 void TestingBrowserProcess::SetSafeBrowsingService(
435 SafeBrowsingService
* sb_service
) {
439 sb_service_
= sb_service
;
443 ///////////////////////////////////////////////////////////////////////////////
445 TestingBrowserProcessInitializer::TestingBrowserProcessInitializer() {
446 TestingBrowserProcess::CreateInstance();
449 TestingBrowserProcessInitializer::~TestingBrowserProcessInitializer() {
450 TestingBrowserProcess::DeleteInstance();