add getEarnedUser function on staking contract

LedgityLabs · Apr 30, 2024 · a6fba67 · a6fba67
1 parent d2609c4
commit a6fba67
Show file tree

Hide file tree

Showing 9 changed files with 463 additions and 186 deletions.
diff --git a/contracts/abis/LDYStaking.json b/contracts/abis/LDYStaking.json
@@ -238,6 +238,25 @@
     "stateMutability": "view",
     "type": "function"
   },
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "account",
+        "type": "address"
+      }
+    ],
+    "name": "getEarnedUser",
+    "outputs": [
+      {
+        "internalType": "uint256[]",
+        "name": "",
+        "type": "uint256[]"
+      }
+    ],
+    "stateMutability": "view",
+    "type": "function"
+  },
   {
     "inputs": [
       {

diff --git a/contracts/deployments.json b/contracts/deployments.json
@@ -30128,6 +30128,25 @@
               "stateMutability": "view",
               "type": "function"
             },
+            {
+              "inputs": [
+                {
+                  "internalType": "address",
+                  "name": "account",
+                  "type": "address"
+                }
+              ],
+              "name": "getEarnedUser",
+              "outputs": [
+                {
+                  "internalType": "uint256[]",
+                  "name": "",
+                  "type": "uint256[]"
+                }
+              ],
+              "stateMutability": "view",
+              "type": "function"
+            },
             {
               "inputs": [
                 {
@@ -30689,7 +30708,7 @@
           ]
         },
         "LDYStaking_Implementation": {
-          "address": "0x1157f1b171DC382165389c43a70De2E73da3645E",
+          "address": "0xFFF42672fb476afe698142A9B6211b772b89b8Ee",
           "abi": [
             {
               "inputs": [],
@@ -30930,6 +30949,25 @@
               "stateMutability": "view",
               "type": "function"
             },
+            {
+              "inputs": [
+                {
+                  "internalType": "address",
+                  "name": "account",
+                  "type": "address"
+                }
+              ],
+              "name": "getEarnedUser",
+              "outputs": [
+                {
+                  "internalType": "uint256[]",
+                  "name": "",
+                  "type": "uint256[]"
+                }
+              ],
+              "stateMutability": "view",
+              "type": "function"
+            },
             {
               "inputs": [
                 {
@@ -33402,7 +33440,7 @@
           ]
         },
         "LUSDC_Implementation": {
-          "address": "0x2c0F4C6127693C688ff4b8794fdB22FfB235cd87",
+          "address": "0xE0434e00f3088Ed0cE950F985f9dbe9898A18153",
           "abi": [
             {
               "anonymous": false,

diff --git a/contracts/foundry/test/LDYStaking.t.sol b/contracts/foundry/test/LDYStaking.t.sol
@@ -329,6 +329,10 @@ contract LDYStakingTest is Test, ModifiersExpectations {
         ldyStaking.stake(amount2, stakingPeriodIndex2);
         vm.stopPrank();
 
+        // check the lenght of earnedArray
+        uint256[] memory earnedArray = ldyStaking.getEarnedUser(account1);
+        assertEq(earnedArray.length, 2);
+
         // account1 unstakes token from staking pool 1
         uint256 skipDuration1 = stakingDurations[stakingPeriodIndex1];
         skip(skipDuration1);
@@ -338,6 +342,10 @@ contract LDYStakingTest is Test, ModifiersExpectations {
         vm.stopPrank();
         assertEq(ldyToken.balanceOf(account1), amount1 + earned1);
 
+        // check the lenght of earnedArray
+        earnedArray = ldyStaking.getEarnedUser(account1);
+        assertEq(earnedArray.length, 1);
+
         // account1 unstakes token from staking pool 2
         uint256 skipDuration2 = stakingDurations[stakingPeriodIndex2];
         skip(skipDuration2);
@@ -346,6 +354,10 @@ contract LDYStakingTest is Test, ModifiersExpectations {
         ldyStaking.unstake(amount2, 0);
         vm.stopPrank();
         assertEq(ldyToken.balanceOf(account1), amount1 + earned1 + amount2 + earned2);
+
+        // check the lenght of earnedArray
+        earnedArray = ldyStaking.getEarnedUser(account1);
+        assertEq(earnedArray.length, 0);
     }
 
     function testFuzz_Unstake_2(address account, uint256 amount, uint8 stakingPeriodIndex) public {
@@ -374,6 +386,10 @@ contract LDYStakingTest is Test, ModifiersExpectations {
         vm.stopPrank();
         assertEq(ldyToken.balanceOf(account), partAmount);
 
+        // check the lenght of earnedArray
+        uint256[] memory earnedArray = ldyStaking.getEarnedUser(account);
+        assertEq(earnedArray.length, 1);
+
         // account unstakes rest of token after 100 secs
         uint256 restAmount = amount - partAmount;
         skip(100);
@@ -383,6 +399,10 @@ contract LDYStakingTest is Test, ModifiersExpectations {
         ldyStaking.unstake(restAmount, 0);
         vm.stopPrank();
         assertEq(ldyToken.balanceOf(account), amount + earned);
+
+        // check the lenght of earnedArray
+        earnedArray = ldyStaking.getEarnedUser(account);
+        assertEq(earnedArray.length, 0);
     }
 
     function testFuzz_GetReward_Failed(

diff --git a/contracts/hardhat/deployments/baseSepolia/LDYStaking.json b/contracts/hardhat/deployments/baseSepolia/LDYStaking.json
diff --git a/contracts/hardhat/deployments/baseSepolia/LDYStaking_Implementation.json b/contracts/hardhat/deployments/baseSepolia/LDYStaking_Implementation.json
diff --git a/contracts/hardhat/deployments/baseSepolia/LUSDC.json b/contracts/hardhat/deployments/baseSepolia/LUSDC.json
diff --git a/contracts/hardhat/deployments/baseSepolia/LUSDC_Implementation.json b/contracts/hardhat/deployments/baseSepolia/LUSDC_Implementation.json