Remove registerStakeCredential from ShelleyEraImp

Author: Lucsanszky

eras/allegra/impl/testlib/Test/Cardano/Ledger/Allegra/ImpTest.hs

@@ -42,7 +42,6 @@ instance ShelleyEraImp AllegraEra where
 
   fixupTx = shelleyFixupTx
   expectTxSuccess = impShelleyExpectTxSuccess
-  registerStakeCredential = shelleyRegisterStakeCredential
   genRegTxCert = shelleyGenRegTxCert
   genUnRegTxCert = shelleyGenUnRegTxCert
 

eras/alonzo/impl/testlib/Test/Cardano/Ledger/Alonzo/ImpTest.hs

@@ -431,7 +431,6 @@ instance ShelleyEraImp AlonzoEra where
   impSatisfyNativeScript = impAllegraSatisfyNativeScript
   fixupTx = alonzoFixupTx
   expectTxSuccess = impAlonzoExpectTxSuccess
-  registerStakeCredential = shelleyRegisterStakeCredential
   genRegTxCert = shelleyGenRegTxCert
   genUnRegTxCert = shelleyGenUnRegTxCert
 

eras/babbage/impl/testlib/Test/Cardano/Ledger/Babbage/ImpTest.hs

@@ -53,7 +53,6 @@ instance ShelleyEraImp BabbageEra where
   impSatisfyNativeScript = impAllegraSatisfyNativeScript
   fixupTx = babbageFixupTx
   expectTxSuccess = impBabbageExpectTxSuccess
-  registerStakeCredential = shelleyRegisterStakeCredential
   genRegTxCert = shelleyGenRegTxCert
   genUnRegTxCert = shelleyGenUnRegTxCert
 

eras/conway/impl/testlib/Test/Cardano/Ledger/Conway/ImpTest.hs

@@ -130,7 +130,6 @@ module Test.Cardano.Ledger.Conway.ImpTest (
   FailBoth (..),
   delegateSPORewardAddressToDRep_,
   getCommittee,
-  conwayRegisterStakeCredential,
 ) where
 
 import Cardano.Ledger.Address (RewardAccount (..))
@@ -144,7 +143,6 @@ import Cardano.Ledger.BaseTypes (
   addEpochInterval,
   binOpEpochNo,
   inject,
-  networkId,
   textToUrl,
  )
 import Cardano.Ledger.Coin (Coin (..))
@@ -172,7 +170,7 @@ import Cardano.Ledger.Conway.Rules (
  )
 import Cardano.Ledger.Conway.State
 import Cardano.Ledger.Conway.TxCert (Delegatee (..))
