You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.1 KiB
38 lines
1.1 KiB
const { network } = require('hardhat');
|
|
const { promisify } = require('util');
|
|
|
|
const queue = promisify(setImmediate);
|
|
|
|
async function countPendingTransactions() {
|
|
return parseInt(await network.provider.send('eth_getBlockTransactionCountByNumber', ['pending']));
|
|
}
|
|
|
|
async function batchInBlock(txs) {
|
|
try {
|
|
// disable auto-mining
|
|
await network.provider.send('evm_setAutomine', [false]);
|
|
// send all transactions
|
|
const promises = txs.map(fn => fn());
|
|
// wait for node to have all pending transactions
|
|
while (txs.length > (await countPendingTransactions())) {
|
|
await queue();
|
|
}
|
|
// mine one block
|
|
await network.provider.send('evm_mine');
|
|
// fetch receipts
|
|
const receipts = await Promise.all(promises);
|
|
// Sanity check, all tx should be in the same block
|
|
const minedBlocks = new Set(receipts.map(({ receipt }) => receipt.blockNumber));
|
|
expect(minedBlocks.size).to.equal(1);
|
|
|
|
return receipts;
|
|
} finally {
|
|
// enable auto-mining
|
|
await network.provider.send('evm_setAutomine', [true]);
|
|
}
|
|
}
|
|
|
|
module.exports = {
|
|
countPendingTransactions,
|
|
batchInBlock,
|
|
};
|
|
|