Add golden test for querySetSnapshotStakePoolDistr.

aniketd · aniketd · commit eccb3a0d5beb · 2026-05-04T17:39:18.000+05:30
Also export exampleVrfVerKeyHash from Shelley.Examples.
diff --git a/eras/shelley/impl/CHANGELOG.md b/eras/shelley/impl/CHANGELOG.md
@@ -49,13 +49,11 @@
   - `ChainPredicateFailure`
 * Add protocol version validation to `createInitialState`:
   - Validate that current protocol version is within the era's bounds
-
-### `cddl`
-
 * Remove `withAntiGen` re-export, since it's deprecated
 
 ### `testlib`
 
+* Export `exampleVrfVerKeyHash` from `Test.Cardano.Ledger.Shelley.Examples`
 * Add `HuddleRule "int32" ShelleyEra` instance
 * Add `withIssuerAndTxsInBlock_` and `withIssuerAndTxsInBlock`
 * Add a `Maybe (KeyHash BlockIssuer)` parameter to `withTxsInBlockEither`
diff --git a/eras/shelley/impl/testlib/Test/Cardano/Ledger/Shelley/Examples.hs b/eras/shelley/impl/testlib/Test/Cardano/Ledger/Shelley/Examples.hs
@@ -28,6 +28,7 @@ module Test.Cardano.Ledger.Shelley.Examples (
   examplePoolDistr,
   exampleStakePoolParams,
   exampleTxIns,
+  exampleVrfVerKeyHash,
   exampleProposedPPUpdates,
   exampleByronAddress,
   exampleShelleyScript,
diff --git a/libs/cardano-ledger-api/golden/conway/cbor/querySetSnapshotStakePoolDistr.cbor b/libs/cardano-ledger-api/golden/conway/cbor/querySetSnapshotStakePoolDistr.cbor
diff --git a/libs/cardano-ledger-api/golden/conway/json/querySetSnapshotStakePoolDistr.json b/libs/cardano-ledger-api/golden/conway/json/querySetSnapshotStakePoolDistr.json
@@ -0,0 +1,40 @@
+[
+    {
+        "pdTotalActiveStake": 1,
+        "unPoolDistr": {}
+    },
+    {
+        "pdTotalActiveStake": 1,
+        "unPoolDistr": {
+            "e0a714319812c3f773ba04ec5d6b3ffcd5aad85006805b047b082541": {
+                "individualPoolStake": {
+                    "denominator": 1,
+                    "numerator": 1
+                },
+                "individualPoolStakeVrf": "c5e21ab1c9f6022d81c3b25e3436cb7f1df77f9652ae3e1310c28e621dd87b4c",
+                "individualTotalPoolStake": 1
+            }
+        }
+    },
+    {
+        "pdTotalActiveStake": 6000000000,
+        "unPoolDistr": {
+            "a646474b8f5431261506b6c273d307c7569a4eb6c96b42dd4a29520a": {
+                "individualPoolStake": {
+                    "denominator": 8,
+                    "numerator": 3
+                },
+                "individualPoolStakeVrf": "c5e21ab1c9f6022d81c3b25e3436cb7f1df77f9652ae3e1310c28e621dd87b4c",
+                "individualTotalPoolStake": 5000000000
+            },
+            "e0a714319812c3f773ba04ec5d6b3ffcd5aad85006805b047b082541": {
+                "individualPoolStake": {
+                    "denominator": 4,
+                    "numerator": 1
+                },
+                "individualPoolStakeVrf": "c5e21ab1c9f6022d81c3b25e3436cb7f1df77f9652ae3e1310c28e621dd87b4c",
+                "individualTotalPoolStake": 1000000000
+            }
+        }
+    }
+]
diff --git a/libs/cardano-ledger-api/golden/dijkstra/cbor/querySetSnapshotStakePoolDistr.cbor b/libs/cardano-ledger-api/golden/dijkstra/cbor/querySetSnapshotStakePoolDistr.cbor
diff --git a/libs/cardano-ledger-api/golden/dijkstra/json/querySetSnapshotStakePoolDistr.json b/libs/cardano-ledger-api/golden/dijkstra/json/querySetSnapshotStakePoolDistr.json
@@ -0,0 +1,40 @@
+[
+    {
+        "pdTotalActiveStake": 1,
+        "unPoolDistr": {}
+    },
+    {
+        "pdTotalActiveStake": 1,
+        "unPoolDistr": {
+            "e0a714319812c3f773ba04ec5d6b3ffcd5aad85006805b047b082541": {
+                "individualPoolStake": {
+                    "denominator": 1,
+                    "numerator": 1
+                },
+                "individualPoolStakeVrf": "c5e21ab1c9f6022d81c3b25e3436cb7f1df77f9652ae3e1310c28e621dd87b4c",
+                "individualTotalPoolStake": 1
+            }
+        }
+    },
+    {
+        "pdTotalActiveStake": 6000000000,
+        "unPoolDistr": {
+            "a646474b8f5431261506b6c273d307c7569a4eb6c96b42dd4a29520a": {
+                "individualPoolStake": {
+                    "denominator": 8,
+                    "numerator": 3
+                },
+                "individualPoolStakeVrf": "c5e21ab1c9f6022d81c3b25e3436cb7f1df77f9652ae3e1310c28e621dd87b4c",
+                "individualTotalPoolStake": 5000000000
+            },
+            "e0a714319812c3f773ba04ec5d6b3ffcd5aad85006805b047b082541": {
+                "individualPoolStake": {
+                    "denominator": 4,
+                    "numerator": 1
+                },
+                "individualPoolStakeVrf": "c5e21ab1c9f6022d81c3b25e3436cb7f1df77f9652ae3e1310c28e621dd87b4c",
+                "individualTotalPoolStake": 1000000000
+            }
+        }
+    }
+]
diff --git a/libs/cardano-ledger-api/test/Test/Cardano/Ledger/Api/State/QuerySpec.hs b/libs/cardano-ledger-api/test/Test/Cardano/Ledger/Api/State/QuerySpec.hs
@@ -59,6 +59,7 @@ import Test.Cardano.Ledger.Api.State.Query.Examples (
   queryPoolParametersExamples,
   queryRegisteredDRepStakeDistrExamples,
   querySPOStakeDistrExamples,
+  querySetSnapshotStakePoolDistrExamples,
   queryStakePoolDefaultVoteExamples,
   queryStakePoolDelegsAndRewardsExamples,
   queryStakePoolRelaysExamples,
@@ -149,6 +150,9 @@ latestErasSpec =
         "queryRegisteredDRepStakeDistr"
         queryRegisteredDRepStakeDistrExamples
       eraLedgerStateQueryGoldenSpec @era "querySPOStakeDistr" querySPOStakeDistrExamples
+      eraLedgerStateQueryNoFromJSONGoldenSpec @era
+        "querySetSnapshotStakePoolDistr"
+        querySetSnapshotStakePoolDistrExamples
       eraLedgerStateQueryCBORGoldenSpec @era
         "queryStakePoolDefaultVote"
         queryStakePoolDefaultVoteExamples
diff --git a/libs/cardano-ledger-api/testlib/Test/Cardano/Ledger/Api/State/Query/Examples.hs b/libs/cardano-ledger-api/testlib/Test/Cardano/Ledger/Api/State/Query/Examples.hs
@@ -1,5 +1,7 @@
+{-# LANGUAGE DataKinds #-}
 {-# LANGUAGE NumericUnderscores #-}
 {-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE TypeApplications #-}
 
 module Test.Cardano.Ledger.Api.State.Query.Examples (
   queryAccountsDepositsExamples,
@@ -14,6 +16,7 @@ module Test.Cardano.Ledger.Api.State.Query.Examples (
   queryPoolParametersExamples,
   queryRegisteredDRepStakeDistrExamples,
   querySPOStakeDistrExamples,
+  querySetSnapshotStakePoolDistrExamples,
   queryStakePoolDefaultVoteExamples,
   queryStakePoolDelegsAndRewardsExamples,
   queryStakePoolRelaysExamples,
@@ -23,12 +26,19 @@ import Cardano.Base.IP (toIPv4, toIPv6)
 import Cardano.Ledger.Api.Governance (Constitution (..))
 import Cardano.Ledger.Api.State.Query (DefaultVote (..))
 import Cardano.Ledger.BaseTypes (AnchorData, EpochNo (..), Port (..), StrictMaybe (..), textToDns)
-import Cardano.Ledger.Coin (Coin (..), CompactForm (..))
+import Cardano.Ledger.Coin (Coin (..), CompactForm (..), knownNonZeroCoin)
 import Cardano.Ledger.Credential (Credential (..))
 import Cardano.Ledger.DRep (DRep (..), DRepState (..))
 import Cardano.Ledger.Hashes (SafeHash)
 import Cardano.Ledger.Keys (KeyHash, KeyRole (..))
-import Cardano.Ledger.State (ChainAccountState (..), StakePoolParams (..), StakePoolRelay (..))
+import Cardano.Ledger.State (
+  ChainAccountState (..),
+  IndividualPoolStake (..),
+  PoolDistr (..),
+  StakePoolParams (..),
+  StakePoolRelay (..),
+ )
+import Data.Default (def)
 import Data.Map.Strict (Map)
 import qualified Data.Map.Strict as Map
 import Data.Maybe (fromJust)
@@ -39,7 +49,13 @@ import Data.Set (Set)
 import qualified Data.Set as Set
 import Test.Cardano.Ledger.Conway.Examples (exampleAnchor)
 import Test.Cardano.Ledger.Core.Utils (mkDummySafeHash)
-import Test.Cardano.Ledger.Shelley.Examples (exampleStakePoolParams, mkKeyHash, mkScriptHash)
+import Test.Cardano.Ledger.Shelley.Examples (
+  examplePoolDistr,
+  exampleStakePoolParams,
+  exampleVrfVerKeyHash,
+  mkKeyHash,
+  mkScriptHash,
+ )
 
 queryConstitutionExamples :: [Constitution era]
 queryConstitutionExamples =
@@ -92,6 +108,25 @@ querySPOStakeDistrExamples =
       ]
   ]
 
+querySetSnapshotStakePoolDistrExamples :: [PoolDistr]
+querySetSnapshotStakePoolDistrExamples =
+  [ def
+  , examplePoolDistr
+  , PoolDistr
+      ( Map.fromList
+          [
+            ( mkKeyHash 1
+            , IndividualPoolStake (1 % 4) (CompactCoin 1_000_000_000) exampleVrfVerKeyHash
+            )
+          ,
+            ( mkKeyHash 2
+            , IndividualPoolStake (3 % 8) (CompactCoin 5_000_000_000) exampleVrfVerKeyHash
+            )
+          ]
+      )
+      (knownNonZeroCoin @6_000_000_000)
+  ]
+
 queryStakePoolDefaultVoteExamples :: [DefaultVote]
 queryStakePoolDefaultVoteExamples =
   [ DefaultNo
