Skip to content

Commit 3141ca1

Browse files
committed
Improve readability of leadership check
1 parent bd2d757 commit 3141ca1

File tree

2 files changed

+20
-18
lines changed

2 files changed

+20
-18
lines changed

cardano-testnet/test/Test/Cli/Alonzo/LeadershipSchedule.hs

+6-5
Original file line numberDiff line numberDiff line change
@@ -504,12 +504,13 @@ hprop_leadershipSchedule = H.integration . H.runFinallies . H.workspace "alonzo"
504504

505505
leadershipDeadline <- H.noteShowM $ DTC.addUTCTime 90 <$> H.noteShowIO DTC.getCurrentTime
506506

507-
H.byDeadlineM 10 leadershipDeadline "Wait for a leadership at least as new as the highest one we expect" $ do
508-
leaderSlots <- H.getRelevantLeaderSlots (TR.nodeStdout $ TR.poolRuntime poolNode1) (minimum expectedLeadershipSlotNumbers)
509-
maxActualSlot <- H.noteShow $ maximum leaderSlots
507+
-- We need enough time to pass such that the expected leadership slots generated by the
508+
-- leadership-schedule command have actually occurred.
509+
leaderSlots <- H.byDeadlineM 10 leadershipDeadline "Wait for chain to surpass all expected leadership slots" $ do
510+
someLeaderSlots <- H.getRelevantLeaderSlots (TR.nodeStdout $ TR.poolRuntime poolNode1) (minimum expectedLeadershipSlotNumbers)
511+
maxActualSlot <- H.noteShow $ maximum someLeaderSlots
510512
H.assert $ maxActualSlot >= maxSlotExpected
511-
512-
leaderSlots <- H.getRelevantLeaderSlots (TR.poolNodeStdout poolNode1) (minimum expectedLeadershipSlotNumbers)
513+
pure someLeaderSlots
513514

514515
-- It's possible for some slots to not be assigned in TPraos when BFT nodes are running.
515516
-- TODO Remove BFT nodes from testnet and assert the schedule is equal to actual slots

cardano-testnet/test/Test/Cli/Babbage/LeadershipSchedule.hs

+14-13
Original file line numberDiff line numberDiff line change
@@ -137,16 +137,16 @@ hprop_leadershipSchedule = H.integration . H.runFinallies . H.workspace "alonzo"
137137

138138
leadershipDeadline <- H.noteShowM $ DTC.addUTCTime 90 <$> H.noteShowIO DTC.getCurrentTime
139139

140-
-- Retrieve actual slots
141-
H.byDeadlineM 10 leadershipDeadline "Wait for a leadership at least as new as the highest one we expect" $ do
142-
leaderSlots <- H.getRelevantLeaderSlots (TR.poolNodeStdout poolNode1) (minimum expectedLeadershipSlotNumbers)
143-
if L.null leaderSlots
140+
-- We need enough time to pass such that the expected leadership slots generated by the
141+
-- leadership-schedule command have actually occurred.
142+
leaderSlots <- H.byDeadlineM 10 leadershipDeadline "Wait for chain to surpass all expected leadership slots" $ do
143+
someLeaderSlots <- H.getRelevantLeaderSlots (TR.poolNodeStdout poolNode1) (minimum expectedLeadershipSlotNumbers)
144+
if L.null someLeaderSlots
144145
then H.failure
145146
else do
146-
maxActualSlot <- H.noteShow $ maximum leaderSlots
147+
maxActualSlot <- H.noteShow $ maximum someLeaderSlots
147148
H.assert $ maxActualSlot >= maxSlotExpected
148-
149-
leaderSlots <- H.getRelevantLeaderSlots (TR.poolNodeStdout poolNode1) (minimum expectedLeadershipSlotNumbers)
149+
pure someLeaderSlots
150150

151151
H.noteShow_ expectedLeadershipSlotNumbers
152152
H.noteShow_ leaderSlots
@@ -179,15 +179,16 @@ hprop_leadershipSchedule = H.integration . H.runFinallies . H.workspace "alonzo"
179179

180180
leadershipDeadline <- H.noteShowM $ DTC.addUTCTime 90 <$> H.noteShowIO DTC.getCurrentTime
181181

182-
H.byDeadlineM 10 leadershipDeadline "Wait for a leadership at least as new as the highest one we expect" $ do
183-
leaderSlots <- H.getRelevantLeaderSlots (TR.poolNodeStdout poolNode1) (minimum expectedLeadershipSlotNumbers)
184-
if L.null leaderSlots
182+
-- We need enough time to pass such that the expected leadership slots generated by the
183+
-- leadership-schedule command have actually occurred.
184+
leaderSlots <- H.byDeadlineM 10 leadershipDeadline "Wait for chain to surpass all expected leadership slots" $ do
185+
someLeaderSlots <- H.getRelevantLeaderSlots (TR.poolNodeStdout poolNode1) (minimum expectedLeadershipSlotNumbers)
186+
if L.null someLeaderSlots
185187
then H.failure
186188
else do
187-
maxActualSlot <- H.noteShow $ maximum leaderSlots
189+
maxActualSlot <- H.noteShow $ maximum someLeaderSlots
188190
H.assert $ maxActualSlot >= maxSlotExpected
189-
190-
leaderSlots <- H.getRelevantLeaderSlots (TR.poolNodeStdout poolNode1) (minimum expectedLeadershipSlotNumbers)
191+
pure someLeaderSlots
191192

192193
H.noteShow_ expectedLeadershipSlotNumbers
193194
H.noteShow_ leaderSlots

0 commit comments

Comments
 (0)