| 
									
										
										
										
											2019-11-18 18:18:28 -08:00
										 |  |  | /** | 
					
						
							|  |  |  |  * Copyright 2017 Google Inc. All rights reserved. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * Licensed under the Apache License, Version 2.0 (the "License"); | 
					
						
							|  |  |  |  * you may not use this file except in compliance with the License. | 
					
						
							|  |  |  |  * You may obtain a copy of the License at | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  *     http://www.apache.org/licenses/LICENSE-2.0
 | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * Unless required by applicable law or agreed to in writing, software | 
					
						
							|  |  |  |  * distributed under the License is distributed on an "AS IS" BASIS, | 
					
						
							|  |  |  |  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
					
						
							|  |  |  |  * See the License for the specific language governing permissions and | 
					
						
							|  |  |  |  * limitations under the License. | 
					
						
							|  |  |  |  */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | const fs = require('fs'); | 
					
						
							|  |  |  | const path = require('path'); | 
					
						
							| 
									
										
										
										
											2020-07-17 13:02:50 -07:00
										 |  |  | const {FFOX, CHROMIUM, WEBKIT, OUTPUT_DIR, CHANNEL} = testOptions; | 
					
						
							| 
									
										
										
										
											2019-11-18 18:18:28 -08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-07-17 13:02:50 -07:00
										 |  |  | registerFixture('outputFile', async ({parallelIndex}, test) => { | 
					
						
							|  |  |  |   const outputFile = path.join(OUTPUT_DIR, `trace-${parallelIndex}.json`); | 
					
						
							|  |  |  |   await test(outputFile); | 
					
						
							|  |  |  |   if (fs.existsSync(outputFile)) | 
					
						
							|  |  |  |     fs.unlinkSync(outputFile); | 
					
						
							|  |  |  | }); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | describe.skip(!CHROMIUM)('Chromium.startTracing', function() { | 
					
						
							| 
									
										
										
										
											2020-04-08 15:19:09 -07:00
										 |  |  |   it('should output a trace', async({browser, page, server, outputFile}) => { | 
					
						
							|  |  |  |     await browser.startTracing(page, {screenshots: true, path: outputFile}); | 
					
						
							|  |  |  |     await page.goto(server.PREFIX + '/grid.html'); | 
					
						
							|  |  |  |     await browser.stopTracing(); | 
					
						
							|  |  |  |     expect(fs.existsSync(outputFile)).toBe(true); | 
					
						
							|  |  |  |   }); | 
					
						
							|  |  |  |   it('should run with custom categories if provided', async({browser, page, outputFile}) => { | 
					
						
							|  |  |  |     await browser.startTracing(page, {path: outputFile, categories: ['disabled-by-default-v8.cpu_profiler.hires']}); | 
					
						
							|  |  |  |     await browser.stopTracing(); | 
					
						
							| 
									
										
										
										
											2019-11-18 18:18:28 -08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-04-08 15:19:09 -07:00
										 |  |  |     const traceJson = JSON.parse(fs.readFileSync(outputFile).toString()); | 
					
						
							|  |  |  |     expect(traceJson.metadata['trace-config']).toContain('disabled-by-default-v8.cpu_profiler.hires', 'Does not contain expected category'); | 
					
						
							|  |  |  |   }); | 
					
						
							| 
									
										
										
										
											2020-07-17 13:02:50 -07:00
										 |  |  |   it('should throw if tracing on two pages', async({browser, page, outputFile}) => { | 
					
						
							| 
									
										
										
										
											2020-04-08 15:19:09 -07:00
										 |  |  |     await browser.startTracing(page, {path: outputFile}); | 
					
						
							|  |  |  |     const newPage = await browser.newPage(); | 
					
						
							|  |  |  |     let error = null; | 
					
						
							|  |  |  |     await browser.startTracing(newPage, {path: outputFile}).catch(e => error = e); | 
					
						
							|  |  |  |     await newPage.close(); | 
					
						
							|  |  |  |     expect(error).toBeTruthy(); | 
					
						
							|  |  |  |     await browser.stopTracing(); | 
					
						
							|  |  |  |   }); | 
					
						
							|  |  |  |   it('should return a buffer', async({browser, page, server, outputFile}) => { | 
					
						
							|  |  |  |     await browser.startTracing(page, {screenshots: true, path: outputFile}); | 
					
						
							|  |  |  |     await page.goto(server.PREFIX + '/grid.html'); | 
					
						
							|  |  |  |     const trace = await browser.stopTracing(); | 
					
						
							|  |  |  |     const buf = fs.readFileSync(outputFile); | 
					
						
							|  |  |  |     expect(trace.toString()).toEqual(buf.toString(), 'Tracing buffer mismatch'); | 
					
						
							|  |  |  |   }); | 
					
						
							| 
									
										
										
										
											2020-07-17 13:02:50 -07:00
										 |  |  |   it('should work without options', async({browser, page, server}) => { | 
					
						
							| 
									
										
										
										
											2020-04-08 15:19:09 -07:00
										 |  |  |     await browser.startTracing(page); | 
					
						
							|  |  |  |     await page.goto(server.PREFIX + '/grid.html'); | 
					
						
							|  |  |  |     const trace = await browser.stopTracing(); | 
					
						
							|  |  |  |     expect(trace).toBeTruthy(); | 
					
						
							|  |  |  |   }); | 
					
						
							|  |  |  |   it('should support a buffer without a path', async({browser, page, server}) => { | 
					
						
							|  |  |  |     await browser.startTracing(page, {screenshots: true}); | 
					
						
							|  |  |  |     await page.goto(server.PREFIX + '/grid.html'); | 
					
						
							|  |  |  |     const trace = await browser.stopTracing(); | 
					
						
							|  |  |  |     expect(trace.toString()).toContain('screenshot', 'Does not contain screenshot'); | 
					
						
							| 
									
										
										
										
											2019-11-18 18:18:28 -08:00
										 |  |  |   }); | 
					
						
							| 
									
										
										
										
											2020-04-08 15:19:09 -07:00
										 |  |  | }); |