AA-21: gas usage tests for different batch scenarios (eth-infinitism#93)

use with "yarn gas-checker"
currently deploys batches of 1,2,20,21, and display the gas diff

Author: drortirosh

gascalc/0-init-gas-checker.ts

@@ -0,0 +1,12 @@
+import { GasCheckCollector, GasChecker } from './GasChecker'
+
+describe('gas calculations', function () {
+  this.timeout(60000)
+  const g = new GasChecker()
+
+  it('warmup', async function () {
+    await GasCheckCollector.init()
+    // dummy run - first run is slower.
+    await g.runTest({ title: 'simple', count: 1, diffLastGas: false })
+  })
+})

gascalc/1-simple-wallet.gas.ts

@@ -0,0 +1,17 @@
+import { GasChecker } from './GasChecker'
+
+context('simple wallet', function () {
+  this.timeout(60000)
+  const g = new GasChecker()
+
+  it('simple 1', async function () {
+    await g.addTestRow({ title: 'simple', count: 1, diffLastGas: false })
+    await g.addTestRow({ title: 'simple - diff from previous', count: 2, diffLastGas: true })
+  })
+
+  it('simple 20', async function () {
+    if (g.skipLong()) this.skip()
+    await g.addTestRow({ title: 'simple', count: 20, diffLastGas: false })
+    await g.addTestRow({ title: 'simple - diff from previous', count: 21, diffLastGas: true })
+  })
+})

gascalc/2-paymaster.gas.ts

@@ -0,0 +1,40 @@
+import { parseEther } from 'ethers/lib/utils'
+import { TestPaymasterAcceptAll__factory } from '../typechain'
+import { ethers } from 'hardhat'
+import { GasChecker } from './GasChecker'
+
+const ethersSigner = ethers.provider.getSigner()
+
+context('Minimal Paymaster', function () {
+  this.timeout(60000)
+  const g = new GasChecker()
+
+  let paymasterAddress: string
+  before(async () => {
+    const paymaster = await new TestPaymasterAcceptAll__factory(ethersSigner).deploy(g.entryPoint().address)
+    paymasterAddress = paymaster.address
+    await paymaster.addStake(0, { value: 1 })
+    await g.entryPoint().depositTo(paymaster.address, { value: parseEther('10') })
+  })
+  it('simple paymaster', async function () {
+    await g.addTestRow({ title: 'simple paymaster', count: 1, paymaster: paymasterAddress, diffLastGas: false })
+    await g.addTestRow({
+      title: 'simple paymaster with diff',
+      count: 2,
+      paymaster: paymasterAddress,
+      diffLastGas: true
+    })
+  })
+
+  it('simple paymaster 20', async function () {
+    if (g.skipLong()) this.skip()
+
+    await g.addTestRow({ title: 'simple paymaster', count: 20, paymaster: paymasterAddress, diffLastGas: false })
+    await g.addTestRow({
+      title: 'simple paymaster with diff',
+      count: 21,
+      paymaster: paymasterAddress,
+      diffLastGas: true
+    })
+  })
+})

gascalc/3-huge-tx-gas.ts

@@ -0,0 +1,17 @@
+import { DefaultGasTestInfo, GasChecker } from './GasChecker'
+
+context('huge tx', function () {
+  this.timeout(60000)
+  const huge = DefaultGasTestInfo.destCallData!.padEnd(20480, 'f')
+  const g = new GasChecker()
+
+  it('big tx', async () => {
+    await g.addTestRow({ title: 'big tx 10k', count: 1, destCallData: huge, diffLastGas: false })
+    await g.addTestRow({ title: 'big tx - diff from previous', count: 2, destCallData: huge, diffLastGas: true })
+  })
+  it('big tx 20', async function () {
+    if (g.skipLong()) this.skip()
+    await g.addTestRow({ title: 'big tx', count: 20, destCallData: huge, diffLastGas: false })
+    await g.addTestRow({ title: 'big tx - diff from previous', count: 21, destCallData: huge, diffLastGas: true })
+  })
+})