59 lines
1.7 KiB
JavaScript
59 lines
1.7 KiB
JavaScript
const { ok } = require('assert');
|
|
const { createHash } = require('crypto');
|
|
const puppeteer = require("puppeteer-core");
|
|
const chromium = require('@sparticuz/chromium');
|
|
|
|
exports.handler = async (event, context) => {
|
|
let browser = null;
|
|
|
|
try {
|
|
const browser = await puppeteer.launch({
|
|
args: chromium.args,
|
|
defaultViewport: chromium.defaultViewport,
|
|
executablePath: await chromium.executablePath,
|
|
headless: chromium.headless,
|
|
ignoreHTTPSErrors: true,
|
|
});
|
|
|
|
const contexts = [
|
|
browser.defaultBrowserContext(),
|
|
];
|
|
|
|
while (contexts.length < event.length) {
|
|
contexts.push(await browser.createIncognitoBrowserContext());
|
|
}
|
|
|
|
for (let context of contexts) {
|
|
const job = event.shift();
|
|
const page = await context.newPage();
|
|
|
|
if (job.hasOwnProperty('url') === true) {
|
|
await page.goto(job.url, { waitUntil: ['domcontentloaded', 'load'] });
|
|
|
|
if (job.hasOwnProperty('expected') === true) {
|
|
if (job.expected.hasOwnProperty('title') === true) {
|
|
ok(await page.title() === job.expected.title, `Title assertion failed.`);
|
|
}
|
|
|
|
if (job.expected.hasOwnProperty('screenshot') === true) {
|
|
if (job.expected.hasOwnProperty('remove') === true ) {
|
|
await page.evaluate((selector) => {
|
|
document.getElementById(selector).remove();
|
|
}, job.expected.remove);
|
|
}
|
|
ok(createHash('sha1').update((await page.screenshot()).toString('base64')).digest('hex') === job.expected.screenshot, `Screenshot assertion failed.`);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} catch (error) {
|
|
throw error.message;
|
|
} finally {
|
|
if (browser !== null) {
|
|
await browser.close();
|
|
}
|
|
}
|
|
|
|
return true;
|
|
};
|