-import Cardano.Ledger.Credential (Credential (..), credToText)
+import Cardano.Ledger.Credential (Credential (..))
 import Cardano.Ledger.DRep
 import Cardano.Ledger.Plutus.Language (Language (..), SLanguage (..), hashPlutusScript)
 import Cardano.Ledger.Shelley.LedgerState (
@@ -212,7 +210,6 @@ import qualified Data.Text as T
 import Data.Tree
 import qualified GHC.Exts as GHC (fromList)
 import Lens.Micro
-import Lens.Micro.Mtl (use)
 import Prettyprinter (align, hsep, viaShow, vsep)
 import Test.Cardano.Ledger.Babbage.ImpTest
 import Test.Cardano.Ledger.Conway.Arbitrary ()
@@ -297,7 +294,6 @@ instance ShelleyEraImp ConwayEra where
 
   fixupTx = babbageFixupTx
   expectTxSuccess = impBabbageExpectTxSuccess
-  registerStakeCredential = conwayRegisterStakeCredential
   genRegTxCert = conwayGenRegTxCert
   genUnRegTxCert = conwayGenUnRegTxCert
 
@@ -1792,19 +1788,3 @@ instance InjectRuleFailure "DELEG" ShelleyDelegPredFailure ConwayEra where
 
 getCommittee :: ConwayEraGov era => ImpTestM era (StrictMaybe (Committee era))
 getCommittee = getsNES $ nesEsL . epochStateGovStateL . committeeGovStateL
-
-conwayRegisterStakeCredential ::
-  forall era.
-  ( HasCallStack
-  , ConwayEraImp era
-  ) =>
-  Credential 'Staking ->
-  ImpTestM era RewardAccount
-conwayRegisterStakeCredential cred = do
-  deposit <- getsNES (nesEsL . curPParamsEpochStateL . ppKeyDepositL)
-  submitTxAnn_ ("Register Reward Account: " <> T.unpack (credToText cred)) $
-    mkBasicTx mkBasicTxBody
-      & bodyTxL . certsTxBodyL
-        .~ SSeq.fromList [RegDepositTxCert cred deposit]
-  networkId <- use (impGlobalsL . to networkId)
-  pure $ RewardAccount networkId cred

eras/dijkstra/testlib/Test/Cardano/Ledger/Dijkstra/ImpTest.hs

@@ -58,7 +58,6 @@ instance ShelleyEraImp DijkstraEra where
 
   fixupTx = babbageFixupTx
   expectTxSuccess = impBabbageExpectTxSuccess
-  registerStakeCredential = conwayRegisterStakeCredential
   genRegTxCert = dijkstraGenRegTxCert
   genUnRegTxCert = dijkstraGenUnRegTxCert
 

eras/mary/impl/testlib/Test/Cardano/Ledger/Mary/ImpTest.hs

@@ -27,7 +27,6 @@ instance ShelleyEraImp MaryEra where
   impSatisfyNativeScript = impAllegraSatisfyNativeScript
   fixupTx = shelleyFixupTx
   expectTxSuccess = impShelleyExpectTxSuccess
-  registerStakeCredential = shelleyRegisterStakeCredential
   genRegTxCert = shelleyGenRegTxCert
   genUnRegTxCert = shelleyGenUnRegTxCert
 

eras/shelley/impl/testlib/Test/Cardano/Ledger/Shelley/ImpTest.hs

@@ -69,7 +69,7 @@ module Test.Cardano.Ledger.Shelley.ImpTest (
   tryRunImpRuleNoAssertions,
   delegateStake,
   registerRewardAccount,
-  shelleyRegisterStakeCredential,
+  registerStakeCredential,
   getRewardAccountFor,
   getReward,
   lookupReward,
@@ -493,8 +493,6 @@ class
 
   expectTxSuccess :: HasCallStack => Tx era -> ImpTestM era ()
 
-  registerStakeCredential :: HasCallStack => Credential 'Staking -> ImpTestM era RewardAccount
-
   genRegTxCert :: HasCallStack => Credential 'Staking -> ImpTestM era (TxCert era)
 
   genUnRegTxCert :: HasCallStack => Credential 'Staking -> ImpTestM era (TxCert era)
@@ -742,7 +740,6 @@ instance
 
   fixupTx = shelleyFixupTx
   expectTxSuccess = impShelleyExpectTxSuccess
-  registerStakeCredential = shelleyRegisterStakeCredential
   genRegTxCert = shelleyGenRegTxCert
   genUnRegTxCert = shelleyGenUnRegTxCert
 
@@ -1477,19 +1474,19 @@ getRewardAccountFor stakingC = do
   networkId <- use (impGlobalsL . to networkId)
   pure $ RewardAccount networkId stakingC
 
-shelleyRegisterStakeCredential ::
+registerStakeCredential ::
   forall era.
   ( HasCallStack
   , ShelleyEraImp era
-  , ShelleyEraTxCert era
   ) =>
   Credential 'Staking ->
   ImpTestM era RewardAccount
-shelleyRegisterStakeCredential cred = do
+registerStakeCredential cred = do
+  regTxCert <- genRegTxCert cred
   submitTxAnn_ ("Register Reward Account: " <> T.unpack (credToText cred)) $
     mkBasicTx mkBasicTxBody
       & bodyTxL . certsTxBodyL
-        .~ SSeq.fromList [RegTxCert cred]
+        .~ SSeq.fromList [regTxCert]
   networkId <- use (impGlobalsL . to networkId)
   pure $ RewardAccount networkId cred