xref: /aosp_15_r20/external/cronet/testing/multiprocess_func_list.cc (revision 6777b5387eb2ff775bb5750e3f5d96f37fb7352b) 
1 // Copyright 2012 The Chromium Authors
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 "multiprocess_func_list.h"
6 
7 #include <map>
8 
9 // Helper functions to maintain mapping of "test name"->test func.
10 // The information is accessed via a global map.
11 namespace multi_process_function_list {
12 
13 namespace {
14 
15 ChildProcessTestRunner g_test_runner = nullptr;
16 
17 struct ProcessFunctions {
ProcessFunctionsmulti_process_function_list::__anon823ae8520111::ProcessFunctions18   ProcessFunctions() : main(NULL), setup(NULL) {}
ProcessFunctionsmulti_process_function_list::__anon823ae8520111::ProcessFunctions19   ProcessFunctions(TestMainFunctionPtr main, SetupFunctionPtr setup)
20       : main(main),
21         setup(setup) {
22   }
23   TestMainFunctionPtr main;
24   SetupFunctionPtr setup;
25 };
26 
27 typedef std::map<std::string, ProcessFunctions> MultiProcessTestMap;
28 
29 // Retrieve a reference to the global 'func name' -> func ptr map.
GetMultiprocessFuncMap()30 MultiProcessTestMap& GetMultiprocessFuncMap() {
31   static MultiProcessTestMap test_name_to_func_ptr_map;
32   return test_name_to_func_ptr_map;
33 }
34 
35 }  // namespace
36 
AppendMultiProcessTest(std::string test_name,TestMainFunctionPtr main_func_ptr,SetupFunctionPtr setup_func_ptr)37 AppendMultiProcessTest::AppendMultiProcessTest(
38     std::string test_name,
39     TestMainFunctionPtr main_func_ptr,
40     SetupFunctionPtr setup_func_ptr) {
41   GetMultiprocessFuncMap()[test_name] =
42       ProcessFunctions(main_func_ptr, setup_func_ptr);
43 }
44 
SetChildProcessTestRunner(ChildProcessTestRunner runner)45 void SetChildProcessTestRunner(ChildProcessTestRunner runner) {
46   g_test_runner = runner;
47 }
48 
InvokeChildProcessTest(const std::string & test_name)49 int InvokeChildProcessTest(const std::string& test_name) {
50   if (g_test_runner) {
51     return g_test_runner(test_name);
52   }
53   return InvokeChildProcessTestMain(test_name);
54 }
55 
InvokeChildProcessTestMain(const std::string & test_name)56 int InvokeChildProcessTestMain(const std::string& test_name) {
57   MultiProcessTestMap& func_lookup_table = GetMultiprocessFuncMap();
58   MultiProcessTestMap::iterator it = func_lookup_table.find(test_name);
59   if (it != func_lookup_table.end()) {
60     const ProcessFunctions& process_functions = it->second;
61     if (process_functions.setup)
62       (*process_functions.setup)();
63     if (process_functions.main)
64       return (*process_functions.main)();
65   }
66 
67   return -1;
68 }
69 
70 }  // namespace multi_process_function_list
71