fix: missing test check

EKOKE-DAO · Jan 22, 2024 · 835083a · 835083a
1 parent f8a60f1
commit 835083a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/ethereum/fly/test/Fly.ts b/ethereum/fly/test/Fly.ts
@@ -76,6 +76,9 @@ describe("Fly", () => {
     const fee = await token.swapFee();
 
     await token.setFlyCanisterAddress(flyCanister.address);
+    const initialFlyCanisterBalance = await ethers.provider.getBalance(
+      flyCanister.address
+    );
 
     const initialBalance = await ethers.provider.getBalance(owner.address);
 
@@ -93,8 +96,13 @@ describe("Fly", () => {
     // check owner has paid FEE ethers
     const finalBalance =
       initialBalance - (await ethers.provider.getBalance(owner.address));
-
     expect(finalBalance).to.greaterThan(fee);
+
+    // check fly canister has received the swap fee
+    const expectedFlyCanisterBalance = initialFlyCanisterBalance + fee;
+    expect(await ethers.provider.getBalance(flyCanister.address)).to.equal(
+      expectedFlyCanisterBalance
+    );
   });
 
   it("should fail swap if fly canister address is not set", async () => {