Fix observation test for increment

cardano-scaling · Oct 9, 2024 · 3383deb · 3383deb
1 parent c6226d1
commit 3383deb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/hydra-node/src/Hydra/Chain/Direct/State.hs b/hydra-node/src/Hydra/Chain/Direct/State.hs
@@ -1197,20 +1197,21 @@ genRecoverTx = do
 
 genIncrementTx :: Int -> Gen (ChainContext, [TxOut CtxUTxO], OpenState, UTxO, Tx)
 genIncrementTx numParties = do
-  (utxo, txDeposit) <- genDepositTx
+  (_utxo, txDeposit) <- genDepositTx
   ctx <- genHydraContextFor numParties
   cctx <- pickChainContext ctx
   let DepositObservation{deposited, depositTxId} = fromJust $ observeDepositTx (ctxNetworkId ctx) txDeposit
   (_, st@OpenState{headId}) <- genStOpen ctx
   let openUTxO = getKnownUTxO st
   let version = 1
   snapshot <- genConfirmedSnapshot headId 2 version openUTxO (Just deposited) Nothing (ctxHydraSigningKeys ctx)
+  let depositUTxO = utxoFromTx txDeposit
   pure
     ( cctx
     , maybe mempty toList (utxoToCommit $ getSnapshot snapshot)
     , st
-    , utxo
-    , unsafeIncrement cctx openUTxO headId (ctxHeadParameters ctx) snapshot depositTxId
+    , depositUTxO
+    , unsafeIncrement cctx (openUTxO <> depositUTxO) headId (ctxHeadParameters ctx) snapshot depositTxId
     )
 
 genDecrementTx :: Int -> Gen (ChainContext, [TxOut CtxUTxO], OpenState, UTxO, Tx)