Makes it possible to run content_browsertests with --as-browser and fake WebRTC devices.

content/public/test/test_launcher.cc

 // Copyright (c) 2012 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 "content/public/test/test_launcher.h"
 
 #include <string>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/environment.h"
 #include "base/file_util.h"
 #include "base/hash_tables.h"
 #include "base/logging.h"
 #include "base/memory/linked_ptr.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/process_util.h"
 #include "base/scoped_temp_dir.h"
 #include "base/string_number_conversions.h"
 #include "base/string_util.h"
 #include "base/test/test_suite.h"
 #include "base/test/test_timeouts.h"
 #include "base/time.h"
 #include "base/utf_string_conversions.h"
+#include "content/public/app/content_main.h"
+#include "content/public/app/content_main_delegate.h"
 #include "content/public/app/startup_helper_win.h"
+#include "content/public/common/content_switches.h"
 #include "content/public/common/sandbox_init.h"
 #include "content/public/test/browser_test.h"
 #include "net/base/escape.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #if defined(OS_WIN)
 #include "base/base_switches.h"
 #include "content/common/sandbox_policy.h"
 #include "sandbox/win/src/dep.h"
 #include "sandbox/win/src/sandbox_factory.h"
@@ skipping 579 matching lines @@
       "  --single-process\n"
       "    Same as above, and also runs Chrome in single-process mode.\n"
       "  --help\n"
       "    Shows this message.\n"
       "  --gtest_help\n"
       "    Shows the gtest help message.\n");
 }
 
 }  // namespace
 
+// The following is kept for historical reasons (so people that are used to
+// using it don't get surprised).
+const char kChildProcessFlag[]   = "child";

[tommi (sloooow) - chröme, 2012/09/06 09:57:29]

only one space on each side of =

+
 const char kGTestFilterFlag[] = "gtest_filter";
 const char kGTestHelpFlag[]   = "gtest_help";

[tommi (sloooow) - chröme, 2012/09/06 09:57:29]

please fix here as well and for "help" + "single-process" + "single_process"
below.

 const char kGTestListTestsFlag[] = "gtest_list_tests";
 const char kGTestRepeatFlag[] = "gtest_repeat";
 const char kGTestRunDisabledTestsFlag[] = "gtest_also_run_disabled_tests";
 const char kGTestOutputFlag[] = "gtest_output";
 
+const char kHelpFlag[]   = "help";
+
+const char kLaunchAsBrowser[] = "as-browser";
+
+// See kManualTestPrefix above.
+const char kRunManualTestsFlag[] = "run-manual";
+
 const char kSingleProcessTestsFlag[]   = "single_process";
 const char kSingleProcessTestsAndChromeFlag[]   = "single-process";
 
-// See kManualTestPrefix above.
-const char kRunManualTestsFlag[] = "run-manual";
-
-// The following is kept for historical reasons (so people that are used to
-// using it don't get surprised).
-const char kChildProcessFlag[]   = "child";
-
-const char kHelpFlag[]   = "help";
-
 const char kWarmupFlag[] = "warmup";
 
+
 TestLauncherDelegate::~TestLauncherDelegate() {
 }
 
+bool ShouldRunContentMain() {
+#if defined(OS_WIN) || defined(OS_LINUX)
+  CommandLine* command_line = CommandLine::ForCurrentProcess();
+  return command_line->HasSwitch(switches::kProcessType) ||
+         command_line->HasSwitch(kLaunchAsBrowser);
+#else
+  return false;
+#endif  // defined(OS_WIN) || defined(OS_LINUX)
+}
+
+int RunContentMain(int argc, char** argv,
+                   TestLauncherDelegate* launcher_delegate) {
+#if defined(OS_WIN)
+  sandbox::SandboxInterfaceInfo sandbox_info = {0};
+  content::InitializeSandboxInfo(&sandbox_info);
+  scoped_ptr<content::ContentMainDelegate> chrome_main_delegate(
+      launcher_delegate->CreateContentMainDelegate());
+  return content::ContentMain(GetModuleHandle(NULL),
+                              &sandbox_info,
+                              chrome_main_delegate.get());
+#elif defined(OS_LINUX)
+  scoped_ptr<content::ContentMainDelegate> chrome_main_delegate(
+      launcher_delegate->CreateContentMainDelegate());
+  return content::ContentMain(argc,
+                              const_cast<const char**>(argv),
+                              chrome_main_delegate.get());
+#endif  // defined(OS_WIN)
+  NOTREACHED();
+  return 0;
+}
+
 int LaunchTests(TestLauncherDelegate* launcher_delegate,
                 int argc,
                 char** argv) {
   DCHECK(!g_launcher_delegate);
   g_launcher_delegate = launcher_delegate;
 
   CommandLine::Init(argc, argv);
   const CommandLine* command_line = CommandLine::ForCurrentProcess();
 
   if (command_line->HasSwitch(kHelpFlag)) {
@@ skipping 12 matching lines @@
 #if defined(OS_WIN)
     if (command_line->HasSwitch(kSingleProcessTestsFlag)) {
       sandbox::SandboxInterfaceInfo sandbox_info;
       content::InitializeSandboxInfo(&sandbox_info);
       content::InitializeSandbox(&sandbox_info);
     }
 #endif
     return launcher_delegate->RunTestSuite(argc, argv);
   }
 
-  int return_code = 0;
-  if (launcher_delegate->Run(argc, argv, &return_code))
-    return return_code;
+  if (ShouldRunContentMain())
+    return RunContentMain(argc, argv, launcher_delegate);
 
   base::AtExitManager at_exit;
 
   int32 total_shards;
   int32 shard_index;
   bool should_shard = ShouldShard(&total_shards, &shard_index);
 
   fprintf(stdout,
       "Starting tests...\n"
       "IMPORTANT DEBUGGING NOTE: each test is run inside its own process.\n"
@@ skipping 49 matching lines @@
       cycles--;
   }
   return exit_code;
 }
 
 TestLauncherDelegate* GetCurrentTestLauncherDelegate() {
   return g_launcher_delegate;
 }
 
 }  // namespace test_launcher