Merge pull request #101 from 1inch/feature/fix-trackReceivedTokenAndTx

[SC-1004] Save context for getBalance in trackReceivedTokenAndTx
1inch · Nov 21, 2023 · 71f36d7 · 71f36d7
2 parents 2b15ce7 + 012d735
commit 71f36d7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@1inch/solidity-utils",
-  "version": "3.5.0",
+  "version": "3.5.1",
   "main": "dist/src/index.js",
   "types": "dist/src/index.d.ts",
   "exports": {

diff --git a/src/utils.ts b/src/utils.ts
@@ -103,7 +103,7 @@ export async function trackReceivedTokenAndTx<T extends unknown[]>(
 ) : Promise<[bigint, ContractTransactionReceipt]> {
     const tokenAddress = 'address' in token ? token.address : await token.getAddress();
     const isETH = tokenAddress === constants.ZERO_ADDRESS || tokenAddress === constants.EEE_ADDRESS;
-    const getBalance = 'balanceOf' in token ? token.balanceOf : provider.getBalance;
+    const getBalance = 'balanceOf' in token ? token.balanceOf.bind(token) : provider.getBalance.bind(provider);
 
     const preBalance: bigint = await getBalance(wallet);
     const txResult = await (await txPromise(...args)).wait();