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()); |
170 bool result = controller->EnableMonitoring( | 174 bool result = controller->EnableMonitoring( |
171 "*", TracingController::ENABLE_SAMPLING, callback); | 175 CategoryFilter("*"), |
| 176 TraceOptions(RECORD_UNTIL_FULL).EnableSampling(true), |
| 177 callback); |
172 ASSERT_TRUE(result); | 178 ASSERT_TRUE(result); |
173 run_loop.Run(); | 179 run_loop.Run(); |
174 EXPECT_EQ(enable_monitoring_done_callback_count(), 1); | 180 EXPECT_EQ(enable_monitoring_done_callback_count(), 1); |
175 } | 181 } |
176 | 182 |
177 { | 183 { |
178 bool is_monitoring; | 184 bool is_monitoring; |
179 std::string category_filter; | 185 CategoryFilter category_filter(""); |
180 TracingController::Options options; | 186 TraceOptions options; |
181 controller->GetMonitoringStatus(&is_monitoring, | 187 controller->GetMonitoringStatus( |
182 &category_filter, | 188 &is_monitoring, &category_filter, &options); |
183 &options); | |
184 EXPECT_TRUE(is_monitoring); | 189 EXPECT_TRUE(is_monitoring); |
185 EXPECT_EQ("*", category_filter); | 190 EXPECT_EQ("*", category_filter.ToString()); |
186 EXPECT_FALSE(options & TracingController::ENABLE_SYSTRACE); | 191 EXPECT_FALSE(options.record_mode == RECORD_CONTINUOUSLY); |
187 EXPECT_FALSE(options & TracingController::RECORD_CONTINUOUSLY); | 192 EXPECT_TRUE(options.enable_sampling); |
188 EXPECT_TRUE(options & TracingController::ENABLE_SAMPLING); | 193 EXPECT_FALSE(options.enable_systrace); |
189 } | 194 } |
190 | 195 |
191 { | 196 { |
192 base::RunLoop run_loop; | 197 base::RunLoop run_loop; |
193 TracingController::TracingFileResultCallback callback = | 198 TracingController::TracingFileResultCallback callback = |
194 base::Bind(&TracingControllerTest:: | 199 base::Bind(&TracingControllerTest:: |
195 CaptureMonitoringSnapshotDoneCallbackTest, | 200 CaptureMonitoringSnapshotDoneCallbackTest, |
196 base::Unretained(this), | 201 base::Unretained(this), |
197 run_loop.QuitClosure()); | 202 run_loop.QuitClosure()); |
198 ASSERT_TRUE(controller->CaptureMonitoringSnapshot(result_file_path, | 203 ASSERT_TRUE(controller->CaptureMonitoringSnapshot(result_file_path, |
199 callback)); | 204 callback)); |
200 run_loop.Run(); | 205 run_loop.Run(); |
201 EXPECT_EQ(capture_monitoring_snapshot_done_callback_count(), 1); | 206 EXPECT_EQ(capture_monitoring_snapshot_done_callback_count(), 1); |
202 } | 207 } |
203 | 208 |
204 { | 209 { |
205 base::RunLoop run_loop; | 210 base::RunLoop run_loop; |
206 TracingController::DisableMonitoringDoneCallback callback = | 211 TracingController::DisableMonitoringDoneCallback callback = |
207 base::Bind(&TracingControllerTest::DisableMonitoringDoneCallbackTest, | 212 base::Bind(&TracingControllerTest::DisableMonitoringDoneCallbackTest, |
208 base::Unretained(this), | 213 base::Unretained(this), |
209 run_loop.QuitClosure()); | 214 run_loop.QuitClosure()); |
210 bool result = controller->DisableMonitoring(callback); | 215 bool result = controller->DisableMonitoring(callback); |
211 ASSERT_TRUE(result); | 216 ASSERT_TRUE(result); |
212 run_loop.Run(); | 217 run_loop.Run(); |
213 EXPECT_EQ(disable_monitoring_done_callback_count(), 1); | 218 EXPECT_EQ(disable_monitoring_done_callback_count(), 1); |
214 } | 219 } |
215 | 220 |
216 { | 221 { |
217 bool is_monitoring; | 222 bool is_monitoring; |
218 std::string category_filter; | 223 CategoryFilter category_filter(""); |
219 TracingController::Options options; | 224 TraceOptions options; |
220 controller->GetMonitoringStatus(&is_monitoring, | 225 controller->GetMonitoringStatus(&is_monitoring, |
221 &category_filter, | 226 &category_filter, |
222 &options); | 227 &options); |
223 EXPECT_FALSE(is_monitoring); | 228 EXPECT_FALSE(is_monitoring); |
224 EXPECT_EQ("", category_filter); | 229 EXPECT_EQ("", category_filter.ToString()); |
225 EXPECT_FALSE(options & TracingController::ENABLE_SYSTRACE); | 230 EXPECT_FALSE(options.record_mode == RECORD_CONTINUOUSLY); |
226 EXPECT_FALSE(options & TracingController::RECORD_CONTINUOUSLY); | 231 EXPECT_FALSE(options.enable_sampling); |
227 EXPECT_FALSE(options & TracingController::ENABLE_SAMPLING); | 232 EXPECT_FALSE(options.enable_systrace); |
228 } | 233 } |
229 } | 234 } |
230 | 235 |
231 private: | 236 private: |
232 int get_categories_done_callback_count_; | 237 int get_categories_done_callback_count_; |
233 int enable_recording_done_callback_count_; | 238 int enable_recording_done_callback_count_; |
234 int disable_recording_done_callback_count_; | 239 int disable_recording_done_callback_count_; |
235 int enable_monitoring_done_callback_count_; | 240 int enable_monitoring_done_callback_count_; |
236 int disable_monitoring_done_callback_count_; | 241 int disable_monitoring_done_callback_count_; |
237 int capture_monitoring_snapshot_done_callback_count_; | 242 int capture_monitoring_snapshot_done_callback_count_; |
(...skipping 27 matching lines...) Expand all Loading... |
265 TestEnableAndDisableRecording(file_path); | 270 TestEnableAndDisableRecording(file_path); |
266 EXPECT_EQ(file_path.value(), last_actual_recording_file_path().value()); | 271 EXPECT_EQ(file_path.value(), last_actual_recording_file_path().value()); |
267 } | 272 } |
268 | 273 |
269 IN_PROC_BROWSER_TEST_F(TracingControllerTest, | 274 IN_PROC_BROWSER_TEST_F(TracingControllerTest, |
270 EnableAndDisableRecordingWithEmptyFileAndNullCallback) { | 275 EnableAndDisableRecordingWithEmptyFileAndNullCallback) { |
271 Navigate(shell()); | 276 Navigate(shell()); |
272 | 277 |
273 TracingController* controller = TracingController::GetInstance(); | 278 TracingController* controller = TracingController::GetInstance(); |
274 EXPECT_TRUE(controller->EnableRecording( | 279 EXPECT_TRUE(controller->EnableRecording( |
275 "", TracingController::DEFAULT_OPTIONS, | 280 CategoryFilter(""), |
| 281 TraceOptions(), |
276 TracingController::EnableRecordingDoneCallback())); | 282 TracingController::EnableRecordingDoneCallback())); |
277 EXPECT_TRUE(controller->DisableRecording( | 283 EXPECT_TRUE(controller->DisableRecording( |
278 base::FilePath(), TracingController::TracingFileResultCallback())); | 284 base::FilePath(), TracingController::TracingFileResultCallback())); |
279 base::RunLoop().RunUntilIdle(); | 285 base::RunLoop().RunUntilIdle(); |
280 } | 286 } |
281 | 287 |
282 IN_PROC_BROWSER_TEST_F(TracingControllerTest, | 288 IN_PROC_BROWSER_TEST_F(TracingControllerTest, |
283 EnableCaptureAndDisableMonitoring) { | 289 EnableCaptureAndDisableMonitoring) { |
284 TestEnableCaptureAndDisableMonitoring(base::FilePath()); | 290 TestEnableCaptureAndDisableMonitoring(base::FilePath()); |
285 } | 291 } |
286 | 292 |
287 IN_PROC_BROWSER_TEST_F(TracingControllerTest, | 293 IN_PROC_BROWSER_TEST_F(TracingControllerTest, |
288 EnableCaptureAndDisableMonitoringWithFilePath) { | 294 EnableCaptureAndDisableMonitoringWithFilePath) { |
289 base::FilePath file_path; | 295 base::FilePath file_path; |
290 base::CreateTemporaryFile(&file_path); | 296 base::CreateTemporaryFile(&file_path); |
291 TestEnableCaptureAndDisableMonitoring(file_path); | 297 TestEnableCaptureAndDisableMonitoring(file_path); |
292 EXPECT_EQ(file_path.value(), last_actual_monitoring_file_path().value()); | 298 EXPECT_EQ(file_path.value(), last_actual_monitoring_file_path().value()); |
293 } | 299 } |
294 | 300 |
295 IN_PROC_BROWSER_TEST_F( | 301 IN_PROC_BROWSER_TEST_F( |
296 TracingControllerTest, | 302 TracingControllerTest, |
297 EnableCaptureAndDisableMonitoringWithEmptyFileAndNullCallback) { | 303 EnableCaptureAndDisableMonitoringWithEmptyFileAndNullCallback) { |
298 Navigate(shell()); | 304 Navigate(shell()); |
299 | 305 |
300 TracingController* controller = TracingController::GetInstance(); | 306 TracingController* controller = TracingController::GetInstance(); |
301 EXPECT_TRUE(controller->EnableMonitoring( | 307 EXPECT_TRUE(controller->EnableMonitoring( |
302 "*", TracingController::ENABLE_SAMPLING, | 308 CategoryFilter("*"), |
| 309 TraceOptions(RECORD_UNTIL_FULL).EnableSampling(true), |
303 TracingController::EnableMonitoringDoneCallback())); | 310 TracingController::EnableMonitoringDoneCallback())); |
304 controller->CaptureMonitoringSnapshot( | 311 controller->CaptureMonitoringSnapshot( |
305 base::FilePath(), TracingController::TracingFileResultCallback()); | 312 base::FilePath(), TracingController::TracingFileResultCallback()); |
306 base::RunLoop().RunUntilIdle(); | 313 base::RunLoop().RunUntilIdle(); |
307 EXPECT_TRUE(controller->DisableMonitoring( | 314 EXPECT_TRUE(controller->DisableMonitoring( |
308 TracingController::DisableMonitoringDoneCallback())); | 315 TracingController::DisableMonitoringDoneCallback())); |
309 base::RunLoop().RunUntilIdle(); | 316 base::RunLoop().RunUntilIdle(); |
310 } | 317 } |
311 | 318 |
312 } // namespace content | 319 } // namespace content |
OLD | NEW |