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-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "LibcTest.h" 10 #include "src/__support/CPP/string_view.h" 11 12 using LIBC_NAMESPACE::cpp::string_view; 13 using LIBC_NAMESPACE::testing::TestOptions; 14 15 namespace { 16 17 // A poor-man's getopt_long. 18 // Run unit tests with --gtest_color=no to disable printing colors, or 19 // --gtest_print_time to print timings in milliseconds only (as GTest does, so 20 // external tools such as Android's atest may expect that format to parse the 21 // output). Other command line flags starting with --gtest_ are ignored. 22 // Otherwise, the last command line arg is used as a test filter, if command 23 // line args are specified. 24 TestOptions 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 else 38 Options.TestFilter = argv[i]; 39 } 40 41 return Options; 42 } 43 44 } // anonymous namespace 45 46 // The C++ standard forbids declaring the main function with a linkage specifier 47 // outisde of 'freestanding' mode, only define the linkage for hermetic tests. 48 #if __STDC_HOSTED__ 49 #define TEST_MAIN int main 50 #else 51 #define TEST_MAIN extern "C" int main 52 #endif 53 54 TEST_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