Add Imp fixup for collateral return txout

Author: teodanciu

eras/alonzo/impl/CHANGELOG.md

@@ -12,6 +12,10 @@
 * Deprecate `Alonzo` type synonym
 * Remove crypto parametrization from `AlonzoEra`
 
+### `testlib`
+
+* Expose `alonzoFixupFees`
+
 ## 1.12.0.0
 
 * Remove deprecated `lookupPlutusScript`

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

@@ -38,6 +38,7 @@ module Test.Cardano.Ledger.Alonzo.ImpTest (
   fixupRedeemers,
   fixupRedeemerIndices,
   fixupScriptWits,
+  alonzoFixupFees,
 ) where
 
 import Cardano.Ledger.Address (Addr (..))

eras/babbage/impl/CHANGELOG.md

@@ -7,6 +7,10 @@
 * Deprecate `Babbage` type synonym
 * Remove crypto parametrization from `BabbageEra`
 
+### `testlib`
+
+* Add `babbageFixupTx`
+
 ## 1.10.1.0
 
 * Use `Mismatch` to clarify predicate failures. #4711

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

@@ -11,6 +11,7 @@
 {-# OPTIONS_GHC -Wno-orphans #-}
 
 module Test.Cardano.Ledger.Babbage.ImpTest (
+  babbageFixupTx,
   module Test.Cardano.Ledger.Alonzo.ImpTest,
   produceRefScript,
   produceRefScripts,
@@ -21,13 +22,14 @@ import Cardano.Ledger.Babbage.Core
 import Cardano.Ledger.BaseTypes (StrictMaybe (..))
 import Cardano.Ledger.Plutus.Language (Language (..), SLanguage (..))
 import Cardano.Ledger.Shelley.LedgerState (curPParamsEpochStateL, nesEsL)
-import Cardano.Ledger.Tools (setMinCoinTxOut)
+import Cardano.Ledger.Tools (ensureMinCoinTxOut, setMinCoinTxOut)
 import Cardano.Ledger.TxIn (TxIn, mkTxInPartial)
-import Control.Monad (forM)
+import Control.Monad (forM, (>=>))
 import Data.List.NonEmpty (NonEmpty (..))
 import qualified Data.List.NonEmpty as NE
 import qualified Data.Sequence.Strict as SSeq
-import Lens.Micro ((&), (.~), (<>~))
+import GHC.Stack (HasCallStack)
+import Lens.Micro
 import Test.Cardano.Ledger.Alonzo.ImpTest
 import Test.Cardano.Ledger.Babbage.TreeDiff ()
 import Test.Cardano.Ledger.Plutus (testingCostModels)
@@ -37,7 +39,40 @@ instance ShelleyEraImp BabbageEra where
     defaultInitNewEpochState
       (nesEsL . curPParamsEpochStateL . ppCostModelsL <>~ testingCostModels [PlutusV2])
   impSatisfyNativeScript = impAllegraSatisfyNativeScript
-  fixupTx = alonzoFixupTx
+  fixupTx = babbageFixupTx
+
+babbageFixupTx ::
+  ( HasCallStack
+  , AlonzoEraImp era
+  , BabbageEraTxBody era
+  ) =>
+  Tx era ->
+  ImpTestM era (Tx era)
+babbageFixupTx =
+  addNativeScriptTxWits
+    >=> fixupAuxDataHash
+    >=> addCollateralInput
+    >=> addRootTxIn
+    >=> fixupScriptWits
+    >=> fixupOutputDatums
+    >=> fixupDatums
+    >=> fixupRedeemerIndices
+    >=> fixupTxOuts
+    >=> fixupCollateralReturn
+    >=> alonzoFixupFees
+    >=> fixupRedeemers
+    >=> fixupPPHash
+    >=> updateAddrTxWits
+
+fixupCollateralReturn ::
+  ( ShelleyEraImp era
+  , BabbageEraTxBody era
+  ) =>
+  Tx era ->
+  ImpTestM era (Tx era)
+fixupCollateralReturn tx = do
+  pp <- getsNES $ nesEsL . curPParamsEpochStateL
+  pure $ tx & bodyTxL . collateralReturnTxBodyL %~ fmap (ensureMinCoinTxOut pp)
 
 instance ShelleyEraImp BabbageEra => MaryEraImp BabbageEra
 

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

@@ -317,7 +317,7 @@ instance ShelleyEraImp ConwayEra where
 
   modifyPParams = conwayModifyPParams
 
-  fixupTx = alonzoFixupTx
+  fixupTx = babbageFixupTx
 
 instance MaryEraImp ConwayEra