| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef BASE_TEST_PERF_TEST_SUITE_H_ | 5 #ifndef BASE_TEST_PERF_TEST_SUITE_H_ |
| 6 #define BASE_TEST_PERF_TEST_SUITE_H_ | 6 #define BASE_TEST_PERF_TEST_SUITE_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 #include "base/basictypes.h" | 9 #include "base/command_line.h" |
| 10 #include "base/debug_util.h" |
| 11 #include "base/file_path.h" |
| 12 #include "base/perftimer.h" |
| 13 #include "base/process_util.h" |
| 14 #include "base/string_util.h" |
| 10 #include "base/test/test_suite.h" | 15 #include "base/test/test_suite.h" |
| 11 | 16 |
| 12 namespace base { | |
| 13 | |
| 14 class PerfTestSuite : public TestSuite { | 17 class PerfTestSuite : public TestSuite { |
| 15 public: | 18 public: |
| 16 PerfTestSuite(int argc, char** argv); | 19 PerfTestSuite(int argc, char** argv) : TestSuite(argc, argv) { |
| 20 } |
| 17 | 21 |
| 18 virtual void Initialize(); | 22 virtual void Initialize() { |
| 19 virtual void Shutdown(); | 23 TestSuite::Initialize(); |
| 20 | 24 |
| 21 private: | 25 // Initialize the perf timer log |
| 22 DISALLOW_COPY_AND_ASSIGN(PerfTestSuite); | 26 FilePath log_path = |
| 27 CommandLine::ForCurrentProcess()->GetSwitchValuePath("log-file"); |
| 28 if (log_path.empty()) { |
| 29 FilePath exe; |
| 30 PathService::Get(base::FILE_EXE, &exe); |
| 31 log_path = exe.ReplaceExtension(FILE_PATH_LITERAL("log")); |
| 32 log_path = log_path.InsertBeforeExtension(FILE_PATH_LITERAL("_perf")); |
| 33 } |
| 34 ASSERT_TRUE(InitPerfLog(log_path)); |
| 35 |
| 36 // Raise to high priority to have more precise measurements. Since we don't |
| 37 // aim at 1% precision, it is not necessary to run at realtime level. |
| 38 if (!DebugUtil::BeingDebugged()) |
| 39 base::RaiseProcessToHighPriority(); |
| 40 } |
| 41 |
| 42 virtual void Shutdown() { |
| 43 TestSuite::Shutdown(); |
| 44 |
| 45 FinalizePerfLog(); |
| 46 } |
| 23 }; | 47 }; |
| 24 | 48 |
| 25 } // namespace base | |
| 26 | |
| 27 #endif // BASE_TEST_PERF_TEST_SUITE_H_ | 49 #endif // BASE_TEST_PERF_TEST_SUITE_H_ |
| OLD | NEW |