Add mock funding to deposit wallet

cardano-foundation · Nov 10, 2024 · da3ff75 · da3ff75
1 parent e29b2d2
commit da3ff75
Show file tree

Hide file tree

Showing 5 changed files with 78 additions and 8 deletions.
diff --git a/lib/customer-deposit-wallet/customer-deposit-wallet.cabal b/lib/customer-deposit-wallet/customer-deposit-wallet.cabal
@@ -178,6 +178,7 @@ library rest
     , customer-deposit-wallet-pure
     , deepseq
     , delta-store
+    , delta-types
     , directory
     , filepath
     , memory

diff --git a/lib/customer-deposit-wallet/rest/Cardano/Wallet/Deposit/REST.hs b/lib/customer-deposit-wallet/rest/Cardano/Wallet/Deposit/REST.hs
@@ -49,6 +49,9 @@ module Cardano.Wallet.Deposit.REST
     , walletPublicIdentity
     , deleteWallet
     , deleteTheDepositWalletOnDisk
+    -- * Internals
+    , onWalletInstance
+
     ) where
 
 import Prelude

diff --git a/lib/customer-deposit-wallet/rest/Cardano/Wallet/Deposit/REST/Start.hs b/lib/customer-deposit-wallet/rest/Cardano/Wallet/Deposit/REST/Start.hs
@@ -1,42 +1,78 @@
+{-# LANGUAGE NumericUnderscores #-}
+
 module Cardano.Wallet.Deposit.REST.Start
     ( loadDepositWalletFromDisk
     , newFakeBootEnv
+    , mockFundTheWallet
     )
 where
 
 import Prelude
 
 import Cardano.Wallet.Deposit.IO
     ( WalletBootEnv (..)
+    , onWalletState
     )
 import Cardano.Wallet.Deposit.IO.Network.Mock
     ( newNetworkEnvMock
     )
 import Cardano.Wallet.Deposit.IO.Network.Type
     ( mapBlock
     )
+import Cardano.Wallet.Deposit.Pure
+    ( customerAddress
+    , rollForwardOne
+    )
+import Cardano.Wallet.Deposit.Read
+    ( ChainPoint (..)
+    , EraValue (..)
+    , mockNextBlock
+    )
 import Cardano.Wallet.Deposit.REST
-    ( WalletResource
+    ( ErrWalletResource
+    , WalletResource
     , loadWallet
+    , onWalletInstance
     , runWalletResourceM
     , walletExists
     )
+import Cardano.Wallet.Deposit.Write
+    ( addTxOut
+    , emptyTxBody
+    , mkAda
+    , mkTx
+    , mkTxOut
+    )
+import Control.Concurrent
+    ( threadDelay
+    )
 import Control.Monad
     ( when
     )
 import Control.Monad.IO.Class
     ( MonadIO (..)
     )
+import Control.Monad.Trans.Except
+    ( ExceptT (..)
+    , runExceptT
+    )
 import Control.Tracer
     ( Tracer
     , stdoutTracer
     , traceWith
     )
+import qualified Data.Delta as Delta
+    ( Replace (..)
+    )
 import Data.Functor.Contravariant
     ( (>$<)
     )
+import Data.Maybe
+    ( fromMaybe
+    )
 
 import qualified Cardano.Wallet.Deposit.Read as Read
+import qualified Data.Delta.Update as Delta
 
 lg :: (MonadIO m, Show a) => Tracer IO String -> String -> a -> m ()
 lg tr p x = liftIO $ traceWith tr $ p <> ": " <> show x
@@ -48,16 +84,43 @@ loadDepositWalletFromDisk
     -> WalletResource
     -> IO ()
 loadDepositWalletFromDisk tr dir env resource = do
-    result <- flip runWalletResourceM resource $ do
-        test <- walletExists dir
-        when test $ do
-            lg tr "Loading wallet from" dir
-            loadWallet env dir
-            lg tr "Wallet loaded from" dir
+    result <- runExceptT $ do
+        ExceptT $ flip runWalletResourceM resource $ do
+            test <- walletExists dir
+            when test $ do
+                lg tr "Loading wallet from" dir
+                loadWallet env dir
+                lg tr "Wallet loaded from" dir
+        liftIO $ threadDelay 1_000_000
+        ExceptT $ mockFundTheWallet resource
     case result of
         Left e -> error $ show e
         Right _ -> pure ()
 
+mockFundTheWallet :: WalletResource -> IO (Either ErrWalletResource ())
+mockFundTheWallet resource = do
+    flip runWalletResourceM resource $ do
+        onWalletInstance $ \w ->
+            onWalletState w
+                $ Delta.update
+                $ Delta.Replace . f
+  where
+    f s =
+        let
+            address =
+                fromMaybe
+                    (error "no address where it should be")
+                    $ customerAddress 0 s
+            tx =
+                mkTx
+                    $ fst
+                    $ addTxOut
+                        (mkTxOut address (mkAda 1_000_000))
+                        emptyTxBody
+            block = mockNextBlock GenesisPoint [tx]
+        in
+            rollForwardOne (EraValue block) s
+
 newFakeBootEnv :: IO (WalletBootEnv IO)
 newFakeBootEnv =
     WalletBootEnv

diff --git a/lib/customer-deposit-wallet/src/Cardano/Wallet/Deposit/IO.hs b/lib/customer-deposit-wallet/src/Cardano/Wallet/Deposit/IO.hs
@@ -44,6 +44,9 @@ module Cardano.Wallet.Deposit.IO
       -- *** Submit transactions
     , submitTx
     , listTxsInSubmission
+
+     -- * Internals
+     , onWalletState
     ) where
 
 import Prelude

diff --git a/lib/customer-deposit-wallet/src/Cardano/Wallet/Deposit/Write.hs b/lib/customer-deposit-wallet/src/Cardano/Wallet/Deposit/Write.hs
@@ -39,8 +39,8 @@ module Cardano.Wallet.Deposit.Write
     , mkTxOut
     , toConwayTx
     , addTxIn
-    , emptyTxBody
     , addTxOut
+    , emptyTxBody
     ) where
 
 import Prelude