Fix integration tests

multiversx · Dec 9, 2024 · cbe8b53 · cbe8b53
1 parent e8aa3e2
commit cbe8b53
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/src/testutils/wallets.ts b/src/testutils/wallets.ts
@@ -116,4 +116,9 @@ export class TestWallet {
         this.account.nonce = accountOnNetwork.nonce;
         return this;
     }
+
+    async getBalance(provider: IAccountFetcher) {
+        let accountOnNetwork = await provider.getAccount(this.address);
+        return accountOnNetwork.balance;
+    }
 }
diff --git a/src/transaction.local.net.spec.ts b/src/transaction.local.net.spec.ts
@@ -55,7 +55,7 @@ describe("test transaction", function () {
 
         await alice.sync(provider);
         await bob.sync(provider);
-        let initialBalanceOfBob = new BigNumber(bob.account.balance.toString());
+        let initialBalanceOfBob = new BigNumber((await bob.getBalance(provider)).toString());
 
         transactionOne.setNonce(alice.account.nonce);
         alice.account.incrementNonce();
@@ -71,7 +71,7 @@ describe("test transaction", function () {
         await watcher.awaitCompleted(transactionTwo.getHash().hex());
 
         await bob.sync(provider);
-        let newBalanceOfBob = new BigNumber(bob.account.balance.toString());
+        let newBalanceOfBob = new BigNumber((await bob.getBalance(provider)).toString());
 
         assert.deepEqual(TokenTransfer.egldFromAmount(85).valueOf(), newBalanceOfBob.minus(initialBalanceOfBob));
     });
@@ -94,15 +94,15 @@ describe("test transaction", function () {
 
         await alice.sync(provider);
         await bob.sync(provider);
-        let initialBalanceOfBob = new BigNumber(bob.account.balance.toString());
+        let initialBalanceOfBob = new BigNumber((await bob.getBalance(provider)).toString());
 
         transactionOne.setNonce(alice.account.nonce);
         await signTransaction({ transaction: transactionOne, wallet: alice });
         await provider.sendTransaction(transactionOne);
         await watcher.awaitCompleted(transactionOne.getHash().hex());
 
         await bob.sync(provider);
-        let newBalanceOfBob = new BigNumber(bob.account.balance.toString());
+        let newBalanceOfBob = new BigNumber((await bob.getBalance(provider)).toString());
 
         assert.deepEqual(TokenTransfer.egldFromAmount(42).valueOf(), newBalanceOfBob.minus(initialBalanceOfBob));
     });
@@ -155,7 +155,7 @@ describe("test transaction", function () {
 
         await alice.sync(provider);
         await bob.sync(provider);
-        const initialBalanceOfBob = new BigNumber(bob.account.balance.toString());
+        const initialBalanceOfBob = new BigNumber((await bob.getBalance(provider)).toString());
 
         const transaction = factory.createTransactionForNativeTokenTransfer(alice.address, {
             receiver: bob.address,
@@ -170,7 +170,7 @@ describe("test transaction", function () {
         await watcher.awaitCompleted(txHash);
 
         await bob.sync(provider);
-        const newBalanceOfBob = new BigNumber(bob.account.balance.toString());
+        const newBalanceOfBob = new BigNumber((await bob.getBalance(provider)).toString());
 
         assert.deepEqual(TokenTransfer.egldFromAmount(42).valueOf(), newBalanceOfBob.minus(initialBalanceOfBob));
     });