test: installment amount test fixed

icon-community · Jan 25, 2024 · 9dd738c · 9dd738c
1 parent d3c2d70
commit 9dd738c
Showing 1 changed file with 22 additions and 4 deletions.
diff --git a/CPSTreasury/src/test/java/community/icon/cps/score/cpstreasury/CPSTreasuryTest.java b/CPSTreasury/src/test/java/community/icon/cps/score/cpstreasury/CPSTreasuryTest.java
@@ -17,6 +17,7 @@
 import score.Context;
 
 import java.math.BigInteger;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 
@@ -205,10 +206,17 @@ void setBnUSDScoreNotContract() {
     void depositProposalFund() {
         /* totalBudget = 100, onsetPayment = 10% installmentCount = 2
          * remainingBudgetAfter onsetPayment = 90
-         * installMentAmount = 90/2 =45
+         * installMentAmount = 50 and 40
          */
         setOnsetPayment();
+        setCpsScoreMethod();
+
         depositProposalFundMethod();
+
+        List<Map<String,?>> remainingMilestone = new ArrayList<>();
+        remainingMilestone.add(Map.of(BUDGET,BigInteger.valueOf(50).multiply(ICX)));
+        remainingMilestone.add(Map.of(BUDGET,BigInteger.valueOf(40).multiply(ICX)));
+        doReturn(remainingMilestone).when(scoreSpy).callScore(eq(List.class),eq(score_address),eq("getRemainingMilestones"),any());
         @SuppressWarnings("unchecked")
         Map<String, ?> proposalDataDetails = (Map<String, ?>) tokenScore.call("getContributorProjectedFund", testing_account2.getAddress());
         @SuppressWarnings("unchecked")
@@ -220,13 +228,13 @@ void depositProposalFund() {
                 TOTAL_INSTALLMENT_PAID, BigInteger.TEN.multiply(ICX),
                 TOTAL_INSTALLMENT_COUNT, 2,
                 TOTAL_TIMES_INSTALLMENT_PAID, 0,
-                consts.INSTALLMENT_AMOUNT, BigInteger.valueOf(45).multiply(ICX));
+                consts.INSTALLMENT_AMOUNT, BigInteger.valueOf(50).multiply(ICX));
         assertEquals(proposalDetails.get(0), expectedData);
 
         assertEquals(proposalDataDetails.get(PROJECT_COUNT),1);
         assertEquals(proposalDataDetails.get(WITHDRAWN_BNUSD),BigInteger.TEN.multiply(ICX));
 
-        Map<String,BigInteger> totalAmount = Map.of(consts.bnUSD,BigInteger.valueOf(45).multiply(ICX));
+        Map<String,BigInteger> totalAmount = Map.of(consts.bnUSD,BigInteger.valueOf(50).multiply(ICX));
         assertEquals(proposalDataDetails.get(TOTAL_AMOUNT),totalAmount);
 
     }
@@ -270,10 +278,16 @@ void updateProposalFund() {
                     installmentAmount = (100-10)/2 = 45
            after budget adjustment :
                     totalBudget = 200 installCount = 3 onsetPayment = 10
-                    installmentAmount = (200-10)/3 = 63.333
+                    each installmentAmount euqally divided = (200-10)/3 = 63.333
         * */
         setOnsetPayment();
+        setCpsScoreMethod();
         depositProposalFundMethod();
+
+        List<Map<String,?>> remainingMilestone = new ArrayList<>();
+        remainingMilestone.add(Map.of(BUDGET,new BigInteger("63333333333333333333")));
+        doReturn(remainingMilestone).when(scoreSpy).callScore(eq(List.class),eq(score_address),eq("getRemainingMilestones"),any());
+
         JsonObject budgetAdjustmentData = new JsonObject();
         budgetAdjustmentData.add("method", "budgetAdjustment");
         JsonObject params = new JsonObject();
@@ -358,6 +372,10 @@ void sendInstallmentToContributor() { // send first installment
         depositProposalFund_MilestoneCheck();
         setCpsScoreMethod();
 
+        List<Map<String,?>> remainingMilestone = new ArrayList<>();
+        remainingMilestone.add(Map.of(BUDGET,BigInteger.valueOf(225).multiply(ICX).divide(BigInteger.valueOf(10))));
+        doReturn(remainingMilestone).when(scoreSpy).callScore(eq(List.class),eq(score_address),eq("getRemainingMilestones"),any());
+
         // proposal details after submission
         Map<String, ?> proposalDataDetails_before = (Map<String, ?>) tokenScore.call("getContributorProjectedFund", testing_account2.getAddress());
         assertEquals(proposalDataDetails_before.get("withdraw_amount_bnUSD"), BigInteger.valueOf(10).multiply(ICX));