@@ -18,7 +18,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
1818 it ( 'returns the correct events for tree views given a flame chart group' , async function ( ) {
1919 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
2020 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'sync-like-timings.json.gz' ) ;
21- dataProvider . setModel ( parsedTrace ) ;
21+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
22+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
2223 const timingsTrackGroup = dataProvider . timelineData ( ) . groups . find ( g => g . name === 'Timings' ) ;
2324 if ( ! timingsTrackGroup ) {
2425 assert . fail ( 'Could not find Timings track flame chart group' ) ;
@@ -36,7 +37,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
3637 it ( 'filters out async events if they cannot be added to the tree' , async function ( ) {
3738 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
3839 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'timings-track.json.gz' ) ;
39- dataProvider . setModel ( parsedTrace ) ;
40+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
41+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
4042 const timingsTrackGroup = dataProvider . timelineData ( ) . groups . find ( g => g . name === 'Timings' ) ;
4143 if ( ! timingsTrackGroup ) {
4244 assert . fail ( 'Could not find Timings track flame chart group' ) ;
@@ -52,7 +54,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
5254 setupIgnoreListManagerEnvironment ( ) ;
5355 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
5456 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'one-second-interaction.json.gz' ) ;
55- dataProvider . setModel ( parsedTrace ) ;
57+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
58+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
5659
5760 // Need to use an index that is not a frame, so jump past the frames.
5861 const event = dataProvider . eventByIndex ( 100 ) ;
@@ -63,7 +66,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
6366 setupIgnoreListManagerEnvironment ( ) ;
6467 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
6568 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'extension-tracks-and-marks.json.gz' ) ;
66- dataProvider . setModel ( parsedTrace ) ;
69+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
70+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
6771 const groupNames = dataProvider . timelineData ( ) . groups . map ( g => g . name ) ;
6872 assert . deepEqual (
6973 groupNames ,
@@ -89,7 +93,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
8993 setupIgnoreListManagerEnvironment ( ) ;
9094 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
9195 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'one-second-interaction.json.gz' ) ;
92- dataProvider . setModel ( parsedTrace ) ;
96+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
97+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
9398 // Force the track appenders to run and populate the chart data.
9499 dataProvider . timelineData ( ) ;
95100
@@ -105,7 +110,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
105110 setupIgnoreListManagerEnvironment ( ) ;
106111 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
107112 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'one-second-interaction.json.gz' ) ;
108- dataProvider . setModel ( parsedTrace ) ;
113+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
114+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
109115 dataProvider . timelineData ( ) ;
110116
111117 const { entryDecorations} = dataProvider . timelineData ( ) ;
@@ -137,7 +143,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
137143 it ( 'populates the frames track with frames and screenshots' , async function ( ) {
138144 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
139145 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'web-dev.json.gz' ) ;
140- dataProvider . setModel ( parsedTrace ) ;
146+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
147+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
141148 const framesTrack = dataProvider . timelineData ( ) . groups . find ( g => {
142149 return g . name . includes ( 'Frames' ) ;
143150 } ) ;
@@ -164,14 +171,15 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
164171
165172 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
166173 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'react-hello-world.json.gz' ) ;
167- dataProvider . setModel ( parsedTrace ) ;
174+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
175+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
168176
169177 const eventCountBeforeIgnoreList = dataProvider . timelineData ( ) . entryStartTimes . length ;
170178
171179 const SCRIPT_TO_IGNORE = urlString `https://unpkg.com/[email protected] /umd/react.development.js` ; 172180 // Clear the data provider cache and add the React script to the ignore list.
173181 dataProvider . reset ( ) ;
174- dataProvider . setModel ( parsedTrace ) ;
182+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
175183 ignoreListManager . ignoreListURL ( SCRIPT_TO_IGNORE ) ;
176184
177185 const eventCountAfterIgnoreList = dataProvider . timelineData ( ) . entryStartTimes . length ;
@@ -181,7 +189,7 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
181189
182190 // Clear the data provider cache and unignore the script again
183191 dataProvider . reset ( ) ;
184- dataProvider . setModel ( parsedTrace ) ;
192+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
185193 ignoreListManager . unIgnoreListURL ( SCRIPT_TO_IGNORE ) ;
186194 // Ensure that now we have un-ignored the URL that we get the full set of events again.
187195 assert . strictEqual ( dataProvider . timelineData ( ) . entryStartTimes . length , eventCountBeforeIgnoreList ) ;
@@ -191,8 +199,9 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
191199 it ( 'filters navigations to only return those that happen on the main frame' , async function ( ) {
192200 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
193201 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'multiple-navigations-with-iframes.json.gz' ) ;
202+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
194203
195- dataProvider . setModel ( parsedTrace ) ;
204+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
196205
197206 const mainFrameID = parsedTrace . Meta . mainFrameId ;
198207 const navigationEvents = dataProvider . mainFrameNavigationStartEvents ( ) ;
@@ -205,7 +214,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
205214 it ( 'can search for entries within a given time-range' , async function ( ) {
206215 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
207216 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'web-dev-with-commit.json.gz' ) ;
208- dataProvider . setModel ( parsedTrace ) ;
217+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
218+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
209219
210220 const bounds = parsedTrace . Meta . traceBounds ;
211221 const filter = new Timeline . TimelineFilters . TimelineRegExp ( / E v a l u a t e s c r i p t / ) ;
@@ -217,7 +227,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
217227 it ( 'delete annotations associated with an event' , async function ( ) {
218228 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
219229 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'web-dev-with-commit.json.gz' ) ;
220- dataProvider . setModel ( parsedTrace ) ;
230+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
231+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
221232 const entryIndex = 0 ;
222233 const eventToFindAssociatedEntriesFor = dataProvider . eventByIndex ( entryIndex ) ;
223234 const event = dataProvider . eventByIndex ( 1 ) ;
@@ -245,7 +256,8 @@ describeWithEnvironment('TimelineFlameChartDataProvider', function() {
245256 it ( 'correctly identifies if an event has annotations' , async function ( ) {
246257 const dataProvider = new Timeline . TimelineFlameChartDataProvider . TimelineFlameChartDataProvider ( ) ;
247258 const { parsedTrace} = await TraceLoader . traceEngine ( this , 'web-dev-with-commit.json.gz' ) ;
248- dataProvider . setModel ( parsedTrace ) ;
259+ const entityMapper = new Timeline . Utils . EntityMapper . EntityMapper ( parsedTrace ) ;
260+ dataProvider . setModel ( parsedTrace , entityMapper ) ;
249261 const eventIndex = 0 ;
250262 const event = dataProvider . eventByIndex ( eventIndex ) ;
251263 assert . exists ( event ) ;
0 commit comments