polish test

AcalaNetwork · Nov 28, 2023 · 5202fd8 · 5202fd8
1 parent 2c920d3
commit 5202fd8
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/src/__tests__/testUtils.ts b/src/__tests__/testUtils.ts
@@ -54,17 +54,18 @@ export const mockXcmToRouter = async (
 ) => {
   const token = ERC20__factory.connect(tokenAddr, signer);
 
-  expect((await token.balanceOf(routerAddr)).toNumber(), 'router already has balance!').to.eq(0);
-
   const decimals = await token.decimals();
-
   const routeAmount = parseUnits(String(amount), decimals);
-  if ((await token.balanceOf(signer.address)).lt(routeAmount)) {
-    throw new Error(`signer ${signer.address} has no enough token [${tokenAddr}] to transfer!`);
-  }
-  await (await token.transfer(routerAddr, routeAmount)).wait();
 
-  expect((await token.balanceOf(routerAddr)).toBigInt()).to.eq(routeAmount.toBigInt());
+  const routerBal = await token.balanceOf(routerAddr);
+  if (routerBal.gt(0)) {
+    expect(routerBal.toBigInt()).to.eq(routeAmount.toBigInt());
+  } else {
+    if ((await token.balanceOf(signer.address)).lt(routeAmount)) {
+      throw new Error(`signer ${signer.address} has no enough token [${tokenAddr}] to transfer!`);
+    }
+    await (await token.transfer(routerAddr, routeAmount)).wait();
+  }
 };
 
 export const expectError = (err: any, msg: any, code: number) => {