OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
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 struct ProcessFunctions { | |
16 ProcessFunctions() : main(NULL), setup(NULL) {} | |
17 ProcessFunctions(TestMainFunctionPtr main, SetupFunctionPtr setup) | |
18 : main(main), setup(setup) {} | |
19 TestMainFunctionPtr main; | |
20 SetupFunctionPtr setup; | |
21 }; | |
22 | |
23 typedef std::map<std::string, ProcessFunctions> MultiProcessTestMap; | |
24 | |
25 // Retrieve a reference to the global 'func name' -> func ptr map. | |
26 MultiProcessTestMap& GetMultiprocessFuncMap() { | |
27 static MultiProcessTestMap test_name_to_func_ptr_map; | |
28 return test_name_to_func_ptr_map; | |
29 } | |
30 | |
31 } // namespace | |
32 | |
33 AppendMultiProcessTest::AppendMultiProcessTest( | |
34 std::string test_name, | |
35 TestMainFunctionPtr main_func_ptr, | |
36 SetupFunctionPtr setup_func_ptr) { | |
37 GetMultiprocessFuncMap()[test_name] = | |
38 ProcessFunctions(main_func_ptr, setup_func_ptr); | |
39 } | |
40 | |
41 int InvokeChildProcessTest(std::string test_name) { | |
42 MultiProcessTestMap& func_lookup_table = GetMultiprocessFuncMap(); | |
43 MultiProcessTestMap::iterator it = func_lookup_table.find(test_name); | |
44 if (it != func_lookup_table.end()) { | |
45 const ProcessFunctions& process_functions = it->second; | |
46 if (process_functions.setup) | |
47 (*process_functions.setup)(); | |
48 if (process_functions.main) | |
49 return (*process_functions.main)(); | |
50 } | |
51 | |
52 return -1; | |
53 } | |
54 | |
55 } // namespace multi_process_function_list | |
OLD | NEW |