| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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 #include "base/file_util.h" | 5 #include "base/file_util.h" |
| 6 #include "base/run_loop.h" | 6 #include "base/run_loop.h" |
| 7 #include "content/browser/tracing/tracing_controller_impl.h" | 7 #include "content/browser/tracing/tracing_controller_impl.h" |
| 8 #include "content/public/test/browser_test_utils.h" | 8 #include "content/public/test/browser_test_utils.h" |
| 9 #include "content/public/test/content_browser_test.h" | 9 #include "content/public/test/content_browser_test.h" |
| 10 #include "content/public/test/content_browser_test_utils.h" | 10 #include "content/public/test/content_browser_test_utils.h" |
| 11 #include "content/shell/browser/shell.h" | 11 #include "content/shell/browser/shell.h" |
| 12 | 12 |
| 13 using base::debug::CategoryFilter; |
| 14 using base::debug::TraceOptions; |
| 15 using base::debug::RECORD_CONTINUOUSLY; |
| 16 using base::debug::RECORD_UNTIL_FULL; |
| 17 |
| 13 namespace content { | 18 namespace content { |
| 14 | 19 |
| 15 class TracingControllerTest : public ContentBrowserTest { | 20 class TracingControllerTest : public ContentBrowserTest { |
| 16 public: | 21 public: |
| 17 TracingControllerTest() {} | 22 TracingControllerTest() {} |
| 18 | 23 |
| 19 virtual void SetUp() OVERRIDE { | 24 virtual void SetUp() OVERRIDE { |
| 20 get_categories_done_callback_count_ = 0; | 25 get_categories_done_callback_count_ = 0; |
| 21 enable_recording_done_callback_count_ = 0; | 26 enable_recording_done_callback_count_ = 0; |
| 22 disable_recording_done_callback_count_ = 0; | 27 disable_recording_done_callback_count_ = 0; |
| (...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 115 | 120 |
| 116 TracingController* controller = TracingController::GetInstance(); | 121 TracingController* controller = TracingController::GetInstance(); |
| 117 | 122 |
| 118 { | 123 { |
| 119 base::RunLoop run_loop; | 124 base::RunLoop run_loop; |
| 120 TracingController::EnableRecordingDoneCallback callback = | 125 TracingController::EnableRecordingDoneCallback callback = |
| 121 base::Bind(&TracingControllerTest::EnableRecordingDoneCallbackTest, | 126 base::Bind(&TracingControllerTest::EnableRecordingDoneCallbackTest, |
| 122 base::Unretained(this), | 127 base::Unretained(this), |
| 123 run_loop.QuitClosure()); | 128 run_loop.QuitClosure()); |
| 124 bool result = controller->EnableRecording( | 129 bool result = controller->EnableRecording( |
| 125 "", TracingController::DEFAULT_OPTIONS, callback); | 130 CategoryFilter(), TraceOptions(), callback); |
| 126 ASSERT_TRUE(result); | 131 ASSERT_TRUE(result); |
| 127 run_loop.Run(); | 132 run_loop.Run(); |
| 128 EXPECT_EQ(enable_recording_done_callback_count(), 1); | 133 EXPECT_EQ(enable_recording_done_callback_count(), 1); |
| 129 } | 134 } |
| 130 | 135 |
| 131 { | 136 { |
| 132 base::RunLoop run_loop; | 137 base::RunLoop run_loop; |
| 133 TracingController::TracingFileResultCallback callback = | 138 TracingController::TracingFileResultCallback callback = |
| 134 base::Bind(&TracingControllerTest::DisableRecordingDoneCallbackTest, | 139 base::Bind(&TracingControllerTest::DisableRecordingDoneCallbackTest, |
| 135 base::Unretained(this), | 140 base::Unretained(this), |
| 136 run_loop.QuitClosure()); | 141 run_loop.QuitClosure()); |
| 137 bool result = controller->DisableRecording(result_file_path, callback); | 142 bool result = controller->DisableRecording(result_file_path, callback); |
| 138 ASSERT_TRUE(result); | 143 ASSERT_TRUE(result); |
| 139 run_loop.Run(); | 144 run_loop.Run(); |
| 140 EXPECT_EQ(disable_recording_done_callback_count(), 1); | 145 EXPECT_EQ(disable_recording_done_callback_count(), 1); |
| 141 } | 146 } |
| 142 } | 147 } |
| 143 | 148 |
| 144 void TestEnableCaptureAndDisableMonitoring( | 149 void TestEnableCaptureAndDisableMonitoring( |
| 145 const base::FilePath& result_file_path) { | 150 const base::FilePath& result_file_path) { |
| 146 Navigate(shell()); | 151 Navigate(shell()); |
| 147 | 152 |
| 148 TracingController* controller = TracingController::GetInstance(); | 153 TracingController* controller = TracingController::GetInstance(); |
| 149 | 154 |
| 150 { | 155 { |
| 151 bool is_monitoring; | 156 bool is_monitoring; |
| 152 std::string category_filter; | 157 CategoryFilter category_filter(""); |
| 153 TracingController::Options options; | 158 TraceOptions options; |
| 154 controller->GetMonitoringStatus(&is_monitoring, | 159 controller->GetMonitoringStatus( |
| 155 &category_filter, | 160 &is_monitoring, &category_filter, &options); |
| 156 &options); | |
| 157 EXPECT_FALSE(is_monitoring); | 161 EXPECT_FALSE(is_monitoring); |
| 158 EXPECT_EQ("-*Debug,-*Test", category_filter); | 162 EXPECT_EQ("-*Debug,-*Test", category_filter.ToString()); |
| 159 EXPECT_FALSE(options & TracingController::ENABLE_SYSTRACE); | 163 EXPECT_FALSE(options.record_mode == RECORD_CONTINUOUSLY); |
| 160 EXPECT_FALSE(options & TracingController::RECORD_CONTINUOUSLY); | 164 EXPECT_FALSE(options.enable_sampling); |
| 161 EXPECT_FALSE(options & TracingController::ENABLE_SAMPLING); | 165 EXPECT_FALSE(options.enable_systrace); |
| 162 } | 166 } |
| 163 | 167 |
| 164 { | 168 { |
| 165 base::RunLoop run_loop; | 169 base::RunLoop run_loop; |
| 166 TracingController::EnableMonitoringDoneCallback callback = | 170 TracingController::EnableMonitoringDoneCallback callback = |
| 167 base::Bind(&TracingControllerTest::EnableMonitoringDoneCallbackTest, | 171 base::Bind(&TracingControllerTest::EnableMonitoringDoneCallbackTest, |
| 168 base::Unretained(this), | 172 base::Unretained(this), |
| 169 run_loop.QuitClosure()); | 173 run_loop.QuitClosure()); |
| 174 |
| 175 TraceOptions trace_options; |
| 176 trace_options.enable_sampling = true; |
| 177 |
| 170 bool result = controller->EnableMonitoring( | 178 bool result = controller->EnableMonitoring( |
| 171 "*", TracingController::ENABLE_SAMPLING, callback); | 179 CategoryFilter("*"), |
| 180 trace_options, |
| 181 callback); |
| 172 ASSERT_TRUE(result); | 182 ASSERT_TRUE(result); |
| 173 run_loop.Run(); | 183 run_loop.Run(); |
| 174 EXPECT_EQ(enable_monitoring_done_callback_count(), 1); | 184 EXPECT_EQ(enable_monitoring_done_callback_count(), 1); |
| 175 } | 185 } |
| 176 | 186 |
| 177 { | 187 { |
| 178 bool is_monitoring; | 188 bool is_monitoring; |
| 179 std::string category_filter; | 189 CategoryFilter category_filter(""); |
| 180 TracingController::Options options; | 190 TraceOptions options; |
| 181 controller->GetMonitoringStatus(&is_monitoring, | 191 controller->GetMonitoringStatus( |
| 182 &category_filter, | 192 &is_monitoring, &category_filter, &options); |
| 183 &options); | |
| 184 EXPECT_TRUE(is_monitoring); | 193 EXPECT_TRUE(is_monitoring); |
| 185 EXPECT_EQ("*", category_filter); | 194 EXPECT_EQ("*", category_filter.ToString()); |
| 186 EXPECT_FALSE(options & TracingController::ENABLE_SYSTRACE); | 195 EXPECT_FALSE(options.record_mode == RECORD_CONTINUOUSLY); |
| 187 EXPECT_FALSE(options & TracingController::RECORD_CONTINUOUSLY); | 196 EXPECT_TRUE(options.enable_sampling); |
| 188 EXPECT_TRUE(options & TracingController::ENABLE_SAMPLING); | 197 EXPECT_FALSE(options.enable_systrace); |
| 189 } | 198 } |
| 190 | 199 |
| 191 { | 200 { |
| 192 base::RunLoop run_loop; | 201 base::RunLoop run_loop; |
| 193 TracingController::TracingFileResultCallback callback = | 202 TracingController::TracingFileResultCallback callback = |
| 194 base::Bind(&TracingControllerTest:: | 203 base::Bind(&TracingControllerTest:: |
| 195 CaptureMonitoringSnapshotDoneCallbackTest, | 204 CaptureMonitoringSnapshotDoneCallbackTest, |
| 196 base::Unretained(this), | 205 base::Unretained(this), |
| 197 run_loop.QuitClosure()); | 206 run_loop.QuitClosure()); |
| 198 ASSERT_TRUE(controller->CaptureMonitoringSnapshot(result_file_path, | 207 ASSERT_TRUE(controller->CaptureMonitoringSnapshot(result_file_path, |
| 199 callback)); | 208 callback)); |
| 200 run_loop.Run(); | 209 run_loop.Run(); |
| 201 EXPECT_EQ(capture_monitoring_snapshot_done_callback_count(), 1); | 210 EXPECT_EQ(capture_monitoring_snapshot_done_callback_count(), 1); |
| 202 } | 211 } |
| 203 | 212 |
| 204 { | 213 { |
| 205 base::RunLoop run_loop; | 214 base::RunLoop run_loop; |
| 206 TracingController::DisableMonitoringDoneCallback callback = | 215 TracingController::DisableMonitoringDoneCallback callback = |
| 207 base::Bind(&TracingControllerTest::DisableMonitoringDoneCallbackTest, | 216 base::Bind(&TracingControllerTest::DisableMonitoringDoneCallbackTest, |
| 208 base::Unretained(this), | 217 base::Unretained(this), |
| 209 run_loop.QuitClosure()); | 218 run_loop.QuitClosure()); |
| 210 bool result = controller->DisableMonitoring(callback); | 219 bool result = controller->DisableMonitoring(callback); |
| 211 ASSERT_TRUE(result); | 220 ASSERT_TRUE(result); |
| 212 run_loop.Run(); | 221 run_loop.Run(); |
| 213 EXPECT_EQ(disable_monitoring_done_callback_count(), 1); | 222 EXPECT_EQ(disable_monitoring_done_callback_count(), 1); |
| 214 } | 223 } |
| 215 | 224 |
| 216 { | 225 { |
| 217 bool is_monitoring; | 226 bool is_monitoring; |
| 218 std::string category_filter; | 227 CategoryFilter category_filter(""); |
| 219 TracingController::Options options; | 228 TraceOptions options; |
| 220 controller->GetMonitoringStatus(&is_monitoring, | 229 controller->GetMonitoringStatus(&is_monitoring, |
| 221 &category_filter, | 230 &category_filter, |
| 222 &options); | 231 &options); |
| 223 EXPECT_FALSE(is_monitoring); | 232 EXPECT_FALSE(is_monitoring); |
| 224 EXPECT_EQ("", category_filter); | 233 EXPECT_EQ("", category_filter.ToString()); |
| 225 EXPECT_FALSE(options & TracingController::ENABLE_SYSTRACE); | 234 EXPECT_FALSE(options.record_mode == RECORD_CONTINUOUSLY); |
| 226 EXPECT_FALSE(options & TracingController::RECORD_CONTINUOUSLY); | 235 EXPECT_FALSE(options.enable_sampling); |
| 227 EXPECT_FALSE(options & TracingController::ENABLE_SAMPLING); | 236 EXPECT_FALSE(options.enable_systrace); |
| 228 } | 237 } |
| 229 } | 238 } |
| 230 | 239 |
| 231 private: | 240 private: |
| 232 int get_categories_done_callback_count_; | 241 int get_categories_done_callback_count_; |
| 233 int enable_recording_done_callback_count_; | 242 int enable_recording_done_callback_count_; |
| 234 int disable_recording_done_callback_count_; | 243 int disable_recording_done_callback_count_; |
| 235 int enable_monitoring_done_callback_count_; | 244 int enable_monitoring_done_callback_count_; |
| 236 int disable_monitoring_done_callback_count_; | 245 int disable_monitoring_done_callback_count_; |
| 237 int capture_monitoring_snapshot_done_callback_count_; | 246 int capture_monitoring_snapshot_done_callback_count_; |
| (...skipping 27 matching lines...) Expand all Loading... |
| 265 TestEnableAndDisableRecording(file_path); | 274 TestEnableAndDisableRecording(file_path); |
| 266 EXPECT_EQ(file_path.value(), last_actual_recording_file_path().value()); | 275 EXPECT_EQ(file_path.value(), last_actual_recording_file_path().value()); |
| 267 } | 276 } |
| 268 | 277 |
| 269 IN_PROC_BROWSER_TEST_F(TracingControllerTest, | 278 IN_PROC_BROWSER_TEST_F(TracingControllerTest, |
| 270 EnableAndDisableRecordingWithEmptyFileAndNullCallback) { | 279 EnableAndDisableRecordingWithEmptyFileAndNullCallback) { |
| 271 Navigate(shell()); | 280 Navigate(shell()); |
| 272 | 281 |
| 273 TracingController* controller = TracingController::GetInstance(); | 282 TracingController* controller = TracingController::GetInstance(); |
| 274 EXPECT_TRUE(controller->EnableRecording( | 283 EXPECT_TRUE(controller->EnableRecording( |
| 275 "", TracingController::DEFAULT_OPTIONS, | 284 CategoryFilter(), |
| 285 TraceOptions(), |
| 276 TracingController::EnableRecordingDoneCallback())); | 286 TracingController::EnableRecordingDoneCallback())); |
| 277 EXPECT_TRUE(controller->DisableRecording( | 287 EXPECT_TRUE(controller->DisableRecording( |
| 278 base::FilePath(), TracingController::TracingFileResultCallback())); | 288 base::FilePath(), TracingController::TracingFileResultCallback())); |
| 279 base::RunLoop().RunUntilIdle(); | 289 base::RunLoop().RunUntilIdle(); |
| 280 } | 290 } |
| 281 | 291 |
| 282 IN_PROC_BROWSER_TEST_F(TracingControllerTest, | 292 IN_PROC_BROWSER_TEST_F(TracingControllerTest, |
| 283 EnableCaptureAndDisableMonitoring) { | 293 EnableCaptureAndDisableMonitoring) { |
| 284 TestEnableCaptureAndDisableMonitoring(base::FilePath()); | 294 TestEnableCaptureAndDisableMonitoring(base::FilePath()); |
| 285 } | 295 } |
| 286 | 296 |
| 287 IN_PROC_BROWSER_TEST_F(TracingControllerTest, | 297 IN_PROC_BROWSER_TEST_F(TracingControllerTest, |
| 288 EnableCaptureAndDisableMonitoringWithFilePath) { | 298 EnableCaptureAndDisableMonitoringWithFilePath) { |
| 289 base::FilePath file_path; | 299 base::FilePath file_path; |
| 290 base::CreateTemporaryFile(&file_path); | 300 base::CreateTemporaryFile(&file_path); |
| 291 TestEnableCaptureAndDisableMonitoring(file_path); | 301 TestEnableCaptureAndDisableMonitoring(file_path); |
| 292 EXPECT_EQ(file_path.value(), last_actual_monitoring_file_path().value()); | 302 EXPECT_EQ(file_path.value(), last_actual_monitoring_file_path().value()); |
| 293 } | 303 } |
| 294 | 304 |
| 295 IN_PROC_BROWSER_TEST_F( | 305 IN_PROC_BROWSER_TEST_F( |
| 296 TracingControllerTest, | 306 TracingControllerTest, |
| 297 EnableCaptureAndDisableMonitoringWithEmptyFileAndNullCallback) { | 307 EnableCaptureAndDisableMonitoringWithEmptyFileAndNullCallback) { |
| 298 Navigate(shell()); | 308 Navigate(shell()); |
| 299 | 309 |
| 300 TracingController* controller = TracingController::GetInstance(); | 310 TracingController* controller = TracingController::GetInstance(); |
| 311 TraceOptions trace_options; |
| 312 trace_options.enable_sampling = true; |
| 301 EXPECT_TRUE(controller->EnableMonitoring( | 313 EXPECT_TRUE(controller->EnableMonitoring( |
| 302 "*", TracingController::ENABLE_SAMPLING, | 314 CategoryFilter("*"), |
| 315 trace_options, |
| 303 TracingController::EnableMonitoringDoneCallback())); | 316 TracingController::EnableMonitoringDoneCallback())); |
| 304 controller->CaptureMonitoringSnapshot( | 317 controller->CaptureMonitoringSnapshot( |
| 305 base::FilePath(), TracingController::TracingFileResultCallback()); | 318 base::FilePath(), TracingController::TracingFileResultCallback()); |
| 306 base::RunLoop().RunUntilIdle(); | 319 base::RunLoop().RunUntilIdle(); |
| 307 EXPECT_TRUE(controller->DisableMonitoring( | 320 EXPECT_TRUE(controller->DisableMonitoring( |
| 308 TracingController::DisableMonitoringDoneCallback())); | 321 TracingController::DisableMonitoringDoneCallback())); |
| 309 base::RunLoop().RunUntilIdle(); | 322 base::RunLoop().RunUntilIdle(); |
| 310 } | 323 } |
| 311 | 324 |
| 312 } // namespace content | 325 } // namespace content |
| OLD | NEW |