mirror of
https://github.com/zen-browser/pdf.js.git
synced 2025-07-08 09:20:06 +02:00
Set the event handlers in the integration tests before any event is triggered
The function evaluateOnNewDocument in Puppeteer allow us to execute some js before the pdf.js one is loaded. It allows us to stub some setters before there are used and then set some event handlers very soon.
This commit is contained in:
parent
d8d9cff715
commit
5e4948062c
4 changed files with 81 additions and 70 deletions
|
@ -14,9 +14,10 @@
|
|||
*/
|
||||
|
||||
import os from "os";
|
||||
|
||||
const isMac = os.platform() === "darwin";
|
||||
|
||||
function loadAndWait(filename, selector, zoom, pageSetup, options) {
|
||||
function loadAndWait(filename, selector, zoom, setups, options) {
|
||||
return Promise.all(
|
||||
global.integrationSessions.map(async session => {
|
||||
const page = await session.browser.newPage();
|
||||
|
@ -52,11 +53,52 @@ function loadAndWait(filename, selector, zoom, pageSetup, options) {
|
|||
global.integrationBaseUrl
|
||||
}?file=/test/pdfs/${filename}#zoom=${zoom ?? "page-fit"}${app_options}`;
|
||||
|
||||
await page.goto(url);
|
||||
if (pageSetup) {
|
||||
await pageSetup(page);
|
||||
if (setups) {
|
||||
// page.evaluateOnNewDocument allows us to run code before the
|
||||
// first js script is executed.
|
||||
// The idea here is to set up some setters for PDFViewerApplication
|
||||
// and EventBus, so we can inject some code to do whatever we want
|
||||
// soon enough especially before the first event in the eventBus is
|
||||
// dispatched.
|
||||
const { prePageSetup, appSetup, eventBusSetup } = setups;
|
||||
await prePageSetup?.(page);
|
||||
if (appSetup || eventBusSetup) {
|
||||
await page.evaluateOnNewDocument(
|
||||
(aSetup, eSetup) => {
|
||||
let app;
|
||||
let eventBus;
|
||||
Object.defineProperty(window, "PDFViewerApplication", {
|
||||
get() {
|
||||
return app;
|
||||
},
|
||||
set(newValue) {
|
||||
app = newValue;
|
||||
if (aSetup) {
|
||||
// eslint-disable-next-line no-eval
|
||||
eval(`(${aSetup})`)(app);
|
||||
}
|
||||
Object.defineProperty(app, "eventBus", {
|
||||
get() {
|
||||
return eventBus;
|
||||
},
|
||||
set(newV) {
|
||||
eventBus = newV;
|
||||
// eslint-disable-next-line no-eval
|
||||
eval(`(${eSetup})`)(eventBus);
|
||||
},
|
||||
});
|
||||
},
|
||||
});
|
||||
},
|
||||
appSetup?.toString(),
|
||||
eventBusSetup?.toString()
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
await page.goto(url);
|
||||
await setups?.postPageSetup?.(page);
|
||||
|
||||
await page.bringToFront();
|
||||
if (selector) {
|
||||
await page.waitForSelector(selector, {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue