test: use emittedUntil in PDF Viewer tests (#27248)

This commit is contained in:
Milan Burda 2021-01-11 02:40:44 +01:00 committed by GitHub
parent 8c305b4113
commit 36f4ee87df
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,6 +1,6 @@
import { expect } from 'chai'; import { expect } from 'chai';
import { BrowserWindow, WebContents, session, ipcMain, app, protocol, webContents } from 'electron/main'; import { BrowserWindow, WebContents, session, ipcMain, app, protocol, webContents } from 'electron/main';
import { emittedOnce } from './events-helpers'; import { emittedOnce, emittedUntil } from './events-helpers';
import { closeAllWindows } from './window-helpers'; import { closeAllWindows } from './window-helpers';
import * as https from 'https'; import * as https from 'https';
import * as http from 'http'; import * as http from 'http';
@ -1283,15 +1283,17 @@ describe('chromium features', () => {
it('opens when loading a pdf resource as top level navigation', async () => { it('opens when loading a pdf resource as top level navigation', async () => {
const w = new BrowserWindow({ show: false }); const w = new BrowserWindow({ show: false });
w.loadURL(pdfSource); w.loadURL(pdfSource);
const [, contents] = await emittedOnce(app, 'web-contents-created'); await emittedUntil(app, 'web-contents-created', (event: Electron.Event, contents: WebContents) => {
expect(contents.getURL()).to.equal('chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/index.html'); return contents.getURL() === 'chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/index.html';
});
}); });
it('opens when loading a pdf resource in a iframe', async () => { it('opens when loading a pdf resource in a iframe', async () => {
const w = new BrowserWindow({ show: false }); const w = new BrowserWindow({ show: false });
w.loadFile(path.join(__dirname, 'fixtures', 'pages', 'pdf-in-iframe.html')); w.loadFile(path.join(__dirname, 'fixtures', 'pages', 'pdf-in-iframe.html'));
const [, contents] = await emittedOnce(app, 'web-contents-created'); await emittedUntil(app, 'web-contents-created', (event: Electron.Event, contents: WebContents) => {
expect(contents.getURL()).to.equal('chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/index.html'); return contents.getURL() === 'chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/index.html';
});
}); });
}); });