Inline putCheckpoint in rollback scenario test

cardano-foundation · Jun 5, 2023 · d171eb2 · d171eb2
1 parent 79889a5
commit d171eb2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/lib/wallet/test/unit/Cardano/Wallet/DB/LayerSpec.hs b/lib/wallet/test/unit/Cardano/Wallet/DB/LayerSpec.hs
@@ -577,6 +577,15 @@ fileModeSpec =  do
                                 (view $ #currentTip . #blockHeight)
                                 epochStability
                                 (currentTip cpB ^. #blockHeight)
+                    let putCheckpoint cp =
+                              Delta.onDBVar walletState
+                            $ Delta.update $ \_ ->
+                                let (prologue, wcp) = WalletState.fromWallet cp
+                                    slot = WalletState.getSlot wcp
+                                in  [ WalletState.UpdateCheckpoints
+                                        [ Checkpoints.PutCheckpoint slot wcp ]
+                                    , WalletState.ReplacePrologue prologue
+                                    ]
 
                     atomically $ do
                         putCheckpoint cpB