src/mojo/shell/shell_test_main.cc

   1 // Copyright 2014 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/bind.h"
   6 #include "base/command_line.h"
   7 #include "base/logging.h"
   8 #include "base/test/launcher/unit_test_launcher.h"
   9 #include "base/test/test_suite.h"
  10 #include "mojo/shell/child_process.h"
  11 #include "mojo/shell/switches.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 int main(int argc, char** argv) {
  15   base::CommandLine::Init(argc, argv);
  16   const base::CommandLine& command_line =
  17       *base::CommandLine::ForCurrentProcess();
  18
  19   if (command_line.HasSwitch(switches::kChildProcessType)) {
  20     scoped_ptr<mojo::shell::ChildProcess> child_process =
  21         mojo::shell::ChildProcess::Create(command_line);
  22     CHECK(child_process);
  23     child_process->Main();
  24     return 0;
  25   }
  26
  27   base::TestSuite test_suite(argc, argv);
  28   return base::LaunchUnitTests(
  29       argc, argv, base::Bind(&base::TestSuite::Run,
  30                              base::Unretained(&test_suite)));
  31 }