feat: l2 estimate gas

lidofinance · Nov 25, 2024 · 589ead5 · 589ead5
1 parent 50bec8c
commit 589ead5
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/packages/sdk/src/l2/__test__/l2.test.ts b/packages/sdk/src/l2/__test__/l2.test.ts
@@ -21,6 +21,7 @@ import { expectContract } from '../../../tests/utils/expect/expect-contract.js';
 import {
   expectAlmostEqualBn,
   expectNonNegativeBn,
+  expectPositiveBn,
 } from '../../../tests/utils/expect/expect-bn.js';
 import {
   SPENDING_TIMEOUT,
@@ -147,6 +148,11 @@ describe('LidoSDKL2 wrap', () => {
     expectAddress(tx.address, stethAddress);
   });
 
+  testSpending('wrap estimate gas', async () => {
+    const gas = await l2.wrapWstethToStethEstimateGas({ value });
+    expectPositiveBn(gas);
+  });
+
   testSpending('wrap wsteth to steth', async () => {
     const stethValue = await l2.steth.convertToSteth(value);
     const stethBalanceBefore = await l2.steth.balance(account.address);
@@ -193,6 +199,11 @@ describe('LidoSDKL2 wrap', () => {
     expectAddress(tx.address, stethAddress);
   });
 
+  testSpending('unwrap estimate gas', async () => {
+    const gas = await l2.unwrapStethEstimateGas({ value });
+    expectPositiveBn(gas);
+  });
+
   testSpending('unwrap', async () => {
     const stethValue = await l2.steth.convertToSteth(value);
     const stethBalanceBefore = await l2.steth.balance(account.address);

diff --git a/packages/sdk/src/l2/l2.ts b/packages/sdk/src/l2/l2.ts
@@ -141,6 +141,18 @@ export class LidoSDKL2 extends LidoSDKModule {
     };
   }
 
+  @Logger('Utils:')
+  public async wrapWstethToStethEstimateGas(
+    props: WrapPropsWithoutCallback,
+  ): Promise<bigint> {
+    const { value, account } = await this.parseProps(props);
+    const contract = await this.getContract();
+
+    return contract.estimateGas.wrap([value], {
+      account,
+    });
+  }
+
   @Logger('Utils:')
   private async wrapWstethToStethParseEvents(
     receipt: TransactionReceipt,
@@ -251,6 +263,19 @@ export class LidoSDKL2 extends LidoSDKModule {
     return request;
   }
 
+  @Logger('Utils:')
+  @ErrorHandler()
+  public async unwrapStethEstimateGas(
+    props: Omit<WrapProps, 'callback'>,
+  ): Promise<bigint> {
+    const { value, account } = await this.parseProps(props);
+    const contract = await this.getContract();
+
+    return contract.estimateGas.unwrap([value], {
+      account,
+    });
+  }
+
   @Logger('Utils:')
   private async unwrapParseEvents(
     receipt: TransactionReceipt,