New genCardanoKeyWitness function

IntersectMBO · Apr 7, 2023 · 5c4ad0b · 5c4ad0b
1 parent be1dd96
commit 5c4ad0b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cardano-api/gen/Test/Gen/Cardano/Api/Typed.hs b/cardano-api/gen/Test/Gen/Cardano/Api/Typed.hs
@@ -1,4 +1,5 @@
 {-# LANGUAGE GADTs #-}
+{-# LANGUAGE LambdaCase #-}
 {-# LANGUAGE NamedFieldPuns #-}
 {-# LANGUAGE OverloadedStrings #-}
 {-# LANGUAGE RecordWildCards #-}
@@ -18,6 +19,7 @@ module Test.Gen.Cardano.Api.Typed
   , genValueNestedRep
   , genValueNestedBundle
   , genByronKeyWitness
+  , genCardanoKeyWitness
   , genShelleyKeyWitness
 
   , genTxId
@@ -761,6 +763,13 @@ genShelleyWitnessSigningKey =
              , WitnessGenesisUTxOKey <$>  genSigningKey AsGenesisUTxOKey
              ]
 
+genCardanoKeyWitness
+  :: CardanoEra era
+  -> Gen (KeyWitness era)
+genCardanoKeyWitness era = case cardanoEraStyle era of
+  LegacyByronEra -> genByronKeyWitness
+  ShelleyBasedEra _ -> genShelleyWitness era
+
 genSeed :: Int -> Gen Crypto.Seed
 genSeed n = Crypto.mkSeedFromBytes <$> Gen.bytes (Range.singleton n)