mirror of
https://github.com/puppeteer/puppeteer
synced 2024-06-14 14:02:48 +00:00
f63a123ece
This PR updates the socket transport code to swap between a Node web socket transport or a web one based on the `isNode` environment. It also adds unit tests to the browser tests that show we can connect in a browser.
56 lines
2.2 KiB
JavaScript
56 lines
2.2 KiB
JavaScript
/**
|
|
* Copyright 2020 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 { Connection } from '../lib/esm/puppeteer/common/Connection.js';
|
|
import { BrowserWebSocketTransport } from '../lib/esm/puppeteer/common/BrowserWebSocketTransport.js';
|
|
import puppeteer from '../lib/esm/puppeteer/web.js';
|
|
import expect from '../node_modules/expect/build-es5/index.js';
|
|
import { getWebSocketEndpoint } from './helper.js';
|
|
|
|
describe('creating a Connection', () => {
|
|
it('can create a real connection to the backend and send messages', async () => {
|
|
const wsUrl = getWebSocketEndpoint();
|
|
const transport = await BrowserWebSocketTransport.create(wsUrl);
|
|
|
|
const connection = new Connection(wsUrl, transport);
|
|
const result = await connection.send('Browser.getVersion');
|
|
/* We can't expect exact results as the version of Chrome/CDP might change
|
|
* and we don't want flakey tests, so let's assert the structure, which is
|
|
* enough to confirm the result was recieved successfully.
|
|
*/
|
|
expect(result).toEqual({
|
|
protocolVersion: expect.any(String),
|
|
jsVersion: expect.any(String),
|
|
revision: expect.any(String),
|
|
userAgent: expect.any(String),
|
|
product: expect.any(String),
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('puppeteer.connect', () => {
|
|
it('can connect over websocket and make requests to the backend', async () => {
|
|
const wsUrl = getWebSocketEndpoint();
|
|
const browser = await puppeteer.connect({
|
|
browserWSEndpoint: wsUrl,
|
|
});
|
|
|
|
const version = await browser.version();
|
|
const versionLooksCorrect = /.+Chrome\/\d{2}/.test(version);
|
|
expect(version).toEqual(expect.any(String));
|
|
expect(versionLooksCorrect).toEqual(true);
|
|
});
|
|
});
|