xref: /aosp_15_r20/external/webrtc/test/test_main.cc (revision d9f758449e529ab9291ac668be2861e7a55c2422)
1 /*
2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #include <memory>
12 #include <regex>
13 #include <string>
14 #include <vector>
15 
16 #include "absl/debugging/failure_signal_handler.h"
17 #include "absl/debugging/symbolize.h"
18 #include "absl/flags/parse.h"
19 #include "test/gmock.h"
20 #include "test/test_main_lib.h"
21 
22 namespace {
23 
ReplaceDashesWithUnderscores(int argc,char * argv[])24 std::vector<std::string> ReplaceDashesWithUnderscores(int argc, char* argv[]) {
25   std::vector<std::string> args(argv, argv + argc);
26   for (std::string& arg : args) {
27     // Only replace arguments that starts with a dash.
28     if (!arg.empty() && arg[0] == '-') {
29       // Don't replace the 2 first characters.
30       auto begin = arg.begin() + 2;
31       // Replace dashes on the left of '=' or on all the arg if no '=' is found.
32       auto end = std::find(arg.begin(), arg.end(), '=');
33       std::replace(begin, end, '-', '_');
34     }
35   }
36   return args;
37 }
38 
VectorOfStringsToVectorOfPointers(std::vector<std::string> & input)39 std::vector<char*> VectorOfStringsToVectorOfPointers(
40     std::vector<std::string>& input) {
41   std::vector<char*> output(input.size());
42   for (size_t i = 0; i < input.size(); ++i) {
43     output[i] = &(input[i][0]);
44   }
45   return output;
46 }
47 
48 }  // namespace
49 
main(int argc,char * argv[])50 int main(int argc, char* argv[]) {
51   // Initialize the symbolizer to get a human-readable stack trace
52   absl::InitializeSymbolizer(argv[0]);
53   testing::InitGoogleMock(&argc, argv);
54   // Before parsing the arguments with the absl flag library, any internal '-'
55   // characters will be converted to '_' characters to make sure the string is a
56   // valid attribute name.
57   std::vector<std::string> new_argv = ReplaceDashesWithUnderscores(argc, argv);
58   std::vector<char*> raw_new_argv = VectorOfStringsToVectorOfPointers(new_argv);
59   absl::ParseCommandLine(argc, &raw_new_argv[0]);
60 
61 // This absl handler use unsupported features/instructions on Fuchsia
62 #if !defined(WEBRTC_FUCHSIA)
63   absl::FailureSignalHandlerOptions options;
64   absl::InstallFailureSignalHandler(options);
65 #endif
66 
67   std::unique_ptr<webrtc::TestMain> main = webrtc::TestMain::Create();
68   int err_code = main->Init();
69   if (err_code != 0) {
70     return err_code;
71   }
72   return main->Run(argc, argv);
73 }
74