Merge pull request #2185 from IntersectMBO/backend-test/epoch-boundary

Backend epoch boundary test

tests/govtool-backend/lib/kuber_api.py

@@ -52,6 +52,9 @@ def get_balance(self, address: str) -> int:
     def get_protocol_params(self):
         return self.__get("api/v3/protocol-params").json()
 
+    def get_query_system_start(self):
+        return self.__get("api/v3/genesis-params").json()
+
     def wait_for_txout(self, txin: str, timeout=400, log: bool = False):
         end = time.time() + timeout
         while time.time() < end:

tests/govtool-backend/test_cases/test_proposal.py

@@ -1,6 +1,8 @@
 from models.TestData import Proposal, ProposalListResponse, GetProposalResponse
+from datetime import datetime, timedelta
 import allure
-
+from lib.kuber_api import KuberApi
+kuber_api = KuberApi.from_env()
 
 def validate_proposal(proposal: Proposal) -> bool:
     assert isinstance(proposal, dict), f"Expected Proposal to be of type dict, got {type(proposal)}"
@@ -12,6 +14,35 @@ def validate_proposal(proposal: Proposal) -> bool:
         ), f"proposal.{key} should be of type {Proposal.__annotations__[key]} got {type(proposal[key])}"
     return True
 
+def validate_expiry_epoch(proposal,gov_action_lifetime) -> bool:
+    created_epoch_no = proposal['createdEpochNo']
+    actual_expiry_epoch_no = proposal['expiryEpochNo']
+    expected_expiry_epoch_no = created_epoch_no + gov_action_lifetime + 1
+    assert actual_expiry_epoch_no == expected_expiry_epoch_no, (
+            f"Expected expiry epoch {expected_expiry_epoch_no}, but got {actual_expiry_epoch_no}"
+        )
+    return True
+
+def validate_expiry_date(proposal, gov_action_lifetime, query_system) -> bool:
+    actual_expiry_date = proposal['expiryDate']
+    actual_expiry_date_in_datetime_format = datetime.strptime(actual_expiry_date, "%Y-%m-%dT%H:%M:%SZ")
+
+    # Calculate the expected expiry date
+    epoch_length = query_system['epochLength']
+    created_epoch_no = proposal['createdEpochNo']
+    start_time_of_0_epoch = query_system['systemStartTime']
+    added_seconds = (created_epoch_no + 1 + gov_action_lifetime) * epoch_length
+
+    # Parse the start time of epoch 0 and add the calculated seconds
+    expected_expiry_date = datetime.strptime(start_time_of_0_epoch, "%Y-%m-%dT%H:%M:%SZ") + timedelta(seconds=added_seconds)
+
+    max_time_difference = timedelta(minutes=10)
+    time_difference = abs(expected_expiry_date - actual_expiry_date_in_datetime_format)
+
+    # Assert that the difference is not greater than 10 minutes
+    assert time_difference <= max_time_difference, f"Time difference is greater than 10 minutes: {time_difference}"
+
+    return True
 
 @allure.story("Proposal")
 def test_list_proposal(govtool_api):
@@ -29,3 +60,15 @@ def test_get_proposal(govtool_api):
             proposal["txHash"] + "%23" + str(proposal["index"])
         ).json()
         assert validate_proposal(proposal_get["proposal"])
+
+@allure.story("Proposal")
+def test_epoch_boundary(govtool_api):
+    query_system = kuber_api.get_query_system_start()
+    epochParams = govtool_api.epoch_params().json()
+    gov_action_lifetime = epochParams['gov_action_lifetime']
+
+    response = govtool_api.proposal_list()
+    proposal_list = response.json()
+    for proposal in proposal_list["elements"]:
+        assert validate_expiry_epoch(proposal,gov_action_lifetime)
+        assert validate_expiry_date(proposal,gov_action_lifetime,query_system)