| # Copyright 2014 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import unittest |
| |
| from telemetry.timeline import model as timeline_model |
| from telemetry.timeline import tab_id_importer |
| from telemetry.timeline import trace_data as trace_data_module |
| |
| class TabIdImporterUnitTest(unittest.TestCase): |
| def testImportOverflowedTrace(self): |
| builder = trace_data_module.TraceDataBuilder() |
| builder.AddEventsTo(trace_data_module.CHROME_TRACE_PART, [ |
| {'name': 'a', 'args': {}, 'pid': 1, 'ts': 7, 'cat': 'foo', |
| 'tid': 1, 'ph': 'B'}, |
| {'name': 'a', 'args': {}, 'pid': 1, 'ts': 8, 'cat': 'foo', |
| 'tid': 1, 'ph': 'E'}, |
| {'name': 'b', 'args': {}, 'pid': 2, 'ts': 9, 'cat': 'foo', |
| 'tid': 2, 'ph': 'B'}, |
| {'name': 'b', 'args': {}, 'pid': 2, 'ts': 10, 'cat': 'foo', |
| 'tid': 2, 'ph': 'E'}, |
| {'name': 'trace_buffer_overflowed', |
| 'args': {'overflowed_at_ts': 12}, |
| 'pid': 2, 'ts': 0, 'tid': 2, 'ph': 'M'} |
| ]) |
| builder.AddEventsTo( |
| trace_data_module.TAB_ID_PART, ['tab-id-1', 'tab-id-2']) |
| |
| with self.assertRaises(tab_id_importer.TraceBufferOverflowException) \ |
| as context: |
| timeline_model.TimelineModel(builder.AsData()) |
| self.assertTrue( |
| 'Trace buffer of process with pid=2 overflowed at timestamp 12' in |
| context.exception.message) |
| |
| def testTraceEventsWithTabIdsMarkers(self): |
| builder = trace_data_module.TraceDataBuilder() |
| builder.AddEventsTo(trace_data_module.CHROME_TRACE_PART, [ |
| {'name': 'a', 'args': {}, 'pid': 1, 'ts': 20, 'tts': 10, 'cat': 'foo', |
| 'tid': 1, 'ph': 'B'}, |
| # tab-id-1 |
| {'name': 'tab-id-1', 'args': {}, 'pid': 1, 'ts': 25, 'cat': 'foo', |
| 'tid': 1, |
| 'ph': 'S', 'id': 72}, |
| {'name': 'a', 'args': {}, 'pid': 1, 'ts': 30, 'tts': 20, 'cat': 'foo', |
| 'tid': 1, 'ph': 'E'}, |
| {'name': 'tab-id-1', 'args': {}, 'pid': 1, 'ts': 35, 'cat': 'foo', |
| 'tid': 1, |
| 'ph': 'F', 'id': 72}, |
| # tab-id-2 |
| {'name': 'tab-id-2', 'args': {}, 'pid': 1, 'ts': 25, 'cat': 'foo', |
| 'tid': 2, |
| 'ph': 'S', 'id': 72}, |
| {'name': 'tab-id-2', 'args': {}, 'pid': 1, 'ts': 26, 'cat': 'foo', |
| 'tid': 2, |
| 'ph': 'F', 'id': 72}, |
| ]) |
| builder.AddEventsTo( |
| trace_data_module.TAB_ID_PART, ['tab-id-1', 'tab-id-2']) |
| |
| m = timeline_model.TimelineModel(builder.AsData()) |
| processes = m.GetAllProcesses() |
| self.assertEqual(1, len(processes)) |
| self.assertIs(processes[0], m.GetRendererProcessFromTabId('tab-id-1')) |
| self.assertIs(processes[0], m.GetRendererProcessFromTabId('tab-id-2')) |
| |
| p = processes[0] |
| self.assertEqual(2, len(p.threads)) |
| self.assertIs(p.threads[1], m.GetRendererThreadFromTabId('tab-id-1')) |
| self.assertIs(p.threads[2], m.GetRendererThreadFromTabId('tab-id-2')) |