| 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 |