88 lines
2.2 KiB
JavaScript
88 lines
2.2 KiB
JavaScript
/* Any copyright is dedicated to the Public Domain.
|
|
* http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
const { Async } = ChromeUtils.importESModule(
|
|
"resource://services-common/async.sys.mjs"
|
|
);
|
|
const { sinon } = ChromeUtils.importESModule(
|
|
"resource://testing-common/Sinon.sys.mjs"
|
|
);
|
|
|
|
function makeArray(length) {
|
|
// Start at 1 so that we can just divide by yieldEvery to get the expected
|
|
// call count. (we exp)
|
|
return Array.from({ length }, (v, i) => i + 1);
|
|
}
|
|
|
|
// Adjust if we ever change the default.
|
|
const DEFAULT_YIELD_EVERY = 50;
|
|
|
|
add_task(async function testYields() {
|
|
let spy = sinon.spy(Async, "promiseYield");
|
|
try {
|
|
await Async.yieldingForEach(makeArray(DEFAULT_YIELD_EVERY * 2), element => {
|
|
// The yield will happen *after* this function is ran.
|
|
Assert.equal(
|
|
spy.callCount,
|
|
Math.floor((element - 1) / DEFAULT_YIELD_EVERY)
|
|
);
|
|
});
|
|
} finally {
|
|
spy.restore();
|
|
}
|
|
});
|
|
|
|
add_task(async function testExistingYieldState() {
|
|
const yieldState = Async.yieldState(DEFAULT_YIELD_EVERY);
|
|
|
|
for (let i = 0; i < 15; i++) {
|
|
Assert.equal(yieldState.shouldYield(), false);
|
|
}
|
|
|
|
let spy = sinon.spy(Async, "promiseYield");
|
|
|
|
try {
|
|
await Async.yieldingForEach(
|
|
makeArray(DEFAULT_YIELD_EVERY * 2),
|
|
element => {
|
|
Assert.equal(
|
|
spy.callCount,
|
|
Math.floor((element + 15 - 1) / DEFAULT_YIELD_EVERY)
|
|
);
|
|
},
|
|
yieldState
|
|
);
|
|
} finally {
|
|
spy.restore();
|
|
}
|
|
});
|
|
|
|
add_task(async function testEarlyReturn() {
|
|
let lastElement = 0;
|
|
await Async.yieldingForEach(makeArray(DEFAULT_YIELD_EVERY), element => {
|
|
lastElement = element;
|
|
return element === 10;
|
|
});
|
|
|
|
Assert.equal(lastElement, 10);
|
|
});
|
|
|
|
add_task(async function testEaryReturnAsync() {
|
|
let lastElement = 0;
|
|
await Async.yieldingForEach(makeArray(DEFAULT_YIELD_EVERY), async element => {
|
|
lastElement = element;
|
|
return element === 10;
|
|
});
|
|
|
|
Assert.equal(lastElement, 10);
|
|
});
|
|
|
|
add_task(async function testEarlyReturnPromise() {
|
|
let lastElement = 0;
|
|
await Async.yieldingForEach(makeArray(DEFAULT_YIELD_EVERY), element => {
|
|
lastElement = element;
|
|
return new Promise(resolve => resolve(element === 10));
|
|
});
|
|
|
|
Assert.equal(lastElement, 10);
|
|
});
|