-
Notifications
You must be signed in to change notification settings - Fork 212
/
Gen.hs
108 lines (97 loc) · 2.95 KB
/
Gen.hs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
{-# LANGUAGE TypeFamilies #-}
-- |
-- Copyright: © 2022 IOHK
-- License: Apache-2.0
--
module Cardano.Wallet.Write.Tx.Gen
( genDatum
, genBinaryData
, genDatumHash
, shrinkBinaryData
, shrinkDatum
, genTxOut
)
where
import Prelude
import Cardano.Ledger.Alonzo.Data
( Data (..), dataToBinaryData )
import Cardano.Wallet.Write.Tx
( BinaryData
, Datum (..)
, DatumHash
, LatestLedgerEra
, RecentEra
, ShelleyLedgerEra
, TxOut
, cardanoEraFromRecentEra
, datumFromCardanoScriptData
, datumHashFromBytes
, datumToCardanoScriptData
, shelleyBasedEraFromRecentEra
)
import Data.ByteString
( ByteString )
import Data.Maybe
( fromMaybe )
import Ouroboros.Consensus.Cardano.Block
( EraCrypto, StandardCrypto )
import Test.QuickCheck
( Gen
, arbitrary
, choose
, listOf
, oneof
, scale
, shrinkMapBy
, sized
, vector
, vectorOf
)
import qualified Cardano.Api.Gen as Cardano
import qualified Cardano.Api.Shelley as Cardano
import qualified Data.ByteString as BS
import qualified PlutusLedgerApi.V1 as PV1
genDatum :: (EraCrypto era ~ StandardCrypto) => Gen (Datum era)
genDatum = oneof
[ Datum <$> genBinaryData
, DatumHash <$> genDatumHash
, pure NoDatum
]
-- Originally from https://github.com/input-output-hk/cardano-ledger/blob/c7c63dabdb215ebdaed8b63274965966f2bf408f/eras/alonzo/test-suite/src/Test/Cardano/Ledger/Alonzo/Serialisation/Generators.hs#L66-L79
genBinaryData :: Gen (BinaryData era)
genBinaryData = dataToBinaryData . Data <$> scale (`div` 10) (sized gendata)
where
gendata n | n > 0 = oneof
[ PV1.I <$> arbitrary
, PV1.B <$> genByteString
, PV1.Map
<$> listOf ((,) <$> gendata (n `div` 2) <*> gendata (n `div` 2))
, PV1.Constr
<$> fmap abs (arbitrary :: Gen Integer)
<*> listOf (gendata (n `div` 2))
, PV1.List
<$> listOf (gendata (n `div` 2))
]
gendata _ = oneof [PV1.I <$> arbitrary, PV1.B <$> genByteString]
shrinkDatum :: Datum LatestLedgerEra -> [Datum LatestLedgerEra]
shrinkDatum (Datum x) = NoDatum : map Datum (shrinkBinaryData x)
shrinkDatum (DatumHash _) = [NoDatum]
shrinkDatum NoDatum = []
shrinkBinaryData :: BinaryData era -> [BinaryData era]
shrinkBinaryData = shrinkMapBy
datumFromCardanoScriptData
datumToCardanoScriptData $
shrinkMapBy
Cardano.unsafeHashableScriptData
Cardano.getScriptData
Cardano.shrinkScriptData
genDatumHash :: Gen DatumHash
genDatumHash =
fromMaybe (error "genDatumHash should always generate valid hashes")
. datumHashFromBytes
. BS.pack <$> vectorOf 32 arbitrary
genByteString :: Gen ByteString
genByteString = BS.pack <$> (choose (0, 64) >>= vector)
genTxOut :: RecentEra era -> Gen (TxOut (ShelleyLedgerEra era))
genTxOut era = Cardano.toShelleyTxOut (shelleyBasedEraFromRecentEra era)
<$> Cardano.genTxOut (cardanoEraFromRecentEra era)