| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 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 #ifndef BASE_TEST_LAUNCHER_PARALLEL_TEST_LAUNCHER_H_ | |
| 6 #define BASE_TEST_LAUNCHER_PARALLEL_TEST_LAUNCHER_H_ | |
| 7 | |
| 8 #include <map> | |
| 9 #include <string> | |
| 10 | |
| 11 #include "base/callback.h" | |
| 12 #include "base/threading/thread_checker.h" | |
| 13 #include "base/time/time.h" | |
| 14 #include "base/timer/timer.h" | |
| 15 | |
| 16 class CommandLine; | |
| 17 | |
| 18 namespace base { | |
| 19 | |
| 20 class SequencedWorkerPoolOwner; | |
| 21 | |
| 22 // Launches child gtest process in parallel. Keeps track of running processes, | |
| 23 // prints a message in case no output is produced for a while. | |
| 24 class ParallelTestLauncher { | |
| 25 public: | |
| 26 // Constructor. |jobs| is the maximum number of tests launched in parallel. | |
| 27 explicit ParallelTestLauncher(size_t jobs); | |
| 28 ~ParallelTestLauncher(); | |
| 29 | |
| 30 // Callback called after a child process finishes. First argument is the exit | |
| 31 // code, second one is child process elapsed time, third one is true if | |
| 32 // the child process was terminated because of a timeout, and fourth one | |
| 33 // contains output of the child (stdout and stderr together). | |
| 34 typedef Callback<void(int, const TimeDelta&, bool, const std::string&)> | |
| 35 LaunchChildGTestProcessCallback; | |
| 36 | |
| 37 // Launches a child process (assumed to be gtest-based binary) using | |
| 38 // |command_line|. If |wrapper| is not empty, it is prepended to the final | |
| 39 // command line. If the child process is still running after |timeout|, it | |
| 40 // is terminated. After the child process finishes |callback| is called | |
| 41 // on the same thread this method was called. | |
| 42 void LaunchChildGTestProcess(const CommandLine& command_line, | |
| 43 const std::string& wrapper, | |
| 44 base::TimeDelta timeout, | |
| 45 const LaunchChildGTestProcessCallback& callback); | |
| 46 | |
| 47 // Resets the output watchdog, indicating some test results have been printed | |
| 48 // out. If a pause between the calls exceeds an internal treshold, a message | |
| 49 // will be printed listing all child processes we're still waiting for. | |
| 50 void ResetOutputWatchdog(); | |
| 51 | |
| 52 private: | |
| 53 // Called on a worker thread after a child process finishes. | |
| 54 void OnLaunchTestProcessFinished( | |
| 55 size_t sequence_number, | |
| 56 const LaunchChildGTestProcessCallback& callback, | |
| 57 int exit_code, | |
| 58 const TimeDelta& elapsed_time, | |
| 59 bool was_timeout, | |
| 60 const std::string& output); | |
| 61 | |
| 62 // Called by the delay timer when no output was made for a while. | |
| 63 void OnOutputTimeout(); | |
| 64 | |
| 65 // Make sure we don't accidentally call the wrong methods e.g. on the worker | |
| 66 // pool thread. With lots of callbacks used this is non-trivial. | |
| 67 // Should be the first member so that it's destroyed last: when destroying | |
| 68 // other members, especially the worker pool, we may check the code is running | |
| 69 // on the correct thread. | |
| 70 ThreadChecker thread_checker_; | |
| 71 | |
| 72 // Watchdog timer to make sure we do not go without output for too long. | |
| 73 DelayTimer<ParallelTestLauncher> timer_; | |
| 74 | |
| 75 // Monotonically increasing sequence number to uniquely identify each | |
| 76 // launched child process. | |
| 77 size_t launch_sequence_number_; | |
| 78 | |
| 79 // Map of currently running child processes, keyed by the sequence number. | |
| 80 typedef std::map<size_t, CommandLine> RunningProcessesMap; | |
| 81 RunningProcessesMap running_processes_map_; | |
| 82 | |
| 83 // Worker pool used to launch processes in parallel. | |
| 84 scoped_ptr<SequencedWorkerPoolOwner> worker_pool_owner_; | |
| 85 | |
| 86 DISALLOW_COPY_AND_ASSIGN(ParallelTestLauncher); | |
| 87 }; | |
| 88 | |
| 89 } // namespace base | |
| 90 | |
| 91 #endif // BASE_TEST_LAUNCHER_PARALLEL_TEST_LAUNCHER_H_ | |
| OLD | NEW |