github/puppeteer
0
0
Fork 0
mirror of https://github.com/puppeteer/puppeteer synced 2024-06-14 14:02:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: add test for page.waitForFunction with script CSP (#2305)

Browse Source 
This patch adds a test to fixate page.waitForFunction behavior
for pages with CSP.

References #1229.
This commit is contained in:
Andrey Lushnikov 2018-04-03 15:21:08 -07:00 committed by GitHub
parent 94c32e4bc8
commit f925158733
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
test/frame.spec.js
View File

@ -110,6 +110,13 @@ module.exports.addTests = function({testRunner, expect}) {
await page.evaluate(() => window.__FOO = 'hit');
await watchdog;
});
xit('should work with strict CSP policy', async({page, server}) => {
server.setCSP('/empty.html', 'script-src ' + server.PREFIX);
await page.goto(server.EMPTY_PAGE);
const watchdog = page.waitForFunction(() => window.__FOO === 'hit', {polling: 'raf'});
await page.evaluate(() => window.__FOO = 'hit');
await watchdog;
});
it('should throw on bad polling value', async({page, server}) => {
let error = null;
try {
@ -469,4 +476,4 @@ module.exports.addTests = function({testRunner, expect}) {
expect(page.frames()[2].parentFrame()).toBe(page.mainFrame());
});
});
};
};

13
test/server/SimpleServer.js
View File

@ -78,6 +78,8 @@ class SimpleServer {
this._routes = new Map();
/** @type {!Map<string, !{username:string, password:string}>} */
this._auths = new Map();
/** @type {!Map<string, string>} */
this._csp = new Map();
/** @type {!Map<string, !Promise>} */
this._requestSubscribers = new Map();
}
@ -109,6 +111,14 @@ class SimpleServer {
this._auths.set(path, {username, password});
}
/**
* @param {string} path
* @param {string} csp
*/
setCSP(path, csp) {
this._csp.set(path, csp);
}
async stop() {
this.reset();
for (const socket of this._sockets)
@ -158,6 +168,7 @@ class SimpleServer {
reset() {
this._routes.clear();
this._auths.clear();
this._csp.clear();
const error = new Error('Static Server has been reset');
for (const subscriber of this._requestSubscribers.values())
subscriber[rejectSymbol].call(null, error);
@ -214,6 +225,8 @@ class SimpleServer {
} else {
response.setHeader('Cache-Control', 'no-cache, no-store');
}
if (this._csp.has(pathName))
response.setHeader('Content-Security-Policy', this._csp.get(pathName));
fs.readFile(filePath, function(err, data) {
if (err) {