| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 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 """Common utilities used in unit tests, within this directory.""" | 5 """Common utilities used in unit tests, within this directory.""" |
| 6 | 6 |
| 7 import clovis_constants | 7 import clovis_constants |
| 8 import dependency_graph | 8 import dependency_graph |
| 9 import devtools_monitor | 9 import devtools_monitor |
| 10 import loading_trace | 10 import loading_trace |
| 11 import page_track | 11 import page_track |
| 12 import request_track | 12 import request_track |
| 13 import tracing | 13 import tracing_track |
| 14 import user_satisfied_lens | 14 import user_satisfied_lens |
| 15 | 15 |
| 16 | 16 |
| 17 class FakeRequestTrack(devtools_monitor.Track): | 17 class FakeRequestTrack(devtools_monitor.Track): |
| 18 def __init__(self, events): | 18 def __init__(self, events): |
| 19 super(FakeRequestTrack, self).__init__(None) | 19 super(FakeRequestTrack, self).__init__(None) |
| 20 self._events = events | 20 self._events = events |
| 21 for e in self._events: | 21 for e in self._events: |
| 22 e.timing.request_time = e.timestamp | 22 e.timing.request_time = e.timestamp |
| 23 | 23 |
| (...skipping 110 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 134 'requestTime': start_time / 1000.0} | 134 'requestTime': start_time / 1000.0} |
| 135 return MakeRequestWithTiming( | 135 return MakeRequestWithTiming( |
| 136 url, source_url, timing_dict, magic_content_type, initiator_type) | 136 url, source_url, timing_dict, magic_content_type, initiator_type) |
| 137 | 137 |
| 138 | 138 |
| 139 def LoadingTraceFromEvents(requests, page_events=None, trace_events=None): | 139 def LoadingTraceFromEvents(requests, page_events=None, trace_events=None): |
| 140 """Returns a LoadingTrace instance from various events.""" | 140 """Returns a LoadingTrace instance from various events.""" |
| 141 request = FakeRequestTrack(requests) | 141 request = FakeRequestTrack(requests) |
| 142 page_event_track = FakePageTrack(page_events if page_events else []) | 142 page_event_track = FakePageTrack(page_events if page_events else []) |
| 143 if trace_events is not None: | 143 if trace_events is not None: |
| 144 tracing_track = tracing.TracingTrack(None, | 144 track = tracing_track.TracingTrack(None, |
| 145 clovis_constants.DEFAULT_CATEGORIES) | 145 clovis_constants.DEFAULT_CATEGORIES) |
| 146 tracing_track.Handle('Tracing.dataCollected', | 146 track.Handle('Tracing.dataCollected', |
| 147 {'params': {'value': [e for e in trace_events]}}) | 147 {'params': {'value': [e for e in trace_events]}}) |
| 148 else: | 148 else: |
| 149 tracing_track = None | 149 track = None |
| 150 return loading_trace.LoadingTrace( | 150 return loading_trace.LoadingTrace( |
| 151 None, None, page_event_track, request, tracing_track) | 151 None, None, page_event_track, request, track) |
| 152 | 152 |
| 153 | 153 |
| 154 class SimpleLens(object): | 154 class SimpleLens(object): |
| 155 """A simple replacement for RequestDependencyLens. | 155 """A simple replacement for RequestDependencyLens. |
| 156 | 156 |
| 157 Uses only the initiator url of a request for determining a dependency. | 157 Uses only the initiator url of a request for determining a dependency. |
| 158 """ | 158 """ |
| 159 def __init__(self, trace): | 159 def __init__(self, trace): |
| 160 self._trace = trace | 160 self._trace = trace |
| 161 | 161 |
| (...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 248 self._request_index += 1 | 248 self._request_index += 1 |
| 249 return rq | 249 return rq |
| 250 | 250 |
| 251 def CreateTrace(self, requests, events, main_frame_id): | 251 def CreateTrace(self, requests, events, main_frame_id): |
| 252 page_event = {'method': 'Page.frameStartedLoading', | 252 page_event = {'method': 'Page.frameStartedLoading', |
| 253 'frame_id': main_frame_id} | 253 'frame_id': main_frame_id} |
| 254 trace = LoadingTraceFromEvents( | 254 trace = LoadingTraceFromEvents( |
| 255 requests, trace_events=events, page_events=[page_event]) | 255 requests, trace_events=events, page_events=[page_event]) |
| 256 trace.tracing_track.SetMainFrameID(main_frame_id) | 256 trace.tracing_track.SetMainFrameID(main_frame_id) |
| 257 return trace | 257 return trace |
| OLD | NEW |