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/logging.h"
   7 #include "base/test/test_suite.h"
   8 #include "sandbox/linux/tests/test_utils.h"
   9 #include "sandbox/linux/tests/unit_tests.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace sandbox {
  13 namespace {
  14
  15 // Check for leaks in our tests.
  16 void RunPostTestsChecks() {
  17   if (TestUtils::CurrentProcessHasChildren()) {
  18     LOG(ERROR) << "One of the tests created a child that was not waited for. "
  19                << "Please, clean-up after your tests!";
  20   }
  21 }
  22
  23 }  // namespace
  24 }  // namespace sandbox
  25
  26 #if defined(OS_ANDROID)
  27 void UnitTestAssertHandler(const std::string& str) {
  28   _exit(1);
  29 }
  30 #endif
  31
  32 int main(int argc, char* argv[]) {
  33 #if defined(OS_ANDROID)
  34   // The use of Callbacks requires an AtExitManager.
  35   base::AtExitManager exit_manager;
  36   testing::InitGoogleTest(&argc, argv);
  37   // Death tests rely on LOG(FATAL) triggering an exit (the default behavior is
  38   // SIGABRT).  The normal test launcher does this at initialization, but since
  39   // we still do not use this on Android, we must install the handler ourselves.
  40   logging::SetLogAssertHandler(UnitTestAssertHandler);
  41 #endif
  42   // Always go through re-execution for death tests.
  43   // This makes gtest only marginally slower for us and has the
  44   // additional side effect of getting rid of gtest warnings about fork()
  45   // safety.
  46   ::testing::FLAGS_gtest_death_test_style = "threadsafe";
  47 #if defined(OS_ANDROID)
  48   int tests_result = RUN_ALL_TESTS();
  49 #else
  50   int tests_result = base::RunUnitTestsUsingBaseTestSuite(argc, argv);
  51 #endif
  52
  53   sandbox::RunPostTestsChecks();
  54   return tests_result;
  55 }