Restore changes to EntryPointSimulations

eth-infinitism · Dec 25, 2024 · 745b7a0 · 745b7a0
1 parent 1b28398
commit 745b7a0
Showing 1 changed file with 30 additions and 13 deletions.
diff --git a/test/entrypointsimulations.test.ts b/test/entrypointsimulations.test.ts
@@ -276,20 +276,37 @@ describe('EntryPointSimulations', function () {
         }, owner, entryPoint)
       }
 
-      await simulateValidation(packUserOp(userOp), entryPoint.address)
-        .catch(e => { throw new Error(decodeRevertReason(e)!) })
-    })
-    it('should revert with AA2x if verificationGasLimit is low', async function () {
-      expect(await simulateValidation(packUserOp(await userOpWithGas(vgl - 1, pmVgl)), entryPoint.address)
-        .catch(decodeRevertReason))
-        .to.match(/AA26/)
-    })
-    it('should revert with AA3x if paymasterVerificationGasLimit is low', async function () {
-      expect(await simulateValidation(packUserOp(await userOpWithGas(vgl, pmVgl - 1)), entryPoint.address)
-        .catch(decodeRevertReason))
-        .to.match(/AA36/)
+      before(async () => {
+        owner = createAccountOwner()
+        paymaster = await new TestPaymasterWithPostOp__factory(ethersSigner).deploy(entryPoint.address)
+        await entryPoint.depositTo(paymaster.address, { value: parseEther('1') })
+        const { proxy: account } = await createAccount(ethersSigner, owner.address, entryPoint.address)
+        sender = account.address
+        await fund(account)
+        pmVgl = withPaymaster === 'without' ? 0 : await findSimulationUserOpWithMin(async n => userOpWithGas(1e6, n), entryPoint, 1, 500000)
+        vgl = await findSimulationUserOpWithMin(async n => userOpWithGas(n, pmVgl), entryPoint, 3000, 500000)
+
+        const userOp = await userOpWithGas(vgl, pmVgl)
+
+        await simulateValidation(packUserOp(userOp), entryPoint.address)
+          .catch(e => {
+            throw new Error(decodeRevertReason(e)!)
+          })
+      })
+      it('should revert with AA2x if verificationGasLimit is low', async function () {
+        expect(await simulateValidation(packUserOp(await userOpWithGas(vgl - 1, pmVgl)), entryPoint.address)
+          .catch(decodeRevertReason))
+          .to.match(/AA26/)
+      })
+      if (withPaymaster === 'with') {
+        it('should revert with AA3x if paymasterVerificationGasLimit is low', async function () {
+          expect(await simulateValidation(packUserOp(await userOpWithGas(vgl, pmVgl - 1)), entryPoint.address)
+            .catch(decodeRevertReason))
+            .to.match(/AA36/)
+        })
+      }
     })
-  })
+  }
 
   describe('#simulateHandleOp', () => {
     it('should simulate creation', async () => {