chore: fix integration tests.

logion-network · May 7, 2024 · 503726f · 503726f
1 parent f02b626
commit 503726f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/packages/node-api/integration/Balances.ts b/packages/node-api/integration/Balances.ts
@@ -4,9 +4,10 @@ import { setup, signAndSend } from "./Util.js";
 export async function transferTokens() {
     const { alice, requester, api } = await setup();
 
+    const transferred = Lgnt.from(20000n);
     const transferExtrinsic = api.polkadot.tx.balances.transferAllowDeath(
         requester.address,
-        Lgnt.from(20000n).canonical
+        transferred.canonical,
     );
     const result = await signAndSend(alice, transferExtrinsic);
 
@@ -17,8 +18,6 @@ export async function transferTokens() {
     expect(data?.from as string).toBe(alice.address);
     expect(data?.to as string).toBe(requester.address);
 
-    const balances = await api.queries.getCoinBalances(ValidAccountId.polkadot(requester.address));
-    const logionTokenBalance = balances.find(balance => balance.coin.id === 'lgnt');
-    expect(logionTokenBalance?.available.coefficient.toNumber()).toBe(20);
-    expect(logionTokenBalance?.available.prefix.tenExponent).toBe(3);
+    const balance = await api.queries.getAccountData(ValidAccountId.polkadot(requester.address));
+    expect(balance?.available).toEqual(transferred);
 }
diff --git a/packages/node-api/integration/Fees.ts b/packages/node-api/integration/Fees.ts
@@ -33,10 +33,10 @@ export async function ensureEnoughFunds() {
     const accountData = await api.queries.getAccountData(aliceAccountId);
     const existentialDeposit = api.polkadot.consts.balances.existentialDeposit.toBigInt();
 
-    await expectAsync(testEnsureEnoughFunds(accountData.available)).toBeRejectedWithError("Not enough funds");
-    await expectAsync(testEnsureEnoughFunds(accountData.available + 1000n)).toBeRejectedWithError("Not enough funds");
-    await expectAsync(testEnsureEnoughFunds(accountData.available - existentialDeposit)).toBeResolved();
-    await expectAsync(testEnsureEnoughFunds(accountData.available - (10n * existentialDeposit))).toBeResolved();
+    await expectAsync(testEnsureEnoughFunds(accountData.available.canonical)).toBeRejectedWithError("Not enough funds");
+    await expectAsync(testEnsureEnoughFunds(accountData.available.canonical + 1000n)).toBeRejectedWithError("Not enough funds");
+    await expectAsync(testEnsureEnoughFunds(accountData.available.canonical - existentialDeposit)).toBeResolved();
+    await expectAsync(testEnsureEnoughFunds(accountData.available.canonical - (10n * existentialDeposit))).toBeResolved();
 }
 
 async function testEnsureEnoughFunds(fees: bigint) {