From 4583803981b6ce5288c045bf58b8bf0e65b476ad Mon Sep 17 00:00:00 2001
From: MikaIsaak <michaelisayan@gmail.com>
Date: Tue, 16 Jul 2024 10:50:25 +0300
Subject: [PATCH] Fix difference

---
 test/erc4626/vault.test.ts | 64 ++++++--------------------------------
 1 file changed, 9 insertions(+), 55 deletions(-)

diff --git a/test/erc4626/vault.test.ts b/test/erc4626/vault.test.ts
index 53ae4bb..e16c741 100644
--- a/test/erc4626/vault.test.ts
+++ b/test/erc4626/vault.test.ts
@@ -267,67 +267,21 @@ describe("Vault", function () {
         });
     });
 
-    // describe("claimAllReward", function () {
-    //     it("Should claim all reward", async function () {
-    //         const { hederaVault, owner, rewardToken } = await deployFixture();
+    describe("claimAllReward", function () {
+        it("Should claim all reward", async function () {
+            const { hederaVault, owner, rewardToken } = await deployFixture();
 
-    //         // await rewardToken.approve(hederaVault.target, 10 * 1e8);
-
-    //         console.log(await hederaVault.calculateReward(0));
-
-    //         const reward = await hederaVault.calculateReward(0);
-
-    //         const tx = await hederaVault.claimAllReward(0);
-
-    //         console.log(tx.hash);
-
-    //         await expect(tx).to.changeTokenBalance(rewardToken, owner, reward);
-    //     });
-    // });
-
-    describe("linear unlock", function () {
-        it("Should unlock rewards linearly over time", async function () {
-            const { hederaVault, owner, rewardToken, stakingToken } = await deployFixture();
-            const amountToDeposit = 1000;
-            const rewardAmount = 1000;
-
-            // Approve and deposit staking tokens
-            await stakingToken.approve(hederaVault.target, amountToDeposit);
-            await hederaVault.deposit(amountToDeposit, owner.address);
-
-            // Approve and add reward tokens
-            await rewardToken.approve(hederaVault.target, rewardAmount);
-            await hederaVault.addReward(rewardToken.target, rewardAmount);
-
-            // Simulate time passing for linear unlock
-            const rewardStart = (await hederaVault.userContribution(owner.address)).rewardStart;
-            const rewardEnd = (await hederaVault.userContribution(owner.address)).rewardEnd;
-            const duration = rewardEnd - rewardStart;
-
-            // Check initial rewards (should be zero)
-            let rewards = await hederaVault.getUserReward(owner, rewardToken.target);
-            expect(rewards).to.equal(0);
+            // await rewardToken.approve(hederaVault.target, 10 * 1e8);
 
-            // Increase time by half the duration
-            await ethers.provider.send("evm_increaseTime", [Number(duration) / 2]);
-            await ethers.provider.send("evm_mine", []);
+            console.log(await hederaVault.calculateReward(0));
 
-            // Check rewards after half the duration
-            rewards = await hederaVault.getUserReward(owner, rewardToken.target);
-            expect(rewards).to.be.closeTo(rewardAmount / 2, 1);
+            const reward = await hederaVault.calculateReward(0);
 
-            // Increase time to the end of the vesting period
-            await ethers.provider.send("evm_increaseTime", [Number(duration) / 2]);
-            await ethers.provider.send("evm_mine", []);
+            const tx = await hederaVault.claimAllReward(0);
 
-            // Check rewards after full duration
-            rewards = await hederaVault.getUserReward(owner, rewardToken.target);
-            expect(rewards).to.be.closeTo(rewardAmount, 1);
+            console.log(tx.hash);
 
-            // Claim rewards
-            await hederaVault.claimAllReward(0);
-            const balance = await rewardToken.balanceOf(owner.address);
-            expect(balance).to.be.closeTo(rewardAmount, 1);
+            await expect(tx).to.changeTokenBalance(rewardToken, owner, reward);
         });
     });
 });