New genCardanoKeyWitness function

IntersectMBO · Apr 11, 2023 · 0ad15f2 · 0ad15f2
1 parent e60ac65
commit 0ad15f2
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)