| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | /** | 
					
						
							|  |  |  |  * Copyright 2018 Google Inc. All rights reserved. | 
					
						
							|  |  |  |  * Modifications copyright (c) Microsoft Corporation. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * 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. | 
					
						
							|  |  |  |  */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-05-06 07:08:22 -07:00
										 |  |  | import { test as it, expect } from './pageTest'; | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | it('should evaluate before anything else on the page', async ({ page, server }) => { | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   await page.addInitScript(function() { | 
					
						
							|  |  |  |     window['injected'] = 123; | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  |   }); | 
					
						
							|  |  |  |   await page.goto(server.PREFIX + '/tamperable.html'); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   expect(await page.evaluate(() => window['result'])).toBe(123); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | }); | 
					
						
							| 
									
										
										
										
											2020-08-05 11:43:40 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-05 15:51:45 -07:00
										 |  |  | it('should work with a path', async ({ page, server, asset }) => { | 
					
						
							|  |  |  |   await page.addInitScript({ path: asset('injectedfile.js') }); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  |   await page.goto(server.PREFIX + '/tamperable.html'); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   expect(await page.evaluate(() => window['result'])).toBe(123); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | }); | 
					
						
							| 
									
										
										
										
											2020-08-05 11:43:40 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-03-10 19:42:52 +01:00
										 |  |  | it('should work with content @smoke', async ({ page, server }) => { | 
					
						
							| 
									
										
										
										
											2020-08-11 15:50:53 -07:00
										 |  |  |   await page.addInitScript({ content: 'window["injected"] = 123' }); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  |   await page.goto(server.PREFIX + '/tamperable.html'); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   expect(await page.evaluate(() => window['result'])).toBe(123); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | }); | 
					
						
							| 
									
										
										
										
											2020-08-05 11:43:40 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-19 10:31:54 +08:00
										 |  |  | it('should throw without path and content', async ({ page }) => { | 
					
						
							| 
									
										
										
										
											2020-09-09 03:06:52 -07:00
										 |  |  |   // @ts-expect-error foo is not a real option of addInitScript
 | 
					
						
							|  |  |  |   const error = await page.addInitScript({ foo: 'bar' }).catch(e => e); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  |   expect(error.message).toContain('Either path or content property must be present'); | 
					
						
							|  |  |  | }); | 
					
						
							| 
									
										
										
										
											2020-08-05 11:43:40 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-04-03 17:47:12 -08:00
										 |  |  | it('should work with trailing comments', async ({ page, asset }) => { | 
					
						
							|  |  |  |   await page.addInitScript({ content: '// comment' }); | 
					
						
							|  |  |  |   await page.addInitScript({ content: 'window.secret = 42;' }); | 
					
						
							|  |  |  |   await page.goto('data:text/html,<html></html>'); | 
					
						
							|  |  |  |   expect(await page.evaluate('secret')).toBe(42); | 
					
						
							|  |  |  | }); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | it('should support multiple scripts', async ({ page, server }) => { | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   await page.addInitScript(function() { | 
					
						
							|  |  |  |     window['script1'] = 1; | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  |   }); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   await page.addInitScript(function() { | 
					
						
							|  |  |  |     window['script2'] = 2; | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  |   }); | 
					
						
							|  |  |  |   await page.goto(server.PREFIX + '/tamperable.html'); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   expect(await page.evaluate(() => window['script1'])).toBe(1); | 
					
						
							|  |  |  |   expect(await page.evaluate(() => window['script2'])).toBe(2); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | }); | 
					
						
							| 
									
										
										
										
											2020-08-05 11:43:40 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | it('should work with CSP', async ({ page, server }) => { | 
					
						
							|  |  |  |   server.setCSP('/empty.html', 'script-src ' + server.PREFIX); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   await page.addInitScript(function() { | 
					
						
							|  |  |  |     window['injected'] = 123; | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  |   }); | 
					
						
							|  |  |  |   await page.goto(server.PREFIX + '/empty.html'); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   expect(await page.evaluate(() => window['injected'])).toBe(123); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |   // Make sure CSP works.
 | 
					
						
							|  |  |  |   await page.addScriptTag({ content: 'window.e = 10;' }).catch(e => void e); | 
					
						
							| 
									
										
										
										
											2020-08-11 15:50:53 -07:00
										 |  |  |   expect(await page.evaluate(() => window['e'])).toBe(undefined); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | }); | 
					
						
							| 
									
										
										
										
											2020-08-05 11:43:40 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | it('should work after a cross origin navigation', async ({ page, server }) => { | 
					
						
							|  |  |  |   await page.goto(server.CROSS_PROCESS_PREFIX); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   await page.addInitScript(function() { | 
					
						
							|  |  |  |     window['injected'] = 123; | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  |   }); | 
					
						
							|  |  |  |   await page.goto(server.PREFIX + '/tamperable.html'); | 
					
						
							| 
									
										
										
										
											2020-08-28 04:20:29 -07:00
										 |  |  |   expect(await page.evaluate(() => window['result'])).toBe(123); | 
					
						
							| 
									
										
										
										
											2020-08-03 15:23:53 -07:00
										 |  |  | }); | 
					
						
							| 
									
										
										
										
											2023-09-13 09:46:59 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | it('init script should run only once in iframe', async ({ page, server, browserName }) => { | 
					
						
							|  |  |  |   it.info().annotations.push({ type: 'issue', description: 'https://github.com/microsoft/playwright/issues/26992' }); | 
					
						
							|  |  |  |   const messages = []; | 
					
						
							|  |  |  |   page.on('console', event => { | 
					
						
							|  |  |  |     if (event.text().startsWith('init script:')) | 
					
						
							|  |  |  |       messages.push(event.text()); | 
					
						
							|  |  |  |   }); | 
					
						
							|  |  |  |   await page.addInitScript(() => console.log('init script:', location.pathname || 'no url yet')); | 
					
						
							|  |  |  |   await page.goto(server.PREFIX + '/frames/one-frame.html'); | 
					
						
							|  |  |  |   expect(messages).toEqual([ | 
					
						
							|  |  |  |     'init script: /frames/one-frame.html', | 
					
						
							|  |  |  |     'init script: ' + (browserName === 'firefox' ? 'no url yet' : '/frames/frame.html'), | 
					
						
							|  |  |  |   ]); | 
					
						
							|  |  |  | }); | 
					
						
							| 
									
										
										
										
											2024-05-30 15:38:10 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | it('init script should run only once in popup', async ({ page, browserName }) => { | 
					
						
							|  |  |  |   await page.context().addInitScript(() => { | 
					
						
							|  |  |  |     window['callCount'] = (window['callCount'] || 0) + 1; | 
					
						
							|  |  |  |   }); | 
					
						
							|  |  |  |   const [popup] = await Promise.all([ | 
					
						
							|  |  |  |     page.waitForEvent('popup'), | 
					
						
							|  |  |  |     page.evaluate(() => window.open('about:blank')), | 
					
						
							|  |  |  |   ]); | 
					
						
							|  |  |  |   expect(await popup.evaluate('callCount')).toEqual(1); | 
					
						
							|  |  |  | }); |