puppeteer/test/utils.js

118 lines
3.1 KiB
JavaScript
Raw Normal View History

/**
* Copyright 2017 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 utils = module.exports = {
2017-06-21 20:51:06 +00:00
/**
2017-06-21 20:58:49 +00:00
* @param {!Page} page
* @param {string} frameId
* @param {string} url
*/
2017-06-21 20:51:06 +00:00
attachFrame: async function(page, frameId, url) {
await page.evaluate(attachFrame, frameId, url);
2017-06-21 20:51:06 +00:00
function attachFrame(frameId, url) {
const frame = document.createElement('iframe');
2017-06-21 20:51:06 +00:00
frame.src = url;
frame.id = frameId;
document.body.appendChild(frame);
return new Promise(x => frame.onload = x);
}
},
2017-06-21 20:51:06 +00:00
/**
2017-06-21 20:58:49 +00:00
* @param {!Page} page
* @param {string} frameId
*/
2017-06-21 20:51:06 +00:00
detachFrame: async function(page, frameId) {
await page.evaluate(detachFrame, frameId);
2017-06-21 20:51:06 +00:00
function detachFrame(frameId) {
const frame = document.getElementById(frameId);
2017-06-21 20:51:06 +00:00
frame.remove();
}
},
2017-06-21 20:51:06 +00:00
/**
2017-06-21 20:58:49 +00:00
* @param {!Page} page
* @param {string} frameId
* @param {string} url
*/
2017-06-21 20:51:06 +00:00
navigateFrame: async function(page, frameId, url) {
await page.evaluate(navigateFrame, frameId, url);
2017-06-21 20:51:06 +00:00
function navigateFrame(frameId, url) {
const frame = document.getElementById(frameId);
2017-06-21 20:51:06 +00:00
frame.src = url;
return new Promise(x => frame.onload = x);
}
},
2017-06-21 20:51:06 +00:00
/**
2017-06-21 20:58:49 +00:00
* @param {!Frame} frame
* @param {string=} indentation
* @return {string}
*/
2017-06-21 20:51:06 +00:00
dumpFrames: function(frame, indentation) {
indentation = indentation || '';
let result = indentation + frame.url().replace(/:\d{4}\//, ':<PORT>/');
for (const child of frame.childFrames())
2017-06-21 20:51:06 +00:00
result += '\n' + utils.dumpFrames(child, ' ' + indentation);
return result;
},
/**
* @param {!EventEmitter} emitter
* @param {string} eventName
* @return {!Promise<!Object>}
*/
waitEvent: function(emitter, eventName) {
return new Promise(fulfill => emitter.once(eventName, fulfill));
},
/**
* @param {!Buffer} pdfBuffer
* @return {!Promise<!Array<!Object>>}
*/
getPDFPages: async function(pdfBuffer) {
const PDFJS = require('pdfjs-dist');
PDFJS.disableWorker = true;
const data = new Uint8Array(pdfBuffer);
const doc = await PDFJS.getDocument(data);
const pages = [];
for (let i = 0; i < doc.numPages; ++i) {
const page = await doc.getPage(i + 1);
const viewport = page.getViewport(1);
// Viewport width and height is in PDF points, which is
// 1/72 of an inch.
pages.push({
width: viewport.width / 72,
height: viewport.height / 72,
});
page.cleanup();
}
doc.cleanup();
return pages;
},
/**
* @param {number} px
* @return {number}
*/
cssPixelsToInches: function(px) {
return px / 96;
},
};