mirror of
https://github.com/puppeteer/puppeteer
synced 2024-06-14 14:02:48 +00:00
1c2adf61e9
Some of the worker tests were failing on the bots. After investigating, I found one race in the test, and one race upstream in Chromium which I filed upstream as https://crbug.com/846099. But I added a small hack here as a temporary workaround. References #2632
66 lines
2.3 KiB
JavaScript
66 lines
2.3 KiB
JavaScript
/**
|
|
* 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 EventEmitter = require('events');
|
|
const {helper, debugError} = require('./helper');
|
|
const {ExecutionContext, JSHandle} = require('./ExecutionContext');
|
|
|
|
class Worker extends EventEmitter {
|
|
/**
|
|
* @param {Puppeteer.CDPSession} client
|
|
* @param {string} url
|
|
* @param {function(!Protocol.Log.entryAddedPayload)} logEntryAdded
|
|
*/
|
|
constructor(client, url, logEntryAdded) {
|
|
super();
|
|
this._client = client;
|
|
this._url = url;
|
|
this._executionContextPromise = new Promise(x => this._executionContextCallback = x);
|
|
this._client.once('Runtime.executionContextCreated', async event => {
|
|
// Get a reference to the main object on the worker as a hack around https://crbug.com/846099.
|
|
await this._client.send('Runtime.evaluate', {
|
|
expression: 'this',
|
|
returnByValue: false,
|
|
contextId: event.executionContextId
|
|
}).catch(debugError);
|
|
const jsHandleFactory = remoteObject => new JSHandle(executionContext, client, remoteObject);
|
|
const executionContext = new ExecutionContext(client, event.context, jsHandleFactory, null);
|
|
this._executionContextCallback(executionContext);
|
|
});
|
|
// This might fail if the target is closed before we recieve all execution contexts.
|
|
this._client.send('Runtime.enable', {}).catch(debugError);
|
|
|
|
this._client.on('Log.entryAdded', logEntryAdded);
|
|
this._client.send('Log.enable', {}).catch(debugError);
|
|
}
|
|
|
|
/**
|
|
* @return {string}
|
|
*/
|
|
url() {
|
|
return this._url;
|
|
}
|
|
|
|
/**
|
|
* @return {!Promise<ExecutionContext>}
|
|
*/
|
|
async executionContext() {
|
|
return this._executionContextPromise;
|
|
}
|
|
}
|
|
|
|
module.exports = Worker;
|
|
helper.tracePublicAPI(Worker);
|