-
Notifications
You must be signed in to change notification settings - Fork 213
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add tests for
rollForward
and rollBackward
- Loading branch information
1 parent
859ca67
commit d9d9799
Showing
2 changed files
with
121 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
119 changes: 119 additions & 0 deletions
119
lib/customer-deposit-wallet/test/unit/Cardano/Wallet/Deposit/PureSpec.hs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,119 @@ | ||
{-# LANGUAGE DuplicateRecordFields #-} | ||
{-# OPTIONS_GHC -Wno-incomplete-uni-patterns #-} | ||
{-| | ||
Copyright: © 2024 Cardano Foundation | ||
License: Apache-2.0 | ||
Property tests for the deposit wallet. | ||
-} | ||
module Cardano.Wallet.Deposit.PureSpec | ||
( spec | ||
) where | ||
|
||
import Prelude | ||
|
||
import Cardano.Crypto.Wallet | ||
( XPub | ||
, generate | ||
, toXPub | ||
) | ||
import Test.Hspec | ||
( Spec | ||
, describe | ||
, it | ||
) | ||
import Test.QuickCheck | ||
( Property | ||
, (.&&.) | ||
, (===) | ||
) | ||
|
||
import qualified Cardano.Wallet.Deposit.Pure as Wallet | ||
import qualified Cardano.Wallet.Deposit.Read as Read | ||
import qualified Cardano.Wallet.Deposit.Write as Write | ||
import qualified Data.ByteString.Char8 as B8 | ||
import qualified Data.Map.Strict as Map | ||
|
||
spec :: Spec | ||
spec = do | ||
describe "UTxO availableBalance" $ do | ||
it "rollForward twice" | ||
prop_availableBalance_rollForward_twice | ||
it "rollBackward . rollForward" | ||
prop_availableBalance_rollForward_rollBackward | ||
|
||
{----------------------------------------------------------------------------- | ||
Properties | ||
------------------------------------------------------------------------------} | ||
prop_availableBalance_rollForward_twice :: Property | ||
prop_availableBalance_rollForward_twice = | ||
Wallet.availableBalance w2 === Write.mkAda 3 | ||
where | ||
w0 = emptyWalletWith17Addresses | ||
Just addr1 = Wallet.customerAddress 1 w0 | ||
Just addr2 = Wallet.customerAddress 2 w0 | ||
|
||
tx1 = payFromFaucet [(addr1, Write.mkAda 1)] | ||
block1 = Read.mockNextBlock Read.GenesisPoint [tx1] | ||
chainPoint1 = Read.getChainPoint block1 | ||
w1 = Wallet.rollForwardOne block1 w0 | ||
|
||
tx2 = payFromFaucet [(addr2, Write.mkAda 2)] | ||
block2 = Read.mockNextBlock chainPoint1 [tx2] | ||
w2 = Wallet.rollForwardOne block2 w1 | ||
|
||
prop_availableBalance_rollForward_rollBackward :: Property | ||
prop_availableBalance_rollForward_rollBackward = | ||
Wallet.availableBalance (fst $ Wallet.rollBackward chainPoint0 w2) | ||
=== Wallet.availableBalance w0 | ||
.&&. | ||
Wallet.availableBalance (fst $ Wallet.rollBackward chainPoint1 w2) | ||
=== Wallet.availableBalance w1 | ||
where | ||
w0 = emptyWalletWith17Addresses | ||
Just addr1 = Wallet.customerAddress 1 w0 | ||
Just addr2 = Wallet.customerAddress 2 w0 | ||
chainPoint0 = Read.GenesisPoint | ||
|
||
tx1 = payFromFaucet [(addr1, Write.mkAda 1)] | ||
block1 = Read.mockNextBlock chainPoint0 [tx1] | ||
chainPoint1 = Read.getChainPoint block1 | ||
w1 = Wallet.rollForwardOne block1 w0 | ||
|
||
tx2 = payFromFaucet [(addr2, Write.mkAda 2)] | ||
block2 = Read.mockNextBlock chainPoint1 [tx2] | ||
w2 = Wallet.rollForwardOne block2 w1 | ||
|
||
emptyWalletWith17Addresses :: Wallet.WalletState | ||
emptyWalletWith17Addresses = | ||
Wallet.fromXPubAndGenesis testXPub 17 testGenesis | ||
|
||
testXPub :: XPub | ||
testXPub = | ||
toXPub | ||
$ generate (B8.pack "random seed for a testing xpub lala") B8.empty | ||
|
||
{----------------------------------------------------------------------------- | ||
Test blockchain | ||
------------------------------------------------------------------------------} | ||
|
||
testGenesis :: Read.GenesisData | ||
testGenesis = undefined | ||
|
||
payFromFaucet :: [(Write.Address, Write.Value)] -> Read.Tx | ||
payFromFaucet destinations = | ||
Write.toConwayTx txid tx | ||
where | ||
toTxOut (addr, value) = Write.mkTxOut addr value | ||
txBody = Write.TxBody | ||
{ Write.spendInputs = mempty | ||
, Write.collInputs = mempty | ||
, Write.txouts = | ||
Map.fromList $ zip [toEnum 0..] $ map toTxOut destinations | ||
, Write.collRet = Nothing | ||
} | ||
tx = Write.Tx | ||
{ Write.txbody = txBody | ||
, Write.txwits = () | ||
} | ||
txid = Write.mockTxId txBody |