Home | History | Annotate | Download | only in test 
      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.
      4 
      5 #include "base/test/multiprocess_test.h"
      6 
      7 #include "base/base_switches.h"
      8 #include "base/command_line.h"
      9 #include "base/files/file_path.h"
     10 #include "base/files/file_util.h"
     11 #include "build/build_config.h"
     12 
     13 namespace base {
     14 
     15 #if !defined(OS_ANDROID) && !defined(__ANDROID__) && !defined(__ANDROID_HOST__)
     16 SpawnChildResult SpawnMultiProcessTestChild(
     17     const std::string& procname,
     18     const CommandLine& base_command_line,
     19     const LaunchOptions& options) {
     20   CommandLine command_line(base_command_line);
     21   // TODO(viettrungluu): See comment above |MakeCmdLine()| in the header file.
     22   // This is a temporary hack, since |MakeCmdLine()| has to provide a full
     23   // command line.
     24   if (!command_line.HasSwitch(switches::kTestChildProcess))
     25     command_line.AppendSwitchASCII(switches::kTestChildProcess, procname);
     26 
     27   SpawnChildResult result;
     28   result.process = LaunchProcess(command_line, options);
     29   return result;
     30 }
     31 
     32 bool WaitForMultiprocessTestChildExit(const Process& process,
     33                                       TimeDelta timeout,
     34                                       int* exit_code) {
     35   return process.WaitForExitWithTimeout(timeout, exit_code);
     36 }
     37 
     38 bool TerminateMultiProcessTestChild(const Process& process,
     39                                     int exit_code,
     40                                     bool wait) {
     41   return process.Terminate(exit_code, wait);
     42 }
     43 
     44 #endif  // !OS_ANDROID && !__ANDROID__ && !__ANDROID_HOST__
     45 
     46 CommandLine GetMultiProcessTestChildBaseCommandLine() {
     47   CommandLine cmd_line = *CommandLine::ForCurrentProcess();
     48   cmd_line.SetProgram(MakeAbsoluteFilePath(cmd_line.GetProgram()));
     49   return cmd_line;
     50 }
     51 
     52 // MultiProcessTest ------------------------------------------------------------
     53 
     54 MultiProcessTest::MultiProcessTest() {
     55 }
     56 
     57 // Don't compile on Arc++.
     58 #if 0
     59 SpawnChildResult MultiProcessTest::SpawnChild(const std::string& procname) {
     60   LaunchOptions options;
     61 #if defined(OS_WIN)
     62   options.start_hidden = true;
     63 #endif
     64   return SpawnChildWithOptions(procname, options);
     65 }
     66 
     67 SpawnChildResult MultiProcessTest::SpawnChildWithOptions(
     68     const std::string& procname,
     69     const LaunchOptions& options) {
     70   return SpawnMultiProcessTestChild(procname, MakeCmdLine(procname), options);
     71 }
     72 #endif
     73 
     74 CommandLine MultiProcessTest::MakeCmdLine(const std::string& procname) {
     75   CommandLine command_line = GetMultiProcessTestChildBaseCommandLine();
     76   command_line.AppendSwitchASCII(switches::kTestChildProcess, procname);
     77   return command_line;
     78 }
     79 
     80 }  // namespace base
     81