2020-08-04 01:23:53 +03: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 17:08:22 +03:00
|
|
|
import { test as it, expect } from './pageTest';
|
2020-08-04 01:23:53 +03:00
|
|
|
|
|
|
|
|
it('should evaluate before anything else on the page', async ({ page, server }) => {
|
2020-08-28 14:20:29 +03:00
|
|
|
await page.addInitScript(function() {
|
|
|
|
|
window['injected'] = 123;
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
|
|
|
|
await page.goto(server.PREFIX + '/tamperable.html');
|
2020-08-28 14:20:29 +03:00
|
|
|
expect(await page.evaluate(() => window['result'])).toBe(123);
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
2020-08-05 21:43:40 +03:00
|
|
|
|
2021-04-06 01:51:45 +03:00
|
|
|
it('should work with a path', async ({ page, server, asset }) => {
|
|
|
|
|
await page.addInitScript({ path: asset('injectedfile.js') });
|
2020-08-04 01:23:53 +03:00
|
|
|
await page.goto(server.PREFIX + '/tamperable.html');
|
2020-08-28 14:20:29 +03:00
|
|
|
expect(await page.evaluate(() => window['result'])).toBe(123);
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
2020-08-05 21:43:40 +03:00
|
|
|
|
2022-03-10 21:42:52 +03:00
|
|
|
it('should work with content @smoke', async ({ page, server }) => {
|
2020-08-12 01:50:53 +03:00
|
|
|
await page.addInitScript({ content: 'window["injected"] = 123' });
|
2020-08-04 01:23:53 +03:00
|
|
|
await page.goto(server.PREFIX + '/tamperable.html');
|
2020-08-28 14:20:29 +03:00
|
|
|
expect(await page.evaluate(() => window['result'])).toBe(123);
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
2020-08-05 21:43:40 +03:00
|
|
|
|
2021-03-19 05:31:54 +03:00
|
|
|
it('should throw without path and content', async ({ page }) => {
|
2020-09-09 13:06:52 +03:00
|
|
|
// @ts-expect-error foo is not a real option of addInitScript
|
|
|
|
|
const error = await page.addInitScript({ foo: 'bar' }).catch(e => e);
|
2020-08-04 01:23:53 +03:00
|
|
|
expect(error.message).toContain('Either path or content property must be present');
|
|
|
|
|
});
|
2020-08-05 21:43:40 +03:00
|
|
|
|
2022-04-04 04:47:12 +03: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-04 01:23:53 +03:00
|
|
|
it('should support multiple scripts', async ({ page, server }) => {
|
2020-08-28 14:20:29 +03:00
|
|
|
await page.addInitScript(function() {
|
|
|
|
|
window['script1'] = 1;
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
2020-08-28 14:20:29 +03:00
|
|
|
await page.addInitScript(function() {
|
|
|
|
|
window['script2'] = 2;
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
|
|
|
|
await page.goto(server.PREFIX + '/tamperable.html');
|
2020-08-28 14:20:29 +03:00
|
|
|
expect(await page.evaluate(() => window['script1'])).toBe(1);
|
|
|
|
|
expect(await page.evaluate(() => window['script2'])).toBe(2);
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
2020-08-05 21:43:40 +03:00
|
|
|
|
2020-08-04 01:23:53 +03:00
|
|
|
it('should work with CSP', async ({ page, server }) => {
|
|
|
|
|
server.setCSP('/empty.html', 'script-src ' + server.PREFIX);
|
2020-08-28 14:20:29 +03:00
|
|
|
await page.addInitScript(function() {
|
|
|
|
|
window['injected'] = 123;
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
|
|
|
|
await page.goto(server.PREFIX + '/empty.html');
|
2020-08-28 14:20:29 +03:00
|
|
|
expect(await page.evaluate(() => window['injected'])).toBe(123);
|
2020-08-04 01:23:53 +03:00
|
|
|
|
|
|
|
|
// Make sure CSP works.
|
|
|
|
|
await page.addScriptTag({ content: 'window.e = 10;' }).catch(e => void e);
|
2020-08-12 01:50:53 +03:00
|
|
|
expect(await page.evaluate(() => window['e'])).toBe(undefined);
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
2020-08-05 21:43:40 +03:00
|
|
|
|
2020-08-04 01:23:53 +03:00
|
|
|
it('should work after a cross origin navigation', async ({ page, server }) => {
|
|
|
|
|
await page.goto(server.CROSS_PROCESS_PREFIX);
|
2020-08-28 14:20:29 +03:00
|
|
|
await page.addInitScript(function() {
|
|
|
|
|
window['injected'] = 123;
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
|
|
|
|
await page.goto(server.PREFIX + '/tamperable.html');
|
2020-08-28 14:20:29 +03:00
|
|
|
expect(await page.evaluate(() => window['result'])).toBe(123);
|
2020-08-04 01:23:53 +03:00
|
|
|
});
|
