| 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 |