diff --git a/test/main.cpp b/test/main.cpp index 0bf1e8182..68c24dc31 100644 --- a/test/main.cpp +++ b/test/main.cpp @@ -6,9 +6,11 @@ // #include "td/utils/common.h" #include "td/utils/crypto.h" +#include "td/utils/ExitGuard.h" #include "td/utils/logging.h" #include "td/utils/OptionParser.h" #include "td/utils/port/detail/ThreadIdGuard.h" +#include "td/utils/port/stacktrace.h" #include "td/utils/Slice.h" #include "td/utils/tests.h" @@ -17,22 +19,34 @@ #endif int main(int argc, char **argv) { + SET_VERBOSITY_LEVEL(VERBOSITY_NAME(FATAL)); + td::ExitGuard exit_guard; td::detail::ThreadIdGuard thread_id_guard; + td::Stacktrace::init(); td::init_openssl_threads(); td::TestsRunner &runner = td::TestsRunner::get_default(); - SET_VERBOSITY_LEVEL(VERBOSITY_NAME(ERROR)); + int default_verbosity_level = 1; td::OptionParser options; - options.add_option('f', "filter", "Run only specified tests", + options.add_option('f', "filter", "run only specified tests", [&](td::Slice filter) { runner.add_substr_filter(filter.str()); }); - options.add_option('s', "stress", "Run tests infinitely", [&] { runner.set_stress_flag(true); }); + options.add_option('s', "stress", "run tests infinitely", [&] { runner.set_stress_flag(true); }); + options.add_checked_option('v', "verbosity", "log verbosity level", + td::OptionParser::parse_integer(default_verbosity_level)); + options.add_check([&] { + if (default_verbosity_level < 0) { + return td::Status::Error("Wrong verbosity level specified"); + } + return td::Status::OK(); + }); auto r_non_options = options.run(argc, argv, 0); if (r_non_options.is_error()) { LOG(PLAIN) << argv[0] << ": " << r_non_options.error().message(); LOG(PLAIN) << options; return 1; } + SET_VERBOSITY_LEVEL(default_verbosity_level); #if TD_EMSCRIPTEN emscripten_set_main_loop(