test: update the cases for CallCheckpointPayload

valory-xyz · Dec 11, 2024 · e0d9f7a · e0d9f7a
1 parent c4b604a
commit e0d9f7a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/packages/valory/skills/staking_abci/tests/test_payloads.py b/packages/valory/skills/staking_abci/tests/test_payloads.py
@@ -41,6 +41,8 @@
                 "service_staking_state": 1,
                 "tx_submitter": "dummy tx submitter",
                 "tx_hash": "dummy tx hash",
+                "ts_checkpoint": 1,
+                "is_checkpoint_reached": True,
             },
         ),
     ],

diff --git a/packages/valory/skills/staking_abci/tests/test_rounds.py b/packages/valory/skills/staking_abci/tests/test_rounds.py
@@ -64,6 +64,8 @@ def abci_app() -> StakingAbciApp:
     "service_staking_state": StakingState.UNSTAKED.value,
     "tx_submitter": "dummy_submitter",
     "tx_hash": "dummy_tx_hash",
+    "ts_checkpoint": 0,
+    "is_checkpoint_reached": True,
 }
 
 
@@ -148,12 +150,16 @@ class TestCallCheckpointRound(BaseStakingRoundTestClass):
                         "service_staking_state": StakingState.STAKED.value,
                         "tx_submitter": "dummy_submitter",
                         "tx_hash": "dummy_tx_hash",
+                        "ts_checkpoint": 0,
+                        "is_checkpoint_reached": True,
                     }
                 ),
                 final_data={
                     "service_staking_state": StakingState.STAKED.value,
                     "tx_submitter": "dummy_submitter",
                     "tx_hash": "dummy_tx_hash",
+                    "ts_checkpoint": 0,
+                    "is_checkpoint_reached": True,
                 },
                 event=Event.DONE,
                 most_voted_payload=DUMMY_SERVICE_STATE["tx_submitter"],
@@ -184,6 +190,8 @@ class TestCallCheckpointRound(BaseStakingRoundTestClass):
                         "service_staking_state": StakingState.EVICTED.value,
                         "tx_submitter": "dummy_submitter",
                         "tx_hash": "dummy_tx_hash",
+                        "ts_checkpoint": 0,
+                        "is_checkpoint_reached": True,
                     }
                 ),
                 final_data={},
@@ -202,6 +210,8 @@ class TestCallCheckpointRound(BaseStakingRoundTestClass):
                         "service_staking_state": StakingState.STAKED.value,
                         "tx_submitter": "dummy_submitter",
                         "tx_hash": None,
+                        "ts_checkpoint": 0,
+                        "is_checkpoint_reached": True,
                     }
                 ),
                 final_data={},
@@ -279,6 +289,8 @@ def test_staking_abci_app_initialization(abci_app: StakingAbciApp) -> None:
             get_name(SynchronizedData.tx_submitter),
             get_name(SynchronizedData.most_voted_tx_hash),
             get_name(SynchronizedData.service_staking_state),
+            get_name(SynchronizedData.previous_checkpoint),
+            get_name(SynchronizedData.is_checkpoint_reached),
         },
         FinishedStakingRound: {
             get_name(SynchronizedData.service_staking_state),