Fix compilation errors (to be merged back)

cardano-foundation · Jun 2, 2023 · 1ccc89c · 1ccc89c
1 parent 1918bc0
commit 1ccc89c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 9 deletions.
diff --git a/lib/wallet/src/Cardano/Wallet/DB/Store/WalletState/Store.hs b/lib/wallet/src/Cardano/Wallet/DB/Store/WalletState/Store.hs
@@ -25,6 +25,8 @@ import Cardano.Wallet.DB.Store.Submissions.Operations
     ( mkStoreSubmissions )
 import Cardano.Wallet.DB.WalletState
     ( DeltaWalletState, DeltaWalletState1 (..), WalletState (..) )
+import Cardano.Wallet.Flavor
+    ( WalletFlavorS, keyOfWallet )
 import Control.Monad.Class.MonadThrow
     ( throwIO )
 import Data.Generics.Internal.VL.Lens
@@ -80,7 +82,7 @@ mkStoreWallet wF wid = mkUpdateStore load write update
 
     update = updateLoad load throwIO $ updateSequence update1
       where
-        update1 _ (ReplacePrologue prologue) = insertPrologue wid prologue
+        update1 _ (ReplacePrologue prologue') = insertPrologue wid prologue'
         update1 s (UpdateCheckpoints delta) =
             updateS checkpointsStore (Just $ checkpoints s) delta
         update1 s (UpdateSubmissions deltas) =

diff --git a/lib/wallet/test/unit/Cardano/Wallet/DB/Store/WalletState/StoreSpec.hs b/lib/wallet/test/unit/Cardano/Wallet/DB/Store/WalletState/StoreSpec.hs
@@ -12,10 +12,8 @@ import Prelude
 
 import Cardano.DB.Sqlite
     ( ForeignKeysSetting (..), SqliteContext (runQuery) )
-import Cardano.Wallet.Address.Derivation.Shelley
-    ( ShelleyKey )
-import Cardano.Wallet.Address.Discovery.Sequential
-    ( SeqState )
+import Cardano.Wallet.Address.Derivation
+    ( Depth (..) )
 import Cardano.Wallet.Checkpoints
     ( DeltaCheckpoints (..) )
 import Cardano.Wallet.DB.Arbitrary
@@ -77,6 +75,8 @@ import Test.QuickCheck.Monadic
 import UnliftIO.Exception
     ( impureThrow )
 
+import Cardano.Address.Derivation
+    ( XPrv )
 import qualified Data.Map.Strict as Map
 
 spec :: Spec
@@ -92,12 +92,13 @@ spec = do
 prop_StoreWallet
     :: forall s
      . ( PersistAddressBook s
-       , GenState s
+       , GenState s, Eq (KeyOf s 'RootK XPrv)
        )
-    => SqliteContext
+    => WalletFlavorS s
+    -> SqliteContext
     -> (WalletId, InitialCheckpoint s)
     -> Property
-prop_StoreWallet db (wid, InitialCheckpoint cp0) =
+prop_StoreWallet wF db (wid, InitialCheckpoint cp0) =
     monadicIO (setup >> prop)
   where
     toIO = runQuery db
@@ -111,7 +112,7 @@ prop_StoreWallet db (wid, InitialCheckpoint cp0) =
     prop = do
         prop_StoreUpdates
             toIO
-            (mkStoreWallet wid)
+            (mkStoreWallet wF wid)
             genState
             genDeltaWalletState