test: Add a new committee member rollback test

Author: sgillespie

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

@@ -246,6 +246,7 @@ unitTests iom knownMigrations =
         , test "parameter change" Governance.parameterChange
         , test "hard fork" Governance.hardFork
         , test "info action" Governance.infoAction
+        , test "rollback new committee member" Governance.rollbackNewCommittee
         ]
     ]
   where

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

@@ -16,6 +16,7 @@ module Test.Cardano.Db.Mock.Unit.Conway.Governance (
   parameterChange,
   hardFork,
   infoAction,
+  rollbackNewCommittee,
 ) where
 
 import qualified Cardano.Db as Db
@@ -29,8 +30,8 @@ import Cardano.Ledger.Core (txIdTx)
 import Cardano.Ledger.Credential (Credential (..))
 import Cardano.Ledger.Keys (KeyHash (..))
 import Cardano.Ledger.SafeHash (SafeToHash (..))
-import Cardano.Mock.ChainSync.Server (IOManager)
-import Cardano.Mock.Forging.Interpreter (getCurrentEpoch)
+import Cardano.Mock.ChainSync.Server (IOManager, ServerHandle)
+import Cardano.Mock.Forging.Interpreter (Interpreter, getCurrentEpoch)
 import qualified Cardano.Mock.Forging.Tx.Conway as Conway
 import qualified Cardano.Mock.Forging.Tx.Generic as Forging
 import Cardano.Mock.Forging.Types
@@ -39,11 +40,13 @@ import Cardano.Prelude
 import Cardano.Slotting.Slot (EpochNo (..))
 import qualified Data.Map as Map
 import Data.Maybe (fromJust)
+import Data.Maybe.Strict (StrictMaybe (..))
 import qualified Ouroboros.Consensus.Shelley.Eras as Consensus
+import Ouroboros.Network.Block (blockPoint)
 import Test.Cardano.Db.Mock.Config
 import qualified Test.Cardano.Db.Mock.UnifiedApi as Api
 import Test.Cardano.Db.Mock.Validate
-import Test.Tasty.HUnit (Assertion)
+import Test.Tasty.HUnit (Assertion, assertFailure)
 import qualified Prelude
 
 drepDistr :: IOManager -> [(Text, Text)] -> Assertion
@@ -80,48 +83,108 @@ newCommittee =
 
     -- Add stake
     void (Api.registerAllStakeCreds interpreter server)
-
     -- Register a DRep and delegate votes to it
     void (Api.registerDRepsAndDelegateVotes interpreter server)
 
-    -- Create and vote for gov action
-    let committeeHash = "e0a714319812c3f773ba04ec5d6b3ffcd5aad85006805b047b082541"
-        committeeCred = KeyHashObj (KeyHash committeeHash)
+    -- Propose, ratify, and enact a new committee member
+    epochs <- enactNewCommittee interpreter server
 
-    void $
-      Api.withConwayFindLeaderAndSubmit interpreter server $ \ledger -> do
-        let
-          -- Create gov action tx
-          addCcTx = Conway.mkAddCommitteeTx committeeCred
-          -- Create votes for all stake pools. We start in the Conway bootstrap phase, so
-          -- DRep votes are not yet required.
-          addVoteTx =
-            Conway.mkGovVoteYesTx
-              govActionId
-              (Forging.drepVoters ++ Forging.resolveStakePoolVoters ledger)
-          govActionId =
-            GovActionId
-              { gaidTxId = txIdTx addCcTx
-              , gaidGovActionIx = GovActionIx 0
-              }
+    -- Wait for it to sync
+    assertBlockNoBackoff dbSync (length epochs + 2)
+    -- Should now have a committee member
+    assertEqQuery
+      dbSync
+      Query.queryGovActionCounts
+      (1, 1, 0, 0)
+      "Unexpected governance action counts"
+  where
+    testLabel = "conwayNewCommittee"
 
