fix(page): respect timeout 0 in page.waitForFunction (#2563)

The in-page task should not set timeout when timeout is 0.

Fixes #2540.
This commit is contained in:
Andrey Lushnikov 2018-05-25 16:45:04 -07:00 committed by GitHub
parent 6a0627a411
commit debfe7e0b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 5 deletions

View File

@ -911,6 +911,7 @@ class WaitTask {
async function waitForPredicatePageFunction(predicateBody, polling, timeout, ...args) {
const predicate = new Function('...args', predicateBody);
let timedOut = false;
if (timeout)
setTimeout(() => timedOut = true, timeout);
if (polling === 'raf')
return await pollRaf();

View File

@ -159,10 +159,13 @@ module.exports.addTests = function({testRunner, expect}) {
expect(error.message).toContain('waiting for function failed: timeout');
});
it('should disable timeout when its set to 0', async({page}) => {
let error = null;
const res = await page.waitForFunction(() => new Promise(res => setTimeout(() => res(42), 100)), {timeout: 0}).catch(e => error = e);
expect(error).toBe(null);
expect(await res.jsonValue()).toBe(42);
const watchdog = page.waitForFunction(() => {
window.__counter = (window.__counter || 0) + 1;
return window.__injected;
}, {timeout: 0, polling: 10});
await page.waitForFunction(() => window.__counter > 10);
await page.evaluate(() => window.__injected = true);
await watchdog;
});
});