Add missing drep voting thresholds lenses in PParams

IntersectMBO · Apr 29, 2024 · 4d71f61 · 4d71f61
1 parent 9db2a4d
commit 4d71f61
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 14 deletions.
diff --git a/eras/conway/impl/CHANGELOG.md b/eras/conway/impl/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## 1.14.0 0
 
+* Add lenses:
+  * `dvtHardForkInitiationL`
+  * `dvtMotionNoConfidenceL`
+  * `dvtTreasuryWithdrawalL`
 * Add`DisallowedProposalDuringBootstrap` and `DisallowedVotesDuringBootstrap` to `ConwayGovPredFailure`
 * Make `DRepDistr` calculation include rewards when no UTxO stake is delegated. #4273
   * Rename `computeDrepPulser` to `computeDRepPulser`.

diff --git a/eras/conway/impl/src/Cardano/Ledger/Conway/PParams.hs b/eras/conway/impl/src/Cardano/Ledger/Conway/PParams.hs
@@ -48,10 +48,13 @@ module Cardano.Ledger.Conway.PParams (
   DRepVotingThresholds (..),
   dvtCommitteeNoConfidenceL,
   dvtCommitteeNormalL,
+  dvtHardForkInitiationL,
+  dvtMotionNoConfidenceL,
   dvtPPNetworkGroupL,
   dvtPPGovGroupL,
   dvtPPTechnicalGroupL,
   dvtPPEconomicGroupL,
+  dvtTreasuryWithdrawalL,
   dvtUpdateToConstitutionL,
   ConwayPParams (..),
   getLanguageView,
@@ -361,6 +364,15 @@ dvtCommitteeNoConfidenceL = lens dvtCommitteeNoConfidence (\x y -> x {dvtCommitt
 dvtCommitteeNormalL :: Lens' DRepVotingThresholds UnitInterval
 dvtCommitteeNormalL = lens dvtCommitteeNormal (\x y -> x {dvtCommitteeNormal = y})
 
+dvtMotionNoConfidenceL :: Lens' DRepVotingThresholds UnitInterval
+dvtMotionNoConfidenceL = lens dvtMotionNoConfidence (\x y -> x {dvtMotionNoConfidence = y})
+
+dvtHardForkInitiationL :: Lens' DRepVotingThresholds UnitInterval
+dvtHardForkInitiationL = lens dvtHardForkInitiation (\x y -> x {dvtHardForkInitiation = y})
+
+dvtTreasuryWithdrawalL :: Lens' DRepVotingThresholds UnitInterval
+dvtTreasuryWithdrawalL = lens dvtTreasuryWithdrawal (\x y -> x {dvtTreasuryWithdrawal = y})
+
 instance EncCBOR DRepVotingThresholds where
   encCBOR DRepVotingThresholds {..} =
     encodeListLen 10

diff --git a/eras/conway/impl/testlib/Test/Cardano/Ledger/Conway/Imp/EnactSpec.hs b/eras/conway/impl/testlib/Test/Cardano/Ledger/Conway/Imp/EnactSpec.hs
@@ -187,20 +187,9 @@ hardForkInitiationSpec =
     (committeeMember :| _) <- registerInitialCommittee
     modifyPParams $ \pp ->
       pp
-        & ppDRepVotingThresholdsL
-          %~ ( \dvt ->
-                dvt
-                  { dvtHardForkInitiation = 2 %! 3
-                  }
-             )
-        & ppPoolVotingThresholdsL
-          %~ ( \pvt ->
-                pvt
-                  { pvtHardForkInitiation = 2 %! 3
-                  }
-             )
-        & ppGovActionLifetimeL
-          .~ EpochInterval 20
+        & ppDRepVotingThresholdsL . dvtHardForkInitiationL .~ 2 %! 3
+        & ppPoolVotingThresholdsL . pvtHardForkInitiationL .~ 2 %! 3
+        & ppGovActionLifetimeL .~ EpochInterval 20
     _ <- setupPoolWithStake $ Coin 22_000_000
     (stakePoolId1, _, _) <- setupPoolWithStake $ Coin 22_000_000
     (stakePoolId2, _, _) <- setupPoolWithStake $ Coin 22_000_000