test: Add another check to new committee rollback

Author: sgillespie

cardano-chain-gen/src/Cardano/Mock/Query.hs

@@ -14,6 +14,7 @@ module Cardano.Mock.Query (
   queryRewardRests,
   queryTreasuryDonations,
   queryVoteCounts,
+  queryEpochStateCount,
 ) where
 
 import qualified Cardano.Db as Db
@@ -201,3 +202,15 @@ queryVoteCounts txHash idx = do
             &&. vote ^. Db.VotingProcedureIndex ==. val idx
         pure countRows
       pure (maybe 0 unValue res)
+
+queryEpochStateCount ::
+  MonadIO io =>
+  Word64 ->
+  ReaderT SqlBackend io Word64
+queryEpochStateCount epochNo = do
+  res <- selectOne $ do
+    epochState <- from (table @Db.EpochState)
+    where_ (epochState ^. Db.EpochStateEpochNo ==. val epochNo)
+    pure countRows
+
+  pure (maybe 0 unValue res)

cardano-chain-gen/test/Test/Cardano/Db/Mock/Unit/Conway/Governance.hs

@@ -56,10 +56,8 @@ drepDistr =
 
     -- Add stake
     void (Api.registerAllStakeCreds interpreter server)
-
     -- Register DRep and delegate votes to it
     void (Api.registerDRepsAndDelegateVotes interpreter server)
-
     -- DRep distribution is calculated at end of the current epoch
     epoch1 <- Api.fillUntilNextEpoch interpreter server
 
@@ -124,13 +122,23 @@ rollbackNewCommittee =
     -- Rollback the last 2 blocks
     epoch1' <- rollbackBlocks interpreter server 2 epoch2
     -- Wait for it to sync
-    assertBlockNoBackoff dbSync (length epoch1' + 4)
+    assertBlockNoBackoff dbSync (length epoch1' + 3)
     -- Should not have a new committee member
     assertEqQuery
       dbSync
       Query.queryGovActionCounts
       (1, 0, 0, 0)
       "Unexpected governance action counts"
+
+    -- Fast forward to next epoch
+    epoch2' <- Api.fillUntilNextEpoch interpreter server
+    assertBlockNoBackoff dbSync (length (epoch1' <> epoch2') + 3)
+    -- Should now have 2 identical committees
+    assertEqQuery
+      dbSync
+      (Query.queryEpochStateCount 2)
+      2
+      "Unexpected epoch state count for epoch 2"
   where
     testLabel = "conwayRollbackNewCommittee"
 

cardano-chain-gen/test/testfiles/fingerprint/conwayRollbackNewCommittee

@@ -1 +1 @@
-[12,16,18,21,24,30,31,32,33,40,41,42,43,47,52,60,62,70,80,84,86,92,98,100,106,109,110,111,112,127,134,138,146,149,154,166,168,178,183,188,193,194,198,200,202,220,222,223,224,225,231,239,242,247,261,282,283,288,289,301,302,303,308,313,315,316,320,331,334,344,345,363,364,368,369,375,377,381,389,394,407,418,422,425,430,437,438,439,440,447,450,453,454,456,458,461,467,492,499,507,516,524,538,541,544,546,550,567,573,576,577,579,580,586,589,595,597,603,605,609,616,618,619,623,624,634,636,643,644,659,664,665,672,678,692,705,711,712,719,726,730,739,740,743,747,749,751,754,759,762,763,765,767,773,777,786,788,789,794,801,806,807,829,830,832,849,851,853,869,871,874,875,878,882,888,893,895,896,898,899,903,906,908,911,912,913,922,930,932,938,941,944,950,960,963,966,968,972,977,985,986,988,990,991,994,997,1001,997]
+[12,16,18,21,24,30,31,32,33,40,41,42,43,47,52,60,62,70,80,84,86,92,98,100,106,109,110,111,112,127,134,138,146,149,154,166,168,178,183,188,193,194,198,200,202,220,222,223,224,225,231,239,242,247,261,282,283,288,289,301,302,303,308,313,315,316,320,331,334,344,345,363,364,368,369,375,377,381,389,394,407,418,422,425,430,437,438,439,440,447,450,453,454,456,458,461,467,492,499,507,516,524,538,541,544,546,550,567,573,576,577,579,580,586,589,595,597,603,605,609,616,618,619,623,624,634,636,643,644,659,664,665,672,678,692,705,711,712,719,726,730,739,740,743,747,749,751,754,759,762,763,765,767,773,777,786,788,789,794,801,806,807,829,830,832,849,851,853,869,871,874,875,878,882,888,893,895,896,898,899,903,906,908,911,912,913,922,930,932,938,941,944,950,960,963,966,968,972,977,985,986,988,990,991,994,997,1001,997,1001]