-        -- Create votes
-        pure [addCcTx, addVoteTx]
+rollbackNewCommittee :: IOManager -> [(Text, Text)] -> Assertion
+rollbackNewCommittee =
+  withFullConfig conwayConfigDir testLabel $ \interpreter server dbSync -> do
+    startDBSync dbSync
 
-    -- It takes 2 epochs to enact a proposal--ratification will happen on the next
-    -- epoch and enacted on the following.
-    epochs <- Api.fillEpochs interpreter server 2
+    -- Add stake
+    void (Api.registerAllStakeCreds interpreter server)
+    -- Register a DRep and delegate votes to it
+    void (Api.registerDRepsAndDelegateVotes interpreter server)
 
+    -- Propose, ratify, and enact a new committee member
+    epoch2 <- enactNewCommittee interpreter server
     -- Wait for it to sync
-    assertBlockNoBackoff dbSync (length epochs + 3)
+    assertBlockNoBackoff dbSync (length epoch2 + 2)
     -- Should now have a committee member
     assertEqQuery
       dbSync
       Query.queryGovActionCounts
       (1, 1, 0, 0)
       "Unexpected governance action counts"
+
+    -- Rollback the last 2 blocks
+    epoch1' <- rollbackBlocks interpreter server 2 epoch2
+    -- Wait for it to sync
+    assertBlockNoBackoff dbSync (length epoch1' + 4)
+    -- Should not have a new committee member
+    assertEqQuery
+      dbSync
+      Query.queryGovActionCounts
+      (1, 0, 0, 0)
+      "Unexpected governance action counts"
   where
-    testLabel = "conwayNewCommittee"
+    testLabel = "conwayRollbackNewCommittee"
+
+enactNewCommittee :: Interpreter -> ServerHandle IO CardanoBlock -> IO [CardanoBlock]
+enactNewCommittee interpreter server = do
+  -- Create and vote for gov action
+  let committeeHash = "e0a714319812c3f773ba04ec5d6b3ffcd5aad85006805b047b082541"
+      committeeCred = KeyHashObj (KeyHash committeeHash)
+
+  blk <-
+    Api.withConwayFindLeaderAndSubmit interpreter server $ \ledger -> do
+      let
+        -- Create gov action tx
+        addCcTx = Conway.mkAddCommitteeTx committeeCred
+        -- Create votes for all stake pools. We start in the Conway bootstrap phase, so
+        -- DRep votes are not yet required.
+        addVoteTx =
+          Conway.mkGovVoteYesTx
+            govActionId
+            (Forging.drepVoters ++ Forging.resolveStakePoolVoters ledger)
+        govActionId =
+          GovActionId
+            { gaidTxId = txIdTx addCcTx
+            , gaidGovActionIx = GovActionIx 0
+            }
+
+      -- Create votes
+      pure [addCcTx, addVoteTx]
+
+  -- It takes 2 epochs to enact a proposal--ratification will happen on the next
+  -- epoch and enacted on the following.
+  epochs <- Api.fillEpochs interpreter server 2
+  pure (blk : epochs)
+
+rollbackBlocks ::
+  Interpreter ->
+  ServerHandle IO CardanoBlock ->
+  Int ->
+  [CardanoBlock] ->
+  IO [CardanoBlock]
+rollbackBlocks interpreter server n blocks = do
+  (rollbackPoint, blocks') <-
+    case drop n (reverse blocks) of
+      (blk : blks) -> pure (blockPoint blk, blks)
+      [] -> assertFailure "Expected at least 3 blocks"
+
+  -- Rollback to the previous epoch
+  Api.rollbackTo interpreter server rollbackPoint
+  -- Create a fork
+  newBlock <-
+    Api.withConwayFindLeaderAndSubmitTx interpreter server $
+      Conway.mkSimpleDCertTx [(StakeIndexNew 1, Conway.mkRegTxCert SNothing)]
+
+  pure (blocks' ++ [newBlock])
 
 updateConstitution :: IOManager -> [(Text, Text)] -> Assertion
 updateConstitution =

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

@@ -0,0 +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]