Chromecast: end-to-end browser test based on content::BrowserTest.

chromecast/shell/browser/cast_browser_main_parts.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chromecast/shell/browser/cast_browser_main_parts.h"
 
 #include "base/command_line.h"
 #include "base/prefs/pref_registry_simple.h"
 #include "chromecast/common/chromecast_config.h"
 #include "chromecast/metrics/cast_metrics_service_client.h"
@@ skipping 33 matching lines @@
   }
 }
 
 }  // namespace
 
 CastBrowserMainParts::CastBrowserMainParts(
     const content::MainFunctionParams& parameters,
     URLRequestContextFactory* url_request_context_factory)
     : BrowserMainParts(),
       cast_browser_process_(new CastBrowserProcess()),
+      parameters_(parameters),
       url_request_context_factory_(url_request_context_factory) {
   CommandLine* command_line = CommandLine::ForCurrentProcess();
   AddDefaultCommandLineSwitches(command_line);
 }
 
 CastBrowserMainParts::~CastBrowserMainParts() {
 }
 
 void CastBrowserMainParts::PreMainMessageLoopStart() {
   net::NetworkChangeNotifier::SetFactory(
@@ skipping 21 matching lines @@
   cast_browser_process_->SetRemoteDebuggingServer(new RemoteDebuggingServer());
 
   InitializeWebUI();
 
   cast_browser_process_->SetCastService(
       CastService::Create(cast_browser_process_->browser_context()));
   cast_browser_process_->cast_service()->Start();
 }
 
 bool CastBrowserMainParts::MainMessageLoopRun(int* result_code) {
-  base::MessageLoopForUI::current()->Run();
+  // If parameters_.ui_task is not NULL, we are running browser tests. In this
+  // case, the browser's main message loop will not run.
+  if (parameters_.ui_task) {
+    parameters_.ui_task->Run();
+  } else {
+    base::MessageLoopForUI::current()->Run();
+  }
   return true;
 }
 
 void CastBrowserMainParts::PostMainMessageLoopRun() {
   cast_browser_process_->cast_service()->Stop();
   cast_browser_process_.reset();
 }
 
 }  // namespace shell
 }  // namespace chromecast