mirror of
https://github.com/puppeteer/puppeteer
synced 2024-06-14 14:02:48 +00:00
e2db16f898
Make eval test resilient to variations in error message format between browsers.This will make the test pass without alternations in WebKit as well as Crhrome and Firefox.
307 lines
13 KiB
JavaScript
307 lines
13 KiB
JavaScript
/**
|
|
* Copyright 2018 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 utils = require('./utils');
|
|
|
|
let asyncawait = true;
|
|
try {
|
|
new Function('async function foo() {await 1}');
|
|
} catch (e) {
|
|
asyncawait = false;
|
|
}
|
|
|
|
const bigint = typeof BigInt !== 'undefined';
|
|
|
|
module.exports.addTests = function({testRunner, expect}) {
|
|
const {describe, xdescribe, fdescribe} = testRunner;
|
|
const {it, fit, xit, it_fails_ffox} = testRunner;
|
|
const {beforeAll, beforeEach, afterAll, afterEach} = testRunner;
|
|
|
|
describe('Page.evaluate', function() {
|
|
it('should work', async({page, server}) => {
|
|
const result = await page.evaluate(() => 7 * 3);
|
|
expect(result).toBe(21);
|
|
});
|
|
(bigint ? it : xit)('should transfer BigInt', async({page, server}) => {
|
|
const result = await page.evaluate(a => a, BigInt(42));
|
|
expect(result).toBe(BigInt(42));
|
|
});
|
|
it('should transfer NaN', async({page, server}) => {
|
|
const result = await page.evaluate(a => a, NaN);
|
|
expect(Object.is(result, NaN)).toBe(true);
|
|
});
|
|
it('should transfer -0', async({page, server}) => {
|
|
const result = await page.evaluate(a => a, -0);
|
|
expect(Object.is(result, -0)).toBe(true);
|
|
});
|
|
it('should transfer Infinity', async({page, server}) => {
|
|
const result = await page.evaluate(a => a, Infinity);
|
|
expect(Object.is(result, Infinity)).toBe(true);
|
|
});
|
|
it('should transfer -Infinity', async({page, server}) => {
|
|
const result = await page.evaluate(a => a, -Infinity);
|
|
expect(Object.is(result, -Infinity)).toBe(true);
|
|
});
|
|
it('should transfer arrays', async({page, server}) => {
|
|
const result = await page.evaluate(a => a, [1, 2, 3]);
|
|
expect(result).toEqual([1,2,3]);
|
|
});
|
|
it('should transfer arrays as arrays, not objects', async({page, server}) => {
|
|
const result = await page.evaluate(a => Array.isArray(a), [1, 2, 3]);
|
|
expect(result).toBe(true);
|
|
});
|
|
it('should modify global environment', async({page}) => {
|
|
await page.evaluate(() => window.globalVar = 123);
|
|
expect(await page.evaluate('globalVar')).toBe(123);
|
|
});
|
|
it('should evaluate in the page context', async({page, server}) => {
|
|
await page.goto(server.PREFIX + '/global-var.html');
|
|
expect(await page.evaluate('globalVar')).toBe(123);
|
|
});
|
|
it_fails_ffox('should return undefined for objects with symbols', async({page, server}) => {
|
|
expect(await page.evaluate(() => [Symbol('foo4')])).toBe(undefined);
|
|
});
|
|
(asyncawait ? it : xit)('should work with function shorthands', async({page, server}) => {
|
|
// trick node6 transpiler to not touch our object.
|
|
// TODO(lushnikov): remove eval once Node6 is dropped.
|
|
const a = eval(`({
|
|
sum(a, b) { return a + b; },
|
|
|
|
async mult(a, b) { return a * b; }
|
|
})`);
|
|
expect(await page.evaluate(a.sum, 1, 2)).toBe(3);
|
|
expect(await page.evaluate(a.mult, 2, 4)).toBe(8);
|
|
});
|
|
it('should work with unicode chars', async({page, server}) => {
|
|
const result = await page.evaluate(a => a['中文字符'], {'中文字符': 42});
|
|
expect(result).toBe(42);
|
|
});
|
|
it('should throw when evaluation triggers reload', async({page, server}) => {
|
|
let error = null;
|
|
await page.evaluate(() => {
|
|
location.reload();
|
|
return new Promise(() => {});
|
|
}).catch(e => error = e);
|
|
expect(error.message).toContain('Protocol error');
|
|
});
|
|
it('should await promise', async({page, server}) => {
|
|
const result = await page.evaluate(() => Promise.resolve(8 * 7));
|
|
expect(result).toBe(56);
|
|
});
|
|
it('should work right after framenavigated', async({page, server}) => {
|
|
let frameEvaluation = null;
|
|
page.on('framenavigated', async frame => {
|
|
frameEvaluation = frame.evaluate(() => 6 * 7);
|
|
});
|
|
await page.goto(server.EMPTY_PAGE);
|
|
expect(await frameEvaluation).toBe(42);
|
|
});
|
|
it('should work from-inside an exposed function', async({page, server}) => {
|
|
// Setup inpage callback, which calls Page.evaluate
|
|
await page.exposeFunction('callController', async function(a, b) {
|
|
return await page.evaluate((a, b) => a * b, a, b);
|
|
});
|
|
const result = await page.evaluate(async function() {
|
|
return await callController(9, 3);
|
|
});
|
|
expect(result).toBe(27);
|
|
});
|
|
it('should reject promise with exception', async({page, server}) => {
|
|
let error = null;
|
|
await page.evaluate(() => not_existing_object.property).catch(e => error = e);
|
|
expect(error).toBeTruthy();
|
|
expect(error.message).toContain('not_existing_object');
|
|
});
|
|
it('should support thrown strings as error messages', async({page, server}) => {
|
|
let error = null;
|
|
await page.evaluate(() => { throw 'qwerty'; }).catch(e => error = e);
|
|
expect(error).toBeTruthy();
|
|
expect(error.message).toContain('qwerty');
|
|
});
|
|
it('should support thrown numbers as error messages', async({page, server}) => {
|
|
let error = null;
|
|
await page.evaluate(() => { throw 100500; }).catch(e => error = e);
|
|
expect(error).toBeTruthy();
|
|
expect(error.message).toContain('100500');
|
|
});
|
|
it('should return complex objects', async({page, server}) => {
|
|
const object = {foo: 'bar!'};
|
|
const result = await page.evaluate(a => a, object);
|
|
expect(result).not.toBe(object);
|
|
expect(result).toEqual(object);
|
|
});
|
|
(bigint ? it : xit)('should return BigInt', async({page, server}) => {
|
|
const result = await page.evaluate(() => BigInt(42));
|
|
expect(result).toBe(BigInt(42));
|
|
});
|
|
it('should return NaN', async({page, server}) => {
|
|
const result = await page.evaluate(() => NaN);
|
|
expect(Object.is(result, NaN)).toBe(true);
|
|
});
|
|
it('should return -0', async({page, server}) => {
|
|
const result = await page.evaluate(() => -0);
|
|
expect(Object.is(result, -0)).toBe(true);
|
|
});
|
|
it('should return Infinity', async({page, server}) => {
|
|
const result = await page.evaluate(() => Infinity);
|
|
expect(Object.is(result, Infinity)).toBe(true);
|
|
});
|
|
it('should return -Infinity', async({page, server}) => {
|
|
const result = await page.evaluate(() => -Infinity);
|
|
expect(Object.is(result, -Infinity)).toBe(true);
|
|
});
|
|
it('should accept "undefined" as one of multiple parameters', async({page, server}) => {
|
|
const result = await page.evaluate((a, b) => Object.is(a, undefined) && Object.is(b, 'foo'), undefined, 'foo');
|
|
expect(result).toBe(true);
|
|
});
|
|
it('should properly serialize null fields', async({page}) => {
|
|
expect(await page.evaluate(() => ({a: undefined}))).toEqual({});
|
|
});
|
|
it('should return undefined for non-serializable objects', async({page, server}) => {
|
|
expect(await page.evaluate(() => window)).toBe(undefined);
|
|
});
|
|
it('should fail for circular object', async({page, server}) => {
|
|
const result = await page.evaluate(() => {
|
|
const a = {};
|
|
const b = {a};
|
|
a.b = b;
|
|
return a;
|
|
});
|
|
expect(result).toBe(undefined);
|
|
});
|
|
it_fails_ffox('should be able to throw a tricky error', async({page, server}) => {
|
|
const windowHandle = await page.evaluateHandle(() => window);
|
|
const errorText = await windowHandle.jsonValue().catch(e => e.message);
|
|
const error = await page.evaluate(errorText => {
|
|
throw new Error(errorText);
|
|
}, errorText).catch(e => e);
|
|
expect(error.message).toContain(errorText);
|
|
});
|
|
it('should accept a string', async({page, server}) => {
|
|
const result = await page.evaluate('1 + 2');
|
|
expect(result).toBe(3);
|
|
});
|
|
it('should accept a string with semi colons', async({page, server}) => {
|
|
const result = await page.evaluate('1 + 5;');
|
|
expect(result).toBe(6);
|
|
});
|
|
it('should accept a string with comments', async({page, server}) => {
|
|
const result = await page.evaluate('2 + 5;\n// do some math!');
|
|
expect(result).toBe(7);
|
|
});
|
|
it('should accept element handle as an argument', async({page, server}) => {
|
|
await page.setContent('<section>42</section>');
|
|
const element = await page.$('section');
|
|
const text = await page.evaluate(e => e.textContent, element);
|
|
expect(text).toBe('42');
|
|
});
|
|
it('should throw if underlying element was disposed', async({page, server}) => {
|
|
await page.setContent('<section>39</section>');
|
|
const element = await page.$('section');
|
|
expect(element).toBeTruthy();
|
|
await element.dispose();
|
|
let error = null;
|
|
await page.evaluate(e => e.textContent, element).catch(e => error = e);
|
|
expect(error.message).toContain('JSHandle is disposed');
|
|
});
|
|
it('should throw if elementHandles are from other frames', async({page, server}) => {
|
|
await utils.attachFrame(page, 'frame1', server.EMPTY_PAGE);
|
|
const bodyHandle = await page.frames()[1].$('body');
|
|
let error = null;
|
|
await page.evaluate(body => body.innerHTML, bodyHandle).catch(e => error = e);
|
|
expect(error).toBeTruthy();
|
|
expect(error.message).toContain('JSHandles can be evaluated only in the context they were created');
|
|
});
|
|
it('should simulate a user gesture', async({page, server}) => {
|
|
const result = await page.evaluate(() => {
|
|
document.body.appendChild(document.createTextNode('test'));
|
|
document.execCommand('selectAll');
|
|
return document.execCommand('copy');
|
|
});
|
|
expect(result).toBe(true);
|
|
});
|
|
it('should throw a nice error after a navigation', async({page, server}) => {
|
|
const executionContext = await page.mainFrame().executionContext();
|
|
|
|
await Promise.all([
|
|
page.waitForNavigation(),
|
|
executionContext.evaluate(() => window.location.reload())
|
|
]);
|
|
const error = await executionContext.evaluate(() => null).catch(e => e);
|
|
expect(error.message).toContain('navigation');
|
|
});
|
|
it_fails_ffox('should not throw an error when evaluation does a navigation', async({page, server}) => {
|
|
await page.goto(server.PREFIX + '/one-style.html');
|
|
const result = await page.evaluate(() => {
|
|
window.location = '/empty.html';
|
|
return [42];
|
|
});
|
|
expect(result).toEqual([42]);
|
|
});
|
|
it_fails_ffox('should transfer 100Mb of data from page to node.js', async({page, server}) => {
|
|
const a = await page.evaluate(() => Array(100 * 1024 * 1024 + 1).join('a'));
|
|
expect(a.length).toBe(100 * 1024 * 1024);
|
|
});
|
|
});
|
|
|
|
describe('Page.evaluateOnNewDocument', function() {
|
|
it('should evaluate before anything else on the page', async({page, server}) => {
|
|
await page.evaluateOnNewDocument(function(){
|
|
window.injected = 123;
|
|
});
|
|
await page.goto(server.PREFIX + '/tamperable.html');
|
|
expect(await page.evaluate(() => window.result)).toBe(123);
|
|
});
|
|
it('should work with CSP', async({page, server}) => {
|
|
server.setCSP('/empty.html', 'script-src ' + server.PREFIX);
|
|
await page.evaluateOnNewDocument(function(){
|
|
window.injected = 123;
|
|
});
|
|
await page.goto(server.PREFIX + '/empty.html');
|
|
expect(await page.evaluate(() => window.injected)).toBe(123);
|
|
|
|
// Make sure CSP works.
|
|
await page.addScriptTag({content: 'window.e = 10;'}).catch(e => void e);
|
|
expect(await page.evaluate(() => window.e)).toBe(undefined);
|
|
});
|
|
});
|
|
|
|
describe('Frame.evaluate', function() {
|
|
it('should have different execution contexts', async({page, server}) => {
|
|
await page.goto(server.EMPTY_PAGE);
|
|
await utils.attachFrame(page, 'frame1', server.EMPTY_PAGE);
|
|
expect(page.frames().length).toBe(2);
|
|
await page.frames()[0].evaluate(() => window.FOO = 'foo');
|
|
await page.frames()[1].evaluate(() => window.FOO = 'bar');
|
|
expect(await page.frames()[0].evaluate(() => window.FOO)).toBe('foo');
|
|
expect(await page.frames()[1].evaluate(() => window.FOO)).toBe('bar');
|
|
});
|
|
it('should have correct execution contexts', async({page, server}) => {
|
|
await page.goto(server.PREFIX + '/frames/one-frame.html');
|
|
expect(page.frames().length).toBe(2);
|
|
expect(await page.frames()[0].evaluate(() => document.body.textContent.trim())).toBe('');
|
|
expect(await page.frames()[1].evaluate(() => document.body.textContent.trim())).toBe(`Hi, I'm frame`);
|
|
});
|
|
it('should execute after cross-site navigation', async({page, server}) => {
|
|
await page.goto(server.EMPTY_PAGE);
|
|
const mainFrame = page.mainFrame();
|
|
expect(await mainFrame.evaluate(() => window.location.href)).toContain('localhost');
|
|
await page.goto(server.CROSS_PROCESS_PREFIX + '/empty.html');
|
|
expect(await mainFrame.evaluate(() => window.location.href)).toContain('127');
|
|
});
|
|
});
|
|
};
|