Add missing tx policies

IntersectMBO · Aug 12, 2024 · f434785 · f434785
1 parent b2e3db6
commit f434785
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/ouroboros-network/sim-tests-lib/Test/Ouroboros/Network/TxSubmission/Common.hs b/ouroboros-network/sim-tests-lib/Test/Ouroboros/Network/TxSubmission/Common.hs
@@ -842,9 +842,11 @@ prop_receivedTxIds_generator (ArbReceivedTxIds _ someTxsToAck _peeraddr _ps st)
 -- `prop_makeDecisions_acknowledged`.
 --
 prop_acknowledgeTxIds :: ArbReceivedTxIds
+                      -> ArbDecisionContextWithReceivedTxIds
                       -> Property
-prop_acknowledgeTxIds (ArbReceivedTxIds _mempoolHasTxFun _txs _peeraddr ps st) =
-    case TXS.acknowledgeTxIds undefined st ps of
+prop_acknowledgeTxIds (ArbReceivedTxIds _ _ _ ps _)
+                      (ArbDecisionContextWithReceivedTxIds policy SharedDecisionContext { sdcSharedTxState = st } _ _ _) =
+    case TXS.acknowledgeTxIds policy st ps of
       (numTxIdsToAck, _, txs, TXS.RefCountDiff { TXS.txIdsToAck }, ps') ->
              counterexample "number of tx ids to ack must agree with RefCountDiff"
              ( fromIntegral numTxIdsToAck
@@ -879,10 +881,12 @@ prop_acknowledgeTxIds (ArbReceivedTxIds _mempoolHasTxFun _txs _peeraddr ps st) =
 --
 prop_hasTxIdsToAcknowledge
   :: ArbReceivedTxIds
+  -> ArbTxDecisionPolicy
   -> Property
-prop_hasTxIdsToAcknowledge (ArbReceivedTxIds _mempoolHasTxFun _txs _peeraddr ps st) =
+prop_hasTxIdsToAcknowledge (ArbReceivedTxIds _mempoolHasTxFun _txs _peeraddr ps st)
+                           (ArbTxDecisionPolicy policy) =
     case ( TXS.hasTxIdsToAcknowledge st ps
-         , TXS.acknowledgeTxIds undefined st ps
+         , TXS.acknowledgeTxIds policy st ps
          ) of
       (canAck, (numTxIdsToAck, _, _, _, _)) ->
         canAck === (numTxIdsToAck > 0)
@@ -1612,7 +1616,7 @@ prop_makeDecisions_acknowledged
 
         ackFromState :: Map PeerAddr NumTxIdsToAck
         ackFromState =
-            Map.map (\ps -> case TXS.acknowledgeTxIds undefined sharedTxState ps of
+            Map.map (\ps -> case TXS.acknowledgeTxIds policy sharedTxState ps of
                               (a, _, _, _, _) -> a)
           . peerTxStates
           $ sharedTxState