61 lines · cpp
1//===-- Main function for implementation of base class for libc unittests -===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "LibcTest.h"10#include "src/__support/CPP/string_view.h"11 12using LIBC_NAMESPACE::cpp::string_view;13using LIBC_NAMESPACE::testing::TestOptions;14 15namespace {16 17// A poor-man's getopt_long.18// Run unit tests with --gtest_color=no to disable printing colors, or19// --gtest_print_time to print timings in milliseconds only (as GTest does, so20// external tools such as Android's atest may expect that format to parse the21// output). Other command line flags starting with --gtest_ are ignored.22// Otherwise, the last command line arg is used as a test filter, if command23// line args are specified.24TestOptions parseOptions(int argc, char **argv) {25 TestOptions Options;26 27 for (int i = 1; i < argc; ++i) {28 string_view arg{argv[i]};29 30 if (arg == "--gtest_color=no")31 Options.PrintColor = false;32 else if (arg == "--gtest_print_time")33 Options.TimeInMs = true;34 // Ignore other unsupported gtest specific flags.35 else if (arg.starts_with("--gtest_"))36 continue;37 else38 Options.TestFilter = argv[i];39 }40 41 return Options;42}43 44} // anonymous namespace45 46// The C++ standard forbids declaring the main function with a linkage specifier47// outisde of 'freestanding' mode, only define the linkage for hermetic tests.48#if __STDC_HOSTED__49#define TEST_MAIN int main50#else51#define TEST_MAIN extern "C" int main52#endif53 54TEST_MAIN(int argc, char **argv, char **envp) {55 LIBC_NAMESPACE::testing::argc = argc;56 LIBC_NAMESPACE::testing::argv = argv;57 LIBC_NAMESPACE::testing::envp = envp;58 59 return LIBC_NAMESPACE::testing::Test::runTests(parseOptions(argc, argv));60}61