mirrors/electron
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions 1

test: load page with some content when testing extension (#23901)

Browse source 
* test: load page with some content when testing extension

* test: use dom-ready to wait for page loaded with extension
This commit is contained in:
Cheng Zhao 2020-06-03 16:56:28 +09:00 • committed by GitHub
parent d8594f7a78
commit 5de6adfb84
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 17 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

24
spec-main/extensions-spec.ts
View file

@ -10,11 +10,13 @@ import { emittedOnce, emittedNTimes } from './events-helpers';
const fixtures = path.join(__dirname, 'fixtures'); const fixtures = path.join(__dirname, 'fixtures');
describe('chrome extensions', () => { describe('chrome extensions', () => {
const emptyPage = '<script>console.log("loaded")</script>';
// NB. extensions are only allowed on http://, https:// and ftp:// (!) urls by default. // NB. extensions are only allowed on http://, https:// and ftp:// (!) urls by default.
let server: http.Server; let server: http.Server;
let url: string; let url: string;
before(async () => { before(async () => {
server = http.createServer((req, res) => res.end()); server = http.createServer((req, res) => res.end(emptyPage));
await new Promise(resolve => server.listen(0, '127.0.0.1', () => { await new Promise(resolve => server.listen(0, '127.0.0.1', () => {
url = `http://127.0.0.1:${(server.address() as AddressInfo).port}`; url = `http://127.0.0.1:${(server.address() as AddressInfo).port}`;
resolve(); resolve();
@ -38,7 +40,8 @@ describe('chrome extensions', () => {
const customSession = session.fromPartition(`persist:${require('uuid').v4()}`); const customSession = session.fromPartition(`persist:${require('uuid').v4()}`);
await customSession.loadExtension(path.join(fixtures, 'extensions', 'red-bg')); await customSession.loadExtension(path.join(fixtures, 'extensions', 'red-bg'));
const w = new BrowserWindow({ show: false, webPreferences: { session: customSession } }); const w = new BrowserWindow({ show: false, webPreferences: { session: customSession } });
await w.loadURL(url); w.loadURL(url);
await emittedOnce(w.webContents, 'dom-ready');
const bg = await w.webContents.executeJavaScript('document.documentElement.style.backgroundColor'); const bg = await w.webContents.executeJavaScript('document.documentElement.style.backgroundColor');
expect(bg).to.equal('red'); expect(bg).to.equal('red');
}); });
@ -92,7 +95,8 @@ describe('chrome extensions', () => {
const customSession = session.fromPartition(`persist:${require('uuid').v4()}`); const customSession = session.fromPartition(`persist:${require('uuid').v4()}`);
await customSession.loadExtension(path.join(fixtures, 'extensions', 'red-bg')); await customSession.loadExtension(path.join(fixtures, 'extensions', 'red-bg'));
const w = new BrowserWindow({ show: false }); // not in the session const w = new BrowserWindow({ show: false }); // not in the session
await w.loadURL(url); w.loadURL(url);
await emittedOnce(w.webContents, 'dom-ready');
const bg = await w.webContents.executeJavaScript('document.documentElement.style.backgroundColor'); const bg = await w.webContents.executeJavaScript('document.documentElement.style.backgroundColor');
expect(bg).to.equal(''); expect(bg).to.equal('');
}); });
@ -115,7 +119,8 @@ describe('chrome extensions', () => {
const customSession = session.fromPartition(`persist:${require('uuid').v4()}`); const customSession = session.fromPartition(`persist:${require('uuid').v4()}`);
extension = await customSession.loadExtension(path.join(fixtures, 'extensions', 'chrome-i18n')); extension = await customSession.loadExtension(path.join(fixtures, 'extensions', 'chrome-i18n'));
w = new BrowserWindow({ show: false, webPreferences: { session: customSession, nodeIntegration: true } }); w = new BrowserWindow({ show: false, webPreferences: { session: customSession, nodeIntegration: true } });
await w.loadURL(url); w.loadURL(url);
await emittedOnce(w.webContents, 'dom-ready');
}); });
it('getAcceptLanguages()', async () => { it('getAcceptLanguages()', async () => {
const result = await exec('getAcceptLanguages'); const result = await exec('getAcceptLanguages');
@ -132,10 +137,11 @@ describe('chrome extensions', () => {
let content: any; let content: any;
before(async () => { before(async () => {
const customSession = session.fromPartition(`persist:${require('uuid').v4()}`); const customSession = session.fromPartition(`persist:${require('uuid').v4()}`);
customSession.loadExtension(path.join(fixtures, 'extensions', 'chrome-runtime')); await customSession.loadExtension(path.join(fixtures, 'extensions', 'chrome-runtime'));
const w = new BrowserWindow({ show: false, webPreferences: { session: customSession } }); const w = new BrowserWindow({ show: false, webPreferences: { session: customSession } });
try { try {
await w.loadURL(url); w.loadURL(url);
await emittedOnce(w.webContents, 'dom-ready');
content = JSON.parse(await w.webContents.executeJavaScript('document.documentElement.textContent')); content = JSON.parse(await w.webContents.executeJavaScript('document.documentElement.textContent'));
expect(content).to.be.an('object'); expect(content).to.be.an('object');
} finally { } finally {
@ -484,7 +490,8 @@ describe('chrome extensions', () => {
it('loads a ui page of an extension', async () => { it('loads a ui page of an extension', async () => {
const { id } = await session.defaultSession.loadExtension(path.join(fixtures, 'extensions', 'ui-page')); const { id } = await session.defaultSession.loadExtension(path.join(fixtures, 'extensions', 'ui-page'));
const w = new BrowserWindow({ show: false }); const w = new BrowserWindow({ show: false });
await w.loadURL(`chrome-extension://${id}/bare-page.html`); w.loadURL(`chrome-extension://${id}/bare-page.html`);
await emittedOnce(w.webContents, 'dom-ready');
const textContent = await w.webContents.executeJavaScript('document.body.textContent'); const textContent = await w.webContents.executeJavaScript('document.body.textContent');
expect(textContent).to.equal('ui page loaded ok\n'); expect(textContent).to.equal('ui page loaded ok\n');
}); });
@ -492,7 +499,8 @@ describe('chrome extensions', () => {
it('can load resources', async () => { it('can load resources', async () => {
const { id } = await session.defaultSession.loadExtension(path.join(fixtures, 'extensions', 'ui-page')); const { id } = await session.defaultSession.loadExtension(path.join(fixtures, 'extensions', 'ui-page'));
const w = new BrowserWindow({ show: false }); const w = new BrowserWindow({ show: false });
await w.loadURL(`chrome-extension://${id}/page-script-load.html`); w.loadURL(`chrome-extension://${id}/page-script-load.html`);
await emittedOnce(w.webContents, 'dom-ready');
const textContent = await w.webContents.executeJavaScript('document.body.textContent'); const textContent = await w.webContents.executeJavaScript('document.body.textContent');
expect(textContent).to.equal('script loaded ok\n'); expect(textContent).to.equal('script loaded ok\n');
}); });

2
spec-main/fixtures/extensions/chrome-runtime/manifest.json
View file

@ -5,7 +5,7 @@
{ {
"matches": ["<all_urls>"], "matches": ["<all_urls>"],
"js": ["main.js"], "js": ["main.js"],
"run_at": "document_start" "run_at": "document_end"
} }
], ],
"manifest_version": 2 "manifest_version": 2