playwright/test/browsercontext-expose-function.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 { it, expect } from './fixtures';

it('expose binding should work', async ({context}) => {
  let bindingSource;
  await context.exposeBinding('add', (source, a, b) => {
    bindingSource = source;
    return a + b;
  });
  const page = await context.newPage();
  const result = await page.evaluate('add(5, 6)');
  expect(bindingSource.context).toBe(context);
  expect(bindingSource.page).toBe(page);
  expect(bindingSource.frame).toBe(page.mainFrame());
  expect(result).toEqual(11);
});

it('should work', async ({context, server}) => {
  await context.exposeFunction('add', (a, b) => a + b);
  const page = await context.newPage();
  await page.exposeFunction('mul', (a, b) => a * b);
  await context.exposeFunction('sub', (a, b) => a - b);
  await context.exposeBinding('addHandle', async ({ frame }, a, b) => {
    const handle = await frame.evaluateHandle(([a, b]) => a + b, [a, b]);
    return handle;
  });
  const result = await page.evaluate('(async () => ({ mul: await mul(9, 4), add: await add(9, 4), sub: await sub(9, 4), addHandle: await addHandle(5, 6) }))()');
  expect(result).toEqual({ mul: 36, add: 13, sub: 5, addHandle: 11 });
});

it('should throw for duplicate registrations', async ({context, server}) => {
  await context.exposeFunction('foo', () => {});
  await context.exposeFunction('bar', () => {});
  let error = await context.exposeFunction('foo', () => {}).catch(e => e);
  expect(error.message).toContain('Function "foo" has been already registered');
  const page = await context.newPage();
  error = await page.exposeFunction('foo', () => {}).catch(e => e);
  expect(error.message).toContain('Function "foo" has been already registered in the browser context');
  await page.exposeFunction('baz', () => {});
  error = await context.exposeFunction('baz', () => {}).catch(e => e);
  expect(error.message).toContain('Function "baz" has been already registered in one of the pages');
});

it('should be callable from-inside addInitScript', async ({context, server}) => {
  let args = [];
  await context.exposeFunction('woof', function(arg) {
    args.push(arg);
  });
  await context.addInitScript('window["woof"]("context")');
  const page = await context.newPage();
  await page.addInitScript('window["woof"]("page")');
  args = [];
  await page.reload();
  expect(args).toEqual(['context', 'page']);
});

it('exposeBindingHandle should work', async ({context}) => {
  let target;
  await context.exposeBinding('logme', (source, t) => {
    target = t;
    return 17;
  }, { handle: true });
  const page = await context.newPage();
  const result = await page.evaluate(async function() {
    return window['logme']({ foo: 42 });
  });
  expect(await target.evaluate(x => x.foo)).toBe(42);
  expect(result).toEqual(17);
});

it('extendInjectedScript should work', async ({ context, server }) => {
  await (context as any)._extendInjectedScript(`var pwExport = (() => {
    class Foo {
      constructor() {
        window._counter = (window._counter || 0) + 1;
      }
    }
    return Foo;
  })()`);

  const page = await context.newPage();
  await page.waitForFunction(() => (window as any)._counter === 1);

  await page.goto(server.EMPTY_PAGE);
  await page.waitForFunction(() => (window as any)._counter === 1);

  await Promise.all([
    page.waitForNavigation(),
    page.evaluate(() => history.pushState({}, '', '/url.html'))
  ]);
  expect(await page.evaluate(() => (window as any)._counter)).toBe(1);
});