sandbox/linux/tests/main.cc

   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/at_exit.h"
   6 #include "base/base_switches.h"
   7 #include "base/command_line.h"
   8 #include "base/files/file_path.h"
   9 #include "base/files/file_util.h"
  10 #include "base/logging.h"
  11 #include "base/test/test_suite.h"
  12 #include "build/build_config.h"
  13 #include "sandbox/linux/tests/test_utils.h"
  14 #include "sandbox/linux/tests/unit_tests.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16 #include "testing/multiprocess_func_list.h"
  17
  18 namespace sandbox {
  19 namespace {
  20
  21 // Check for leaks in our tests.
  22 void RunPostTestsChecks(const base::FilePath& orig_cwd) {
  23   if (TestUtils::CurrentProcessHasChildren()) {
  24     LOG(FATAL) << "One of the tests created a child that was not waited for. "
  25                << "Please, clean up after your tests!";
  26   }
  27
  28   base::FilePath cwd;
  29   CHECK(GetCurrentDirectory(&cwd));
  30   if (orig_cwd != cwd) {
  31     LOG(FATAL) << "One of the tests changed the current working directory. "
  32                << "Please, clean up after your tests!";
  33   }
  34 }
  35
  36 }  // namespace
  37 }  // namespace sandbox
  38
  39 #if !defined(SANDBOX_USES_BASE_TEST_SUITE)
  40 void UnitTestAssertHandler(const std::string& str) {
  41   _exit(1);
  42 }
  43 #endif
  44
  45 int main(int argc, char* argv[]) {
  46   base::CommandLine::Init(argc, argv);
  47   std::string client_func;
  48 #if defined(SANDBOX_USES_BASE_TEST_SUITE)
  49   client_func = base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
  50       switches::kTestChildProcess);
  51 #endif
  52   if (!client_func.empty()) {
  53     base::AtExitManager exit_manager;
  54     return multi_process_function_list::InvokeChildProcessTest(client_func);
  55   }
  56
  57   base::FilePath orig_cwd;
  58   CHECK(GetCurrentDirectory(&orig_cwd));
  59
  60 #if !defined(SANDBOX_USES_BASE_TEST_SUITE)
  61   // The use of Callbacks requires an AtExitManager.
  62   base::AtExitManager exit_manager;
  63   testing::InitGoogleTest(&argc, argv);
  64   // Death tests rely on LOG(FATAL) triggering an exit (the default behavior is
  65   // SIGABRT).  The normal test launcher does this at initialization, but since
  66   // we still do not use this on Android, we must install the handler ourselves.
  67   logging::SetLogAssertHandler(UnitTestAssertHandler);
  68 #endif
  69   // Always go through re-execution for death tests.
  70   // This makes gtest only marginally slower for us and has the
  71   // additional side effect of getting rid of gtest warnings about fork()
  72   // safety.
  73   ::testing::FLAGS_gtest_death_test_style = "threadsafe";
  74 #if !defined(SANDBOX_USES_BASE_TEST_SUITE)
  75   int tests_result = RUN_ALL_TESTS();
  76 #else
  77   int tests_result = base::RunUnitTestsUsingBaseTestSuite(argc, argv);
  78 #endif
  79
  80   sandbox::RunPostTestsChecks(orig_cwd);
  81   return tests_result;
  82 }