Fix cardano-node-test

CardanoSolutions · Oct 31, 2023 · 08bcf37 · 08bcf37
1 parent 5e2fce2
commit 08bcf37
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/cardano-node/src/Cardano/Node/LedgerEvent.hs b/cardano-node/src/Cardano/Node/LedgerEvent.hs
@@ -47,6 +47,7 @@ module Cardano.Node.LedgerEvent (
   , SlotNo (..)
   , BlockNo (..)
   , StandardCrypto
+  , WithOrigin (..)
   , serialize'
   ) where
 

diff --git a/cardano-node/test/Test/Cardano/Node/LedgerEvent.hs b/cardano-node/test/Test/Cardano/Node/LedgerEvent.hs
@@ -1,5 +1,4 @@
 {-# LANGUAGE DataKinds #-}
-{-# LANGUAGE NamedFieldPuns #-}
 {-# LANGUAGE OverloadedStrings #-}
 {-# LANGUAGE TemplateHaskell #-}
 
@@ -72,8 +71,10 @@ type StakeCredential = Credential 'Staking StandardCrypto
 genAnchoredEvent :: Hedgehog.Gen AnchoredEvent
 genAnchoredEvent =
   AnchoredEvent
-    <$> genBlockHeaderHash
+    <$> (At <$> genBlockHeaderHash)
+    <*> genBlockHeaderHash
     <*> genSlotNo
+    <*> genBlockNo
     <*> Gen.choice (mconcat
       [ fmap LedgerNewEpochEvent <$> genLedgerNewEpochEvent
       , fmap LedgerRewardUpdateEvent <$> genLedgerRewardUpdateEvent
@@ -165,6 +166,10 @@ genSlotNo :: Hedgehog.Gen SlotNo
 genSlotNo =
   fromIntegral <$> Gen.word64 Range.constantBounded
 
+genBlockNo :: Hedgehog.Gen BlockNo
+genBlockNo =
+  fromIntegral <$> Gen.int Range.constantBounded
+
 genStakeDistribution :: Hedgehog.Gen (Map StakeCredential Coin)
 genStakeDistribution =
   genStakeCredentialMap genCoin