Merge pull request #5071 from input-output-hk/newhoggy/new-genCardano…

…KeyWitness-function New genCardanoKeyWitness function
IntersectMBO · Apr 11, 2023 · 2dd271b · 2dd271b
2 parents 68186b9 + 0ad15f2
commit 2dd271b
Showing 1 changed file with 8 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
@@ -18,6 +18,7 @@ module Test.Gen.Cardano.Api.Typed
   , genValueNestedRep
   , genValueNestedBundle
   , genByronKeyWitness
+  , genCardanoKeyWitness
   , genShelleyKeyWitness
 
   , genTxId
@@ -761,6 +762,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)