Add arbitraryUnRegTxCert for Imptests.

This chooses between `UnRegTxCert` and `UnRegDepositTxCert`.
IntersectMBO · Jan 10, 2025 · 77d9def · 77d9def
1 parent 5e1ed75
commit 77d9def
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/eras/conway/impl/CHANGELOG.md b/eras/conway/impl/CHANGELOG.md
@@ -5,6 +5,10 @@
 * Deprecate `Conway` type synonym
 * Remove crypto parametrization from `ConwayEra`
 
+### testlib
+
+* Add `arbitraryUnRegTxCert` to choose between `UnRegTxCert` and `UnRegDepositTxCert` in Imp tests.
+
 ## 1.18.0.0
 
 * Remove `SlotNo` from `CertEnv` and `CertsEnv`

diff --git a/eras/conway/impl/testlib/Test/Cardano/Ledger/Conway/ImpTest.hs b/eras/conway/impl/testlib/Test/Cardano/Ledger/Conway/ImpTest.hs
@@ -43,6 +43,7 @@ module Test.Cardano.Ledger.Conway.ImpTest (
   submitYesVote_,
   submitFailingVote,
   trySubmitVote,
+  arbitraryUnRegTxCert,
   registerDRep,
   unRegisterDRep,
   updateDRep,
@@ -390,6 +391,20 @@ unRegisterDRep drep = do
       & bodyTxL . certsTxBodyL
         .~ SSeq.singleton (UnRegDRepTxCert drep refund)
 
+arbitraryUnRegTxCert ::
+  forall era.
+  ( ShelleyEraImp era
+  , ConwayEraTxCert era
+  ) =>
+  Credential 'Staking ->
+  ImpTestM era (TxCert era)
+arbitraryUnRegTxCert stakingCredential = do
+  keyDeposit <- getsNES (nesEsL . curPParamsEpochStateL . ppKeyDepositL)
+  elements
+    [ UnRegTxCert stakingCredential
+    , UnRegDepositTxCert stakingCredential keyDeposit
+    ]
+
 -- | Submit a transaction that updates a given DRep
 updateDRep ::
   forall era.