fix: smart contract account test

MetaMask · Feb 26, 2024 · a954fbf · a954fbf
1 parent 684481e
commit a954fbf
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/packages/user-operation-controller/src/helpers/SnapSmartContractAccount.test.ts b/packages/user-operation-controller/src/helpers/SnapSmartContractAccount.test.ts
@@ -61,6 +61,9 @@ const PATCH_USER_OPERATION_RESPONSE_MOCK: Awaited<
   ReturnType<KeyringController['patchUserOperation']>
 > = {
   paymasterAndData: '0x123',
+  callGasLimit: '0x444',
+  verificationGasLimit: '0x555',
+  preVerificationGas: '0x667',
 };
 
 const SIGN_USER_OPERATION_RESPONSE_MOCK: Awaited<
@@ -222,6 +225,11 @@ describe('SnapSmartContractAccount', () => {
 
       expect(response).toStrictEqual<UpdateUserOperationResponse>({
         paymasterAndData: PATCH_USER_OPERATION_RESPONSE_MOCK.paymasterAndData,
+        callGasLimit: PATCH_USER_OPERATION_RESPONSE_MOCK.callGasLimit,
+        preVerificationGas:
+          PATCH_USER_OPERATION_RESPONSE_MOCK.preVerificationGas,
+        verificationGasLimit:
+          PATCH_USER_OPERATION_RESPONSE_MOCK.verificationGasLimit,
       });
 
       expect(patchMock).toHaveBeenCalledTimes(1);
@@ -244,6 +252,9 @@ describe('SnapSmartContractAccount', () => {
 
       expect(response).toStrictEqual<UpdateUserOperationResponse>({
         paymasterAndData: undefined,
+        callGasLimit: undefined,
+        preVerificationGas: undefined,
+        verificationGasLimit: undefined,
       });
     });
   });