puppeteer/test/src/network.spec.ts

873 lines
30 KiB
TypeScript
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.
*/
import fs from 'fs';
import {ServerResponse} from 'http';
import path from 'path';
import expect from 'expect';
import {HTTPRequest} from 'puppeteer-core/internal/common/HTTPRequest.js';
import {HTTPResponse} from 'puppeteer-core/internal/common/HTTPResponse.js';
import {
2020-05-07 10:54:55 +00:00
getTestState,
setupTestBrowserHooks,
setupTestPageAndContextHooks,
} from './mocha-utils.js';
import utils from './utils.js';
2020-05-07 10:54:55 +00:00
describe('network', function () {
setupTestBrowserHooks();
setupTestPageAndContextHooks();
2020-05-07 10:54:55 +00:00
describe('Page.Events.Request', function () {
it('should fire for navigation requests', async () => {
const {page, server} = getTestState();
const requests: HTTPRequest[] = [];
page.on('request', request => {
return !utils.isFavicon(request) && requests.push(request);
});
(await page.goto(server.EMPTY_PAGE))!;
expect(requests.length).toBe(1);
});
it('should fire for iframes', async () => {
const {page, server} = getTestState();
const requests: HTTPRequest[] = [];
page.on('request', request => {
return !utils.isFavicon(request) && requests.push(request);
});
(await page.goto(server.EMPTY_PAGE))!;
await utils.attachFrame(page, 'frame1', server.EMPTY_PAGE);
expect(requests.length).toBe(2);
});
2020-05-07 10:54:55 +00:00
it('should fire for fetches', async () => {
const {page, server} = getTestState();
const requests: HTTPRequest[] = [];
page.on('request', request => {
return !utils.isFavicon(request) && requests.push(request);
});
(await page.goto(server.EMPTY_PAGE))!;
await page.evaluate(() => {
return fetch('/empty.html');
});
expect(requests.length).toBe(2);
});
});
2020-05-07 10:54:55 +00:00
describe('Request.frame', function () {
it('should work for main frame navigation request', async () => {
const {page, server} = getTestState();
const requests: HTTPRequest[] = [];
page.on('request', request => {
return !utils.isFavicon(request) && requests.push(request);
});
(await page.goto(server.EMPTY_PAGE))!;
expect(requests.length).toBe(1);
expect(requests[0]!.frame()).toBe(page.mainFrame());
});
it('should work for subframe navigation request', async () => {
const {page, server} = getTestState();
(await page.goto(server.EMPTY_PAGE))!;
const requests: HTTPRequest[] = [];
page.on('request', request => {
return !utils.isFavicon(request) && requests.push(request);
});
await utils.attachFrame(page, 'frame1', server.EMPTY_PAGE);
expect(requests.length).toBe(1);
expect(requests[0]!.frame()).toBe(page.frames()[1]!);
});
2020-05-07 10:54:55 +00:00
it('should work for fetch requests', async () => {
const {page, server} = getTestState();
(await page.goto(server.EMPTY_PAGE))!;
let requests: HTTPRequest[] = [];
page.on('request', request => {
return !utils.isFavicon(request) && requests.push(request);
});
await page.evaluate(() => {
return fetch('/digits/1.png');
});
requests = requests.filter(request => {
return !request.url().includes('favicon');
});
expect(requests.length).toBe(1);
expect(requests[0]!.frame()).toBe(page.mainFrame());
});
});
describe('Request.headers', function () {
it('should define Chrome as user agent header', async () => {
const {page, server} = getTestState();
const response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.request().headers()['user-agent']).toContain('Chrome');
});
it('should define Firefox as user agent header', async () => {
const {page, server} = getTestState();
const response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.request().headers()['user-agent']).toContain('Firefox');
});
});
describe('Response.headers', function () {
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
server.setRoute('/empty.html', (_req, res) => {
res.setHeader('foo', 'bar');
res.end();
});
const response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.headers()['foo']).toBe('bar');
});
});
describe('Request.initiator', () => {
it('should return the initiator', async () => {
const {page, server} = getTestState();
const initiators = new Map();
page.on('request', request => {
return initiators.set(
request.url().split('/').pop(),
request.initiator()
);
});
await page.goto(server.PREFIX + '/initiator.html');
expect(initiators.get('initiator.html').type).toBe('other');
expect(initiators.get('initiator.js').type).toBe('parser');
expect(initiators.get('initiator.js').url).toBe(
server.PREFIX + '/initiator.html'
);
expect(initiators.get('frame.html').type).toBe('parser');
expect(initiators.get('frame.html').url).toBe(
server.PREFIX + '/initiator.html'
);
expect(initiators.get('script.js').type).toBe('parser');
expect(initiators.get('script.js').url).toBe(
server.PREFIX + '/frames/frame.html'
);
expect(initiators.get('style.css').type).toBe('parser');
expect(initiators.get('style.css').url).toBe(
server.PREFIX + '/frames/frame.html'
);
expect(initiators.get('initiator.js').type).toBe('parser');
expect(initiators.get('injectedfile.js').type).toBe('script');
expect(initiators.get('injectedfile.js').stack.callFrames[0]!.url).toBe(
server.PREFIX + '/initiator.js'
);
expect(initiators.get('injectedstyle.css').type).toBe('script');
expect(initiators.get('injectedstyle.css').stack.callFrames[0]!.url).toBe(
server.PREFIX + '/initiator.js'
);
expect(initiators.get('initiator.js').url).toBe(
server.PREFIX + '/initiator.html'
);
});
});
describe('Response.fromCache', function () {
2020-05-07 10:54:55 +00:00
it('should return |false| for non-cached content', async () => {
const {page, server} = getTestState();
const response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.fromCache()).toBe(false);
});
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
const responses = new Map();
page.on('response', r => {
return (
2020-05-07 10:54:55 +00:00
!utils.isFavicon(r.request()) &&
responses.set(r.url().split('/').pop(), r)
);
});
// Load and re-load to make sure it's cached.
await page.goto(server.PREFIX + '/cached/one-style.html');
await page.reload();
expect(responses.size).toBe(2);
expect(responses.get('one-style.css').status()).toBe(200);
expect(responses.get('one-style.css').fromCache()).toBe(true);
expect(responses.get('one-style.html').status()).toBe(304);
expect(responses.get('one-style.html').fromCache()).toBe(false);
});
});
describe('Response.fromServiceWorker', function () {
2020-05-07 10:54:55 +00:00
it('should return |false| for non-service-worker content', async () => {
const {page, server} = getTestState();
const response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.fromServiceWorker()).toBe(false);
});
2020-05-07 10:54:55 +00:00
it('Response.fromServiceWorker', async () => {
const {page, server} = getTestState();
const responses = new Map();
page.on('response', r => {
return (
!utils.isFavicon(r) && responses.set(r.url().split('/').pop(), r)
);
});
// Load and re-load to make sure serviceworker is installed and running.
2020-05-07 10:54:55 +00:00
await page.goto(server.PREFIX + '/serviceworkers/fetch/sw.html', {
waitUntil: 'networkidle2',
});
await page.evaluate(async () => {
return await (globalThis as any).activationPromise;
});
await page.reload();
expect(responses.size).toBe(2);
expect(responses.get('sw.html').status()).toBe(200);
expect(responses.get('sw.html').fromServiceWorker()).toBe(true);
expect(responses.get('style.css').status()).toBe(200);
expect(responses.get('style.css').fromServiceWorker()).toBe(true);
});
});
describe('Request.postData', function () {
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
(await page.goto(server.EMPTY_PAGE))!;
server.setRoute('/post', (_req, res) => {
return res.end();
});
let request!: HTTPRequest;
page.on('request', r => {
2022-06-14 11:55:35 +00:00
if (!utils.isFavicon(r)) {
request = r;
}
});
await page.evaluate(() => {
return fetch('./post', {
2020-05-07 10:54:55 +00:00
method: 'POST',
body: JSON.stringify({foo: 'bar'}),
});
});
expect(request).toBeTruthy();
expect(request.postData()).toBe('{"foo":"bar"}');
});
2020-05-07 10:54:55 +00:00
it('should be |undefined| when there is no post data', async () => {
const {page, server} = getTestState();
const response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.request().postData()).toBe(undefined);
});
});
describe('Response.text', function () {
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
const response = (await page.goto(server.PREFIX + '/simple.json'))!;
const responseText = (await response.text()).trimEnd();
expect(responseText).toBe('{"foo": "bar"}');
});
2020-05-07 10:54:55 +00:00
it('should return uncompressed text', async () => {
const {page, server} = getTestState();
server.enableGzip('/simple.json');
const response = (await page.goto(server.PREFIX + '/simple.json'))!;
expect(response.headers()['content-encoding']).toBe('gzip');
const responseText = (await response.text()).trimEnd();
expect(responseText).toBe('{"foo": "bar"}');
});
2020-05-07 10:54:55 +00:00
it('should throw when requesting body of redirected response', async () => {
const {page, server} = getTestState();
server.setRedirect('/foo.html', '/empty.html');
const response = (await page.goto(server.PREFIX + '/foo.html'))!;
const redirectChain = response.request().redirectChain();
expect(redirectChain.length).toBe(1);
const redirected = redirectChain[0]!.response()!;
expect(redirected.status()).toBe(302);
let error!: Error;
await redirected.text().catch(error_ => {
return (error = error_);
});
2020-05-07 10:54:55 +00:00
expect(error.message).toContain(
'Response body is unavailable for redirect responses'
);
});
2020-05-07 10:54:55 +00:00
it('should wait until response completes', async () => {
const {page, server} = getTestState();
(await page.goto(server.EMPTY_PAGE))!;
// Setup server to trap request.
let serverResponse!: ServerResponse;
server.setRoute('/get', (_req, res) => {
serverResponse = res;
// In Firefox, |fetch| will be hanging until it receives |Content-Type| header
// from server.
res.setHeader('Content-Type', 'text/plain; charset=utf-8');
res.write('hello ');
});
// Setup page to trap response.
let requestFinished = false;
page.on('requestfinished', r => {
return (requestFinished = requestFinished || r.url().includes('/get'));
});
// send request and wait for server response
const [pageResponse] = await Promise.all([
page.waitForResponse(r => {
return !utils.isFavicon(r.request());
}),
page.evaluate(() => {
return fetch('./get', {method: 'GET'});
}),
server.waitForRequest('/get'),
]);
expect(serverResponse).toBeTruthy();
expect(pageResponse).toBeTruthy();
expect(pageResponse.status()).toBe(200);
expect(requestFinished).toBe(false);
const responseText = pageResponse.text();
// Write part of the response and wait for it to be flushed.
await new Promise(x => {
return serverResponse.write('wor', x);
});
// Finish response.
await new Promise<void>(x => {
serverResponse.end('ld!', () => {
return x();
});
});
expect(await responseText).toBe('hello world!');
});
});
describe('Response.json', function () {
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
const response = (await page.goto(server.PREFIX + '/simple.json'))!;
expect(await response.json()).toEqual({foo: 'bar'});
});
});
describe('Response.buffer', function () {
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
const response = (await page.goto(server.PREFIX + '/pptr.png'))!;
2020-05-07 10:54:55 +00:00
const imageBuffer = fs.readFileSync(
path.join(__dirname, '../assets', 'pptr.png')
2020-05-07 10:54:55 +00:00
);
const responseBuffer = await response.buffer();
expect(responseBuffer.equals(imageBuffer)).toBe(true);
});
2020-05-07 10:54:55 +00:00
it('should work with compression', async () => {
const {page, server} = getTestState();
server.enableGzip('/pptr.png');
const response = (await page.goto(server.PREFIX + '/pptr.png'))!;
2020-05-07 10:54:55 +00:00
const imageBuffer = fs.readFileSync(
path.join(__dirname, '../assets', 'pptr.png')
2020-05-07 10:54:55 +00:00
);
const responseBuffer = await response.buffer();
expect(responseBuffer.equals(imageBuffer)).toBe(true);
});
it('should throw if the response does not have a body', async () => {
const {page, server} = getTestState();
await page.goto(server.PREFIX + '/empty.html');
server.setRoute('/test.html', (_req, res) => {
res.setHeader('Access-Control-Allow-Origin', '*');
res.setHeader('Access-Control-Allow-Headers', 'x-ping');
res.end('Hello World');
});
const url = server.CROSS_PROCESS_PREFIX + '/test.html';
const responsePromise = new Promise<HTTPResponse>(resolve => {
page.on('response', response => {
// Get the preflight response.
if (
response.request().method() === 'OPTIONS' &&
response.url() === url
) {
resolve(response);
}
});
});
// Trigger a request with a preflight.
feat!: type inference for evaluation types (#8547) This PR greatly improves the types within Puppeteer: - **Almost everything** is auto-deduced. - Parameters don't need to be specified in the function. They are deduced from the spread. - Return types don't need to be specified. They are deduced from the function. (More on this below) - Selections based on tag names correctly deduce element type, similar to TypeScript's mechanism for `getElementByTagName`. - [**BREAKING CHANGE**] We've removed the ability to declare return types in type arguments for the following reasons: 1. Setting them will indubitably break auto-deduction. 2. You can just use `as ...` in TypeScript to coerce the correct type (given it makes sense). - [**BREAKING CHANGE**] `waitFor` is officially gone. To migrate to these changes, there are only four things you may need to change: - If you set a return type using the `ReturnType` type parameter, remove it and use `as ...` and `HandleFor` (if necessary). ⛔ `evaluate<ReturnType>(a: number, b: number) => {...}, a, b)` ✅ `(await evaluate(a, b) => {...}, a, b)) as ReturnType` ⛔ `evaluateHandle<ReturnType>(a: number, b: number) => {...}, a, b)` ✅ `(await evaluateHandle(a, b) => {...}, a, b)) as HandleFor<ReturnType>` - If you set any type parameters in the *parameters* of an evaluation function, remove them. ⛔ `evaluate(a: number, b: number) => {...}, a, b)` ✅ `evaluate(a, b) => {...}, a, b)` - If you set any type parameters in the method's declaration, remove them. ⛔ `evaluate<(a: number, b: number) => void>((a, b) => {...}, a, b)` ✅ `evaluate(a, b) => {...}, a, b)`
2022-06-23 09:29:46 +00:00
await page.evaluate(async src => {
const response = await fetch(src, {
method: 'POST',
headers: {'x-ping': 'pong'},
});
return response;
}, url);
const response = await responsePromise;
await expect(response.buffer()).rejects.toThrowError(
'Could not load body for this request. This might happen if the request is a preflight request.'
);
});
});
describe('Response.statusText', function () {
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
server.setRoute('/cool', (_req, res) => {
res.writeHead(200, 'cool!');
res.end();
});
const response = (await page.goto(server.PREFIX + '/cool'))!;
expect(response.statusText()).toBe('cool!');
});
it('handles missing status text', async () => {
const {page, server} = getTestState();
server.setRoute('/nostatus', (_req, res) => {
res.writeHead(200, '');
res.end();
});
const response = (await page.goto(server.PREFIX + '/nostatus'))!;
expect(response.statusText()).toBe('');
});
});
describe('Response.timing', function () {
it('returns timing information', async () => {
const {page, server} = getTestState();
const responses: HTTPResponse[] = [];
page.on('response', response => {
return responses.push(response);
});
(await page.goto(server.EMPTY_PAGE))!;
expect(responses.length).toBe(1);
expect(responses[0]!.timing()!.receiveHeadersEnd).toBeGreaterThan(0);
});
});
describe('Network Events', function () {
2020-05-07 10:54:55 +00:00
it('Page.Events.Request', async () => {
const {page, server} = getTestState();
const requests: HTTPRequest[] = [];
page.on('request', request => {
return requests.push(request);
});
(await page.goto(server.EMPTY_PAGE))!;
expect(requests.length).toBe(1);
expect(requests[0]!.url()).toBe(server.EMPTY_PAGE);
expect(requests[0]!.resourceType()).toBe('document');
expect(requests[0]!.method()).toBe('GET');
expect(requests[0]!.response()).toBeTruthy();
expect(requests[0]!.frame() === page.mainFrame()).toBe(true);
expect(requests[0]!.frame()!.url()).toBe(server.EMPTY_PAGE);
});
it('Page.Events.RequestServedFromCache', async () => {
const {page, server} = getTestState();
const cached: string[] = [];
page.on('requestservedfromcache', r => {
return cached.push(r.url().split('/').pop()!);
});
await page.goto(server.PREFIX + '/cached/one-style.html');
expect(cached).toEqual([]);
await page.reload();
expect(cached).toEqual(['one-style.css']);
});
2020-05-07 10:54:55 +00:00
it('Page.Events.Response', async () => {
const {page, server} = getTestState();
const responses: HTTPResponse[] = [];
page.on('response', response => {
return responses.push(response);
});
(await page.goto(server.EMPTY_PAGE))!;
expect(responses.length).toBe(1);
expect(responses[0]!.url()).toBe(server.EMPTY_PAGE);
expect(responses[0]!.status()).toBe(200);
expect(responses[0]!.ok()).toBe(true);
expect(responses[0]!.request()).toBeTruthy();
const remoteAddress = responses[0]!.remoteAddress();
// Either IPv6 or IPv4, depending on environment.
2020-05-07 10:54:55 +00:00
expect(
remoteAddress.ip!.includes('::1') || remoteAddress.ip === '127.0.0.1'
2020-05-07 10:54:55 +00:00
).toBe(true);
expect(remoteAddress.port).toBe(server.PORT);
});
2020-05-07 10:54:55 +00:00
it('Page.Events.RequestFailed', async () => {
const {page, server, isChrome} = getTestState();
await page.setRequestInterception(true);
page.on('request', request => {
2022-06-14 11:55:35 +00:00
if (request.url().endsWith('css')) {
request.abort();
} else {
request.continue();
}
});
const failedRequests: HTTPRequest[] = [];
page.on('requestfailed', request => {
return failedRequests.push(request);
});
await page.goto(server.PREFIX + '/one-style.html');
expect(failedRequests.length).toBe(1);
expect(failedRequests[0]!.url()).toContain('one-style.css');
expect(failedRequests[0]!.response()).toBe(null);
expect(failedRequests[0]!.resourceType()).toBe('stylesheet');
2022-06-14 11:55:35 +00:00
if (isChrome) {
expect(failedRequests[0]!.failure()!.errorText).toBe('net::ERR_FAILED');
2022-06-14 11:55:35 +00:00
} else {
expect(failedRequests[0]!.failure()!.errorText).toBe(
'NS_ERROR_FAILURE'
);
2022-06-14 11:55:35 +00:00
}
expect(failedRequests[0]!.frame()).toBeTruthy();
});
2020-05-07 10:54:55 +00:00
it('Page.Events.RequestFinished', async () => {
const {page, server} = getTestState();
const requests: HTTPRequest[] = [];
page.on('requestfinished', request => {
return requests.push(request);
});
(await page.goto(server.EMPTY_PAGE))!;
expect(requests.length).toBe(1);
expect(requests[0]!.url()).toBe(server.EMPTY_PAGE);
expect(requests[0]!.response()).toBeTruthy();
expect(requests[0]!.frame() === page.mainFrame()).toBe(true);
expect(requests[0]!.frame()!.url()).toBe(server.EMPTY_PAGE);
});
2020-05-07 10:54:55 +00:00
it('should fire events in proper order', async () => {
const {page, server} = getTestState();
const events: string[] = [];
page.on('request', () => {
return events.push('request');
});
page.on('response', () => {
return events.push('response');
});
page.on('requestfinished', () => {
return events.push('requestfinished');
});
(await page.goto(server.EMPTY_PAGE))!;
expect(events).toEqual(['request', 'response', 'requestfinished']);
});
2020-05-07 10:54:55 +00:00
it('should support redirects', async () => {
const {page, server} = getTestState();
const events: string[] = [];
page.on('request', request => {
return events.push(`${request.method()} ${request.url()}`);
});
page.on('response', response => {
return events.push(`${response.status()} ${response.url()}`);
});
page.on('requestfinished', request => {
return events.push(`DONE ${request.url()}`);
});
page.on('requestfailed', request => {
return events.push(`FAIL ${request.url()}`);
});
server.setRedirect('/foo.html', '/empty.html');
const FOO_URL = server.PREFIX + '/foo.html';
const response = (await page.goto(FOO_URL))!;
expect(events).toEqual([
`GET ${FOO_URL}`,
`302 ${FOO_URL}`,
`DONE ${FOO_URL}`,
`GET ${server.EMPTY_PAGE}`,
`200 ${server.EMPTY_PAGE}`,
2020-05-07 10:54:55 +00:00
`DONE ${server.EMPTY_PAGE}`,
]);
// Check redirect chain
const redirectChain = response.request().redirectChain();
expect(redirectChain.length).toBe(1);
expect(redirectChain[0]!.url()).toContain('/foo.html');
expect(redirectChain[0]!.response()!.remoteAddress().port).toBe(
2020-05-07 10:54:55 +00:00
server.PORT
);
});
});
describe('Request.isNavigationRequest', () => {
it('should work', async () => {
const {page, server} = getTestState();
const requests = new Map();
page.on('request', request => {
return requests.set(request.url().split('/').pop(), request);
});
server.setRedirect('/rrredirect', '/frames/one-frame.html');
await page.goto(server.PREFIX + '/rrredirect');
expect(requests.get('rrredirect').isNavigationRequest()).toBe(true);
expect(requests.get('one-frame.html').isNavigationRequest()).toBe(true);
expect(requests.get('frame.html').isNavigationRequest()).toBe(true);
expect(requests.get('script.js').isNavigationRequest()).toBe(false);
expect(requests.get('style.css').isNavigationRequest()).toBe(false);
});
it('should work with request interception', async () => {
const {page, server} = getTestState();
const requests = new Map();
page.on('request', request => {
requests.set(request.url().split('/').pop(), request);
request.continue();
});
await page.setRequestInterception(true);
server.setRedirect('/rrredirect', '/frames/one-frame.html');
await page.goto(server.PREFIX + '/rrredirect');
expect(requests.get('rrredirect').isNavigationRequest()).toBe(true);
expect(requests.get('one-frame.html').isNavigationRequest()).toBe(true);
expect(requests.get('frame.html').isNavigationRequest()).toBe(true);
expect(requests.get('script.js').isNavigationRequest()).toBe(false);
expect(requests.get('style.css').isNavigationRequest()).toBe(false);
});
it('should work when navigating to image', async () => {
const {page, server} = getTestState();
const requests: HTTPRequest[] = [];
page.on('request', request => {
return requests.push(request);
});
(await page.goto(server.PREFIX + '/pptr.png'))!;
expect(requests[0]!.isNavigationRequest()).toBe(true);
});
});
describe('Page.setExtraHTTPHeaders', function () {
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
await page.setExtraHTTPHeaders({
2020-05-07 10:54:55 +00:00
foo: 'bar',
});
const [request] = await Promise.all([
server.waitForRequest('/empty.html'),
page.goto(server.EMPTY_PAGE),
]);
expect(request.headers['foo']).toBe('bar');
});
2020-05-07 10:54:55 +00:00
it('should throw for non-string header values', async () => {
const {page} = getTestState();
let error!: Error;
try {
// @ts-expect-error purposeful bad input
await page.setExtraHTTPHeaders({foo: 1});
} catch (error_) {
error = error_ as Error;
}
2020-05-07 10:54:55 +00:00
expect(error.message).toBe(
'Expected value of header "foo" to be String, but "number" is found.'
);
});
});
describe('Page.authenticate', function () {
2020-05-07 10:54:55 +00:00
it('should work', async () => {
const {page, server} = getTestState();
server.setAuth('/empty.html', 'user', 'pass');
let response;
try {
response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.status()).toBe(401);
} catch (error) {
// In headful, an error is thrown instead of 401.
if (
!(error as Error).message.startsWith(
'net::ERR_INVALID_AUTH_CREDENTIALS'
)
) {
throw error;
}
}
await page.authenticate({
username: 'user',
2020-05-07 10:54:55 +00:00
password: 'pass',
});
response = (await page.reload())!;
expect(response.status()).toBe(200);
});
2020-05-07 10:54:55 +00:00
it('should fail if wrong credentials', async () => {
const {page, server} = getTestState();
// Use unique user/password since Chrome caches credentials per origin.
server.setAuth('/empty.html', 'user2', 'pass2');
await page.authenticate({
username: 'foo',
2020-05-07 10:54:55 +00:00
password: 'bar',
});
const response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.status()).toBe(401);
});
2020-05-07 10:54:55 +00:00
it('should allow disable authentication', async () => {
const {page, server} = getTestState();
// Use unique user/password since Chrome caches credentials per origin.
server.setAuth('/empty.html', 'user3', 'pass3');
await page.authenticate({
username: 'user3',
2020-05-07 10:54:55 +00:00
password: 'pass3',
});
let response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.status()).toBe(200);
await page.authenticate({
username: '',
password: '',
});
// Navigate to a different origin to bust Chrome's credential caching.
try {
response = (await page.goto(
server.CROSS_PROCESS_PREFIX + '/empty.html'
))!;
expect(response.status()).toBe(401);
} catch (error) {
// In headful, an error is thrown instead of 401.
if (
!(error as Error).message.startsWith(
'net::ERR_INVALID_AUTH_CREDENTIALS'
)
) {
throw error;
}
}
});
it('should not disable caching', async () => {
const {page, server} = getTestState();
// Use unique user/password since Chrome caches credentials per origin.
server.setAuth('/cached/one-style.css', 'user4', 'pass4');
server.setAuth('/cached/one-style.html', 'user4', 'pass4');
await page.authenticate({
username: 'user4',
password: 'pass4',
});
const responses = new Map();
page.on('response', r => {
return responses.set(r.url().split('/').pop(), r);
});
// Load and re-load to make sure it's cached.
await page.goto(server.PREFIX + '/cached/one-style.html');
await page.reload();
expect(responses.get('one-style.css').status()).toBe(200);
expect(responses.get('one-style.css').fromCache()).toBe(true);
expect(responses.get('one-style.html').status()).toBe(304);
expect(responses.get('one-style.html').fromCache()).toBe(false);
});
});
describe('raw network headers', async () => {
it('Same-origin set-cookie navigation', async () => {
const {page, server} = getTestState();
const setCookieString = 'foo=bar';
server.setRoute('/empty.html', (_req, res) => {
res.setHeader('set-cookie', setCookieString);
res.end('hello world');
});
const response = (await page.goto(server.EMPTY_PAGE))!;
expect(response.headers()['set-cookie']).toBe(setCookieString);
});
it('Same-origin set-cookie subresource', async () => {
const {page, server} = getTestState();
(await page.goto(server.EMPTY_PAGE))!;
const setCookieString = 'foo=bar';
server.setRoute('/foo', (_req, res) => {
res.setHeader('set-cookie', setCookieString);
res.end('hello world');
});
const responsePromise = new Promise<HTTPResponse>(resolve => {
return page.on('response', response => {
return resolve(response);
});
});
page.evaluate(() => {
const xhr = new XMLHttpRequest();
xhr.open('GET', '/foo');
xhr.send();
});
const subresourceResponse = await responsePromise;
expect(subresourceResponse.headers()['set-cookie']).toBe(setCookieString);
});
it('Cross-origin set-cookie', async () => {
const {httpsServer, puppeteer, defaultBrowserOptions} = getTestState();
const browser = await puppeteer.launch({
...defaultBrowserOptions,
ignoreHTTPSErrors: true,
});
const page = await browser.newPage();
try {
await page.goto(httpsServer.PREFIX + '/empty.html');
const setCookieString = 'hello=world';
httpsServer.setRoute('/setcookie.html', (_req, res) => {
res.setHeader('Access-Control-Allow-Origin', '*');
res.setHeader('set-cookie', setCookieString);
res.end();
});
await page.goto(httpsServer.PREFIX + '/setcookie.html');
const url = httpsServer.CROSS_PROCESS_PREFIX + '/setcookie.html';
const response = await new Promise<HTTPResponse>(resolve => {
page.on('response', response => {
if (response.url() === url) {
resolve(response);
}
});
feat!: type inference for evaluation types (#8547) This PR greatly improves the types within Puppeteer: - **Almost everything** is auto-deduced. - Parameters don't need to be specified in the function. They are deduced from the spread. - Return types don't need to be specified. They are deduced from the function. (More on this below) - Selections based on tag names correctly deduce element type, similar to TypeScript's mechanism for `getElementByTagName`. - [**BREAKING CHANGE**] We've removed the ability to declare return types in type arguments for the following reasons: 1. Setting them will indubitably break auto-deduction. 2. You can just use `as ...` in TypeScript to coerce the correct type (given it makes sense). - [**BREAKING CHANGE**] `waitFor` is officially gone. To migrate to these changes, there are only four things you may need to change: - If you set a return type using the `ReturnType` type parameter, remove it and use `as ...` and `HandleFor` (if necessary). ⛔ `evaluate<ReturnType>(a: number, b: number) => {...}, a, b)` ✅ `(await evaluate(a, b) => {...}, a, b)) as ReturnType` ⛔ `evaluateHandle<ReturnType>(a: number, b: number) => {...}, a, b)` ✅ `(await evaluateHandle(a, b) => {...}, a, b)) as HandleFor<ReturnType>` - If you set any type parameters in the *parameters* of an evaluation function, remove them. ⛔ `evaluate(a: number, b: number) => {...}, a, b)` ✅ `evaluate(a, b) => {...}, a, b)` - If you set any type parameters in the method's declaration, remove them. ⛔ `evaluate<(a: number, b: number) => void>((a, b) => {...}, a, b)` ✅ `evaluate(a, b) => {...}, a, b)`
2022-06-23 09:29:46 +00:00
page.evaluate(src => {
const xhr = new XMLHttpRequest();
xhr.open('GET', src);
xhr.send();
}, url);
});
expect(response.headers()['set-cookie']).toBe(setCookieString);
} finally {
await page.close();
await browser.close();
}
});
});
});