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 import unittest | 5 import unittest |
6 | 6 |
7 from telemetry.core.platform import tracing_category_filter | |
8 from telemetry.core.platform import tracing_options | |
9 from telemetry.core import util | 7 from telemetry.core import util |
10 from telemetry.internal.backends.chrome_inspector import tracing_backend | 8 from telemetry.internal.backends.chrome_inspector import tracing_backend |
11 from telemetry.internal.backends.chrome_inspector import websocket | 9 from telemetry.internal.backends.chrome_inspector import websocket |
12 from telemetry.testing import simple_mock | 10 from telemetry.testing import simple_mock |
13 from telemetry.testing import tab_test_case | 11 from telemetry.testing import tab_test_case |
14 from telemetry.timeline import model as model_module | 12 from telemetry.timeline import model as model_module |
| 13 from telemetry.timeline import tracing_category_filter |
| 14 from telemetry.timeline import tracing_options |
15 | 15 |
16 util.AddDirToPythonPath(util.GetTelemetryDir(), 'third_party', 'mock') | 16 util.AddDirToPythonPath(util.GetTelemetryDir(), 'third_party', 'mock') |
17 import mock # pylint:disable=import-error | 17 import mock # pylint:disable=import-error |
18 | 18 |
19 | 19 |
20 class FakeInspectorWebsocket(object): | 20 class FakeInspectorWebsocket(object): |
21 """A fake InspectorWebsocket. | 21 """A fake InspectorWebsocket. |
22 | 22 |
23 A fake that allows tests to send pregenerated data. Normal | 23 A fake that allows tests to send pregenerated data. Normal |
24 InspectorWebsockets allow for any number of domain handlers. This fake only | 24 InspectorWebsockets allow for any number of domain handlers. This fake only |
(...skipping 105 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
130 inspector.AddResponse('Tracing.tracingComplete', 'asdf3', 19) | 130 inspector.AddResponse('Tracing.tracingComplete', 'asdf3', 19) |
131 | 131 |
132 with mock.patch('telemetry.internal.backends.chrome_inspector.' | 132 with mock.patch('telemetry.internal.backends.chrome_inspector.' |
133 'inspector_websocket.InspectorWebsocket') as mock_class: | 133 'inspector_websocket.InspectorWebsocket') as mock_class: |
134 mock_class.return_value = inspector | 134 mock_class.return_value = inspector |
135 backend = tracing_backend.TracingBackend(devtools_port=65000) | 135 backend = tracing_backend.TracingBackend(devtools_port=65000) |
136 | 136 |
137 backend._CollectTracingData(10) | 137 backend._CollectTracingData(10) |
138 self.assertEqual(2, len(backend._trace_events)) | 138 self.assertEqual(2, len(backend._trace_events)) |
139 self.assertTrue(backend._has_received_all_tracing_data) | 139 self.assertTrue(backend._has_received_all_tracing_data) |
OLD | NEW |