playwright/tests/page/page-request-intercept.spec.ts

/**
 * 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.
 */

import { fail } from 'assert';
import type { Route } from '../../index';
import { test as it, expect } from './pageTest';

it('should fulfill intercepted response', async ({page, server, browserName}) => {
  it.fixme(browserName === 'firefox');
  await page.route('**/*', async route => {
    // @ts-expect-error
    await route._intercept({});
    await route.fulfill({
      status: 201,
      headers: {
        foo: 'bar'
      },
      contentType: 'text/plain',
      body: 'Yo, page!'
    });
  });
  const response = await page.goto(server.PREFIX + '/empty.html');
  expect(response.status()).toBe(201);
  expect(response.headers().foo).toBe('bar');
  expect(response.headers()['content-type']).toBe('text/plain');
  expect(await page.evaluate(() => document.body.textContent)).toBe('Yo, page!');
});

it('should throw on continue after intercept', async ({page, server, browserName}) => {
  it.fixme(browserName === 'firefox');

  let routeCallback;
  const routePromise = new Promise<Route>(f => routeCallback = f);
  await page.route('**', routeCallback);

  page.goto(server.EMPTY_PAGE).catch(e => {});
  const route = await routePromise;
  // @ts-expect-error
  await route._intercept();
  try {
    await route.continue();
    fail('did not throw');
  } catch (e) {
    expect(e.message).toContain('Cannot call continue after response interception!');
  }
});

it('should support fulfill after intercept', async ({page, server, browserName, browserMajorVersion}) => {
  it.fixme(browserName === 'firefox');
  it.skip(browserName === 'chromium' && browserMajorVersion <= 91);
  const requestPromise = server.waitForRequest('/empty.html');
  await page.route('**', async route => {
    // @ts-expect-error
    await route._intercept();
    await route.fulfill();
  });
  await page.goto(server.EMPTY_PAGE);
  const request = await requestPromise;
  expect(request.url).toBe('/empty.html');
});


it('should support request overrides', async ({page, server, browserName, browserMajorVersion}) => {
  it.fixme(browserName === 'firefox');
  it.skip(browserName === 'chromium' && browserMajorVersion <= 91);
  const requestPromise = server.waitForRequest('/empty.html');
  await page.route('**/foo', async route => {
    // @ts-expect-error
    await route._intercept({
      url: server.EMPTY_PAGE,
      method: 'POST',
      headers: {'foo': 'bar'},
      postData: 'my data',
    });
    await route.fulfill();
  });
  await page.goto(server.PREFIX + '/foo');
  const request = await requestPromise;
  expect(request.method).toBe('POST');
  expect(request.url).toBe('/empty.html');
  expect(request.headers['foo']).toBe('bar');
  expect((await request.postBody).toString('utf8')).toBe('my data');
});

it('should give access to the intercepted response', async ({page, server, browserName}) => {
  it.fixme(browserName === 'firefox');
  // it.fixme(browserName === 'webkit');

  await page.goto(server.EMPTY_PAGE);

  let routeCallback;
  const routePromise = new Promise<Route>(f => routeCallback = f);
  await page.route('**/title.html', routeCallback);

  const evalPromise = page.evaluate(url => fetch(url), server.PREFIX + '/title.html').catch(console.log);

  const route = await routePromise;
  // @ts-expect-error
  const response = await route._intercept();

  expect(response.status()).toBe(200);
  expect(response.ok()).toBeTruthy();
  expect(response.url()).toBe(server.PREFIX + '/title.html');
  expect(response.headers()['content-type']).toBe('text/html; charset=utf-8');

  await Promise.all([route.fulfill(), evalPromise]);
});

it('should give access to the intercepted response body', async ({page, server, browserName}) => {
  it.fixme(browserName === 'firefox');

  await page.goto(server.EMPTY_PAGE);

  let routeCallback;
  const routePromise = new Promise<Route>(f => routeCallback = f);
  await page.route('**/simple.json', routeCallback);

  const evalPromise = page.evaluate(url => fetch(url), server.PREFIX + '/simple.json').catch(console.log);

  const route = await routePromise;
  // @ts-expect-error
  const response = await route._intercept();

  expect((await response.text())).toBe('{"foo": "bar"}\n');

  await Promise.all([route.fulfill(), evalPromise]);
});

it('should be abortable after interception', async ({page, server, browserName}) => {
  it.fixme(browserName === 'firefox');
  it.fixme(browserName === 'webkit');

  await page.route(/\.css$/, async route => {
    // @ts-expect-error
    await route._intercept();
    await route.abort();
  });
  let failed = false;
  page.on('requestfailed', request => {
    if (request.url().includes('.css'))
      failed = true;
  });
  const response = await page.goto(server.PREFIX + '/one-style.html');
  expect(response.ok()).toBe(true);
  expect(response.request().failure()).toBe(null);
  expect(failed).toBe(true);
});

it('should fulfill after redirects', async ({page, server}) => {
  it.fixme();
  server.setRedirect('/redirect/1.html', '/redirect/2.html');
  server.setRedirect('/redirect/2.html', '/empty.html');
  const expectedUrls = ['/redirect/1.html', '/redirect/2.html', '/empty.html'].map(s => server.PREFIX + s);
  const requestUrls = [];
  const responseUrls = [];
  const requestFinishedUrls = [];
  page.on('request', request => requestUrls.push(request.url()));
  page.on('response', response => responseUrls.push(response.url()));
  page.on('requestfinished', request => requestFinishedUrls.push(request.url()));
  await page.route('**/*', async route => {
    // @ts-expect-error
    await route._intercept({});
    await route.fulfill({
      status: 201,
      headers: {
        foo: 'bar'
      },
      contentType: 'text/plain',
      body: 'Yo, page!'
    });
  });
  const response = await page.goto(server.PREFIX + '/redirect/1.html');
  expect(requestUrls).toEqual(expectedUrls);
  expect(responseUrls).toEqual(expectedUrls);
  await response.finished();
  expect(requestFinishedUrls).toEqual(expectedUrls);

  const redirectChain = [];
  for (let req = response.request(); req; req = req.redirectedFrom())
    redirectChain.unshift(req.url());
  expect(redirectChain).toEqual(expectedUrls);

  expect(response.status()).toBe(201);
  expect(response.headers().foo).toBe('bar');
  expect(response.headers()['content-type']).toBe('text/plain');
  expect(await response.text()).toBe('Yo, page!');
});