puppeteer/test/evaluation.spec.js

467 lines
15 KiB
JavaScript
Raw Normal View History

/**
* 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');
const expect = require('expect');
2020-05-07 10:54:55 +00:00
const {
getTestState,
setupTestBrowserHooks,
setupTestPageAndContextHooks,
} = require('./mocha-utils');
const bigint = typeof BigInt !== 'undefined';
2020-05-07 10:54:55 +00:00
describe('Evaluation specs', function () {
setupTestBrowserHooks();
setupTestPageAndContextHooks();
2020-05-07 10:54:55 +00:00
describe('Page.evaluate', function () {
it('should work', async () => {
const { page } = getTestState();
const result = await page.evaluate(() => 7 * 3);
expect(result).toBe(21);
});
(bigint ? it : xit)('should transfer BigInt', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate((a) => a, BigInt(42));
expect(result).toBe(BigInt(42));
});
it('should transfer NaN', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate((a) => a, NaN);
expect(Object.is(result, NaN)).toBe(true);
});
it('should transfer -0', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate((a) => a, -0);
expect(Object.is(result, -0)).toBe(true);
});
it('should transfer Infinity', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate((a) => a, Infinity);
expect(Object.is(result, Infinity)).toBe(true);
});
it('should transfer -Infinity', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate((a) => a, -Infinity);
expect(Object.is(result, -Infinity)).toBe(true);
});
2020-05-07 10:54:55 +00:00
it('should transfer arrays', async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate((a) => a, [1, 2, 3]);
expect(result).toEqual([1, 2, 3]);
});
2020-05-07 10:54:55 +00:00
it('should transfer arrays as arrays, not objects', async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate((a) => Array.isArray(a), [1, 2, 3]);
expect(result).toBe(true);
});
2020-05-07 10:54:55 +00:00
it('should modify global environment', async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
await page.evaluate(() => (window.globalVar = 123));
expect(await page.evaluate('globalVar')).toBe(123);
});
2020-05-07 10:54:55 +00:00
it('should evaluate in the page context', async () => {
const { page, server } = getTestState();
await page.goto(server.PREFIX + '/global-var.html');
expect(await page.evaluate('globalVar')).toBe(123);
});
2020-05-07 10:54:55 +00:00
itFailsFirefox(
'should return undefined for objects with symbols',
async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
expect(await page.evaluate(() => [Symbol('foo4')])).toBe(undefined);
}
);
it('should work with function shorthands', async () => {
const { page } = getTestState();
const a = {
2020-05-07 10:54:55 +00:00
sum(a, b) {
return a + b;
},
2020-05-07 10:54:55 +00:00
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);
});
2020-05-07 10:54:55 +00:00
it('should work with unicode chars', async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate((a) => a['中文字符'], {
中文字符: 42,
});
expect(result).toBe(42);
});
2020-05-07 10:54:55 +00:00
itFailsFirefox('should throw when evaluation triggers reload', async () => {
const { page } = getTestState();
let error = null;
2020-05-07 10:54:55 +00:00
await page
.evaluate(() => {
location.reload();
return new Promise(() => {});
})
.catch((error_) => (error = error_));
expect(error.message).toContain('Protocol error');
});
2020-05-07 10:54:55 +00:00
it('should await promise', async () => {
const { page } = getTestState();
const result = await page.evaluate(() => Promise.resolve(8 * 7));
expect(result).toBe(56);
});
2020-05-07 10:54:55 +00:00
it('should work right after framenavigated', async () => {
const { page, server } = getTestState();
let frameEvaluation = null;
2020-05-07 10:54:55 +00:00
page.on('framenavigated', async (frame) => {
frameEvaluation = frame.evaluate(() => 6 * 7);
});
await page.goto(server.EMPTY_PAGE);
expect(await frameEvaluation).toBe(42);
});
2020-05-07 10:54:55 +00:00
itFailsFirefox('should work from-inside an exposed function', async () => {
const { page } = getTestState();
// Setup inpage callback, which calls Page.evaluate
2020-05-07 10:54:55 +00:00
await page.exposeFunction('callController', async function (a, b) {
return await page.evaluate((a, b) => a * b, a, b);
});
2020-05-07 10:54:55 +00:00
const result = await page.evaluate(async function () {
return await callController(9, 3);
});
expect(result).toBe(27);
});
2020-05-07 10:54:55 +00:00
it('should reject promise with exception', async () => {
const { page } = getTestState();
let error = null;
2020-05-07 10:54:55 +00:00
await page
.evaluate(() => not_existing_object.property)
.catch((error_) => (error = error_));
expect(error).toBeTruthy();
expect(error.message).toContain('not_existing_object');
});
2020-05-07 10:54:55 +00:00
it('should support thrown strings as error messages', async () => {
const { page } = getTestState();
let error = null;
2020-05-07 10:54:55 +00:00
await page
.evaluate(() => {
throw 'qwerty';
})
.catch((error_) => (error = error_));
expect(error).toBeTruthy();
expect(error.message).toContain('qwerty');
});
2020-05-07 10:54:55 +00:00
it('should support thrown numbers as error messages', async () => {
const { page } = getTestState();
let error = null;
2020-05-07 10:54:55 +00:00
await page
.evaluate(() => {
throw 100500;
})
.catch((error_) => (error = error_));
expect(error).toBeTruthy();
expect(error.message).toContain('100500');
});
2020-05-07 10:54:55 +00:00
it('should return complex objects', async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
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 () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
const result = await page.evaluate(() => BigInt(42));
expect(result).toBe(BigInt(42));
});
it('should return NaN', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
const result = await page.evaluate(() => NaN);
expect(Object.is(result, NaN)).toBe(true);
});
it('should return -0', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
const result = await page.evaluate(() => -0);
expect(Object.is(result, -0)).toBe(true);
});
it('should return Infinity', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
const result = await page.evaluate(() => Infinity);
expect(Object.is(result, Infinity)).toBe(true);
});
it('should return -Infinity', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
const result = await page.evaluate(() => -Infinity);
expect(Object.is(result, -Infinity)).toBe(true);
});
2020-05-07 10:54:55 +00:00
it('should accept "undefined" as one of multiple parameters', async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const result = await page.evaluate(
(a, b) => Object.is(a, undefined) && Object.is(b, 'foo'),
undefined,
'foo'
);
expect(result).toBe(true);
});
2020-05-07 10:54:55 +00:00
it('should properly serialize null fields', async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
expect(await page.evaluate(() => ({ a: undefined }))).toEqual({});
});
2020-05-07 10:54:55 +00:00
itFailsFirefox(
'should return undefined for non-serializable objects',
async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
expect(await page.evaluate(() => window)).toBe(undefined);
}
);
itFailsFirefox('should fail for circular object', async () => {
const { page } = getTestState();
const result = await page.evaluate(() => {
const a = {};
2020-05-07 10:54:55 +00:00
const b = { a };
a.b = b;
return a;
});
expect(result).toBe(undefined);
});
2020-05-07 10:54:55 +00:00
itFailsFirefox('should be able to throw a tricky error', async () => {
const { page } = getTestState();
const windowHandle = await page.evaluateHandle(() => window);
2020-05-07 10:54:55 +00:00
const errorText = await windowHandle
.jsonValue()
.catch((error_) => error_.message);
const error = await page
.evaluate((errorText) => {
throw new Error(errorText);
}, errorText)
.catch((error_) => error_);
expect(error.message).toContain(errorText);
});
2020-05-07 10:54:55 +00:00
it('should accept a string', async () => {
const { page } = getTestState();
const result = await page.evaluate('1 + 2');
expect(result).toBe(3);
});
2020-05-07 10:54:55 +00:00
it('should accept a string with semi colons', async () => {
const { page } = getTestState();
const result = await page.evaluate('1 + 5;');
expect(result).toBe(6);
});
2020-05-07 10:54:55 +00:00
it('should accept a string with comments', async () => {
const { page } = getTestState();
const result = await page.evaluate('2 + 5;\n// do some math!');
expect(result).toBe(7);
});
it('should accept element handle as an argument', async () => {
2020-05-07 10:54:55 +00:00
const { page } = getTestState();
await page.setContent('<section>42</section>');
const element = await page.$('section');
2020-05-07 10:54:55 +00:00
const text = await page.evaluate((e) => e.textContent, element);
expect(text).toBe('42');
});
it('should throw if underlying element was disposed', async () => {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
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((error_) => (error = error_));
expect(error.message).toContain('JSHandle is disposed');
});
2020-05-07 10:54:55 +00:00
itFailsFirefox(
'should throw if elementHandles are from other frames',
async () => {
const { page, server } = getTestState();
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((error_) => (error = error_));
expect(error).toBeTruthy();
expect(error.message).toContain(
'JSHandles can be evaluated only in the context they were created'
);
}
);
itFailsFirefox('should simulate a user gesture', async () => {
const { page } = getTestState();
const result = await page.evaluate(() => {
document.body.appendChild(document.createTextNode('test'));
document.execCommand('selectAll');
return document.execCommand('copy');
});
expect(result).toBe(true);
});
2020-05-07 10:54:55 +00:00
itFailsFirefox('should throw a nice error after a navigation', async () => {
const { page } = getTestState();
const executionContext = await page.mainFrame().executionContext();
await Promise.all([
page.waitForNavigation(),
2020-05-07 10:54:55 +00:00
executionContext.evaluate(() => window.location.reload()),
]);
2020-05-07 10:54:55 +00:00
const error = await executionContext
.evaluate(() => null)
.catch((error_) => error_);
expect(error.message).toContain('navigation');
});
2020-05-07 10:54:55 +00:00
itFailsFirefox(
'should not throw an error when evaluation does a navigation',
async () => {
const { page, server } = getTestState();
await page.goto(server.PREFIX + '/one-style.html');
const result = await page.evaluate(() => {
window.location = '/empty.html';
return [42];
});
expect(result).toEqual([42]);
}
);
it('should transfer 100Mb of data from page to node.js', async function () {
const { page } = getTestState();
2020-05-07 10:54:55 +00:00
const a = await page.evaluate(() =>
Array(100 * 1024 * 1024 + 1).join('a')
);
expect(a.length).toBe(100 * 1024 * 1024);
});
2020-05-07 10:54:55 +00:00
it('should throw error with detailed information on exception inside promise ', async () => {
const { page } = getTestState();
let error = null;
2020-05-07 10:54:55 +00:00
await page
.evaluate(
() =>
new Promise(() => {
throw new Error('Error in promise');
})
)
.catch((error_) => (error = error_));
expect(error.message).toContain('Error in promise');
});
});
2020-05-07 10:54:55 +00:00
describeFailsFirefox('Page.evaluateOnNewDocument', function () {
it('should evaluate before anything else on the page', async () => {
const { page, server } = getTestState();
2020-05-07 10:54:55 +00:00
await page.evaluateOnNewDocument(function () {
window.injected = 123;
});
await page.goto(server.PREFIX + '/tamperable.html');
expect(await page.evaluate(() => window.result)).toBe(123);
});
2020-05-07 10:54:55 +00:00
it('should work with CSP', async () => {
const { page, server } = getTestState();
server.setCSP('/empty.html', 'script-src ' + server.PREFIX);
2020-05-07 10:54:55 +00:00
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.
2020-05-07 10:54:55 +00:00
await page
.addScriptTag({ content: 'window.e = 10;' })
.catch((error) => void error);
expect(await page.evaluate(() => window.e)).toBe(undefined);
});
});
2020-05-07 10:54:55 +00:00
describe('Frame.evaluate', function () {
itFailsFirefox('should have different execution contexts', async () => {
const { page, server } = getTestState();
await page.goto(server.EMPTY_PAGE);
await utils.attachFrame(page, 'frame1', server.EMPTY_PAGE);
expect(page.frames().length).toBe(2);
2020-05-07 10:54:55 +00:00
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');
});
2020-05-07 10:54:55 +00:00
itFailsFirefox('should have correct execution contexts', async () => {
const { page, server } = getTestState();
await page.goto(server.PREFIX + '/frames/one-frame.html');
expect(page.frames().length).toBe(2);
2020-05-07 10:54:55 +00:00
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`);
});
2020-05-07 10:54:55 +00:00
it('should execute after cross-site navigation', async () => {
const { page, server } = getTestState();
await page.goto(server.EMPTY_PAGE);
const mainFrame = page.mainFrame();
2020-05-07 10:54:55 +00:00
expect(await mainFrame.evaluate(() => window.location.href)).toContain(
'localhost'
);
await page.goto(server.CROSS_PROCESS_PREFIX + '/empty.html');
2020-05-07 10:54:55 +00:00
expect(await mainFrame.evaluate(() => window.location.href)).toContain(
'127'
);
});
});
});