Add UTxO.size

Author: locallycompact

hydra-cardano-api/src/Cardano/Api/UTxO.hs

@@ -103,6 +103,10 @@ txOutputs = Map.elems . toMap
 null :: UTxO -> Bool
 null = Map.null . toMap
 
+
+size :: UTxO -> Int
+size = Map.size . toMap
+
 -- * Type Conversions
 
 -- | Transforms a UTxO containing tx outs from any era into Babbage era.

hydra-cluster/test/Test/Hydra/Cluster/FaucetSpec.hs

@@ -34,7 +34,7 @@ spec =
           vk <- generate genVerificationKey
           seedFromFaucet node vk 1_000_000 tracer
         -- 10 unique outputs
-        length (UTxO.txOutputs $ fold utxos) `shouldBe` 10
+        UTxO.size (fold utxos) `shouldBe` 10
 
     describe "returnFundsToFaucet" $ do
       it "does nothing if nothing to return" $ \(tracer, node) -> do
@@ -54,7 +54,7 @@ spec =
               UTxO.foldMap txOutValue remaining `shouldBe` mempty
 
               -- check the faucet has one utxo extra in the end
-              length (UTxO.txOutputs finalFaucetFunds) `shouldBe` length (UTxO.txOutputs initialFaucetFunds) + 1
+              UTxO.size finalFaucetFunds `shouldBe` UTxO.size initialFaucetFunds + 1
 
               let initialFaucetValue = selectLovelace (UTxO.foldMap txOutValue initialFaucetFunds)
               let finalFaucetValue = selectLovelace (UTxO.foldMap txOutValue finalFaucetFunds)
@@ -75,4 +75,4 @@ spec =
         -- it squashed the UTxO
         utxoAfter <- queryUTxOFor networkId nodeSocket QueryTip vk
 
-        length (UTxO.txOutputs utxoAfter) `shouldBe` 1
+        UTxO.size utxoAfter `shouldBe` 1

hydra-node/bench/tx-cost/TxCost.hs

@@ -102,7 +102,7 @@ computeCommitCost = do
       Right tx ->
         case checkSizeAndEvaluate tx (utxo <> knownUtxo) of
           Just (txSize, memUnit, cpuUnit, minFee) ->
-            pure $ Just (NumUTxO $ length $ UTxO.txOutputs utxo, txSize, memUnit, cpuUnit, minFee)
+            pure $ Just (NumUTxO $ UTxO.size utxo, txSize, memUnit, cpuUnit, minFee)
           Nothing ->
             pure Nothing
 

hydra-node/test/Hydra/Chain/Direct/StateSpec.hs

@@ -476,7 +476,7 @@ prop_observeAnyTx =
 -- | Given a UTxO with more than one entry, we can split it into two non-empty UTxO.
 prop_splitUTxO :: UTxO -> Property
 prop_splitUTxO utxo =
-  (length (UTxO.txOutputs utxo) > 1) ==>
+  (UTxO.size utxo > 1) ==>
     let (inHead, toDecommit) = splitUTxO utxo
      in conjoin
           [ not (UTxO.null inHead) & counterexample "inHead is empty"

hydra-tx/src/Hydra/Tx/Decrement.hs

@@ -55,7 +55,7 @@ decrementTx scriptRegistry vk headId headParameters (headInput, headOutput) snap
           { signature = toPlutusSignatures signatures
           , snapshotNumber = fromIntegral number
           , numberOfDecommitOutputs =
-              fromIntegral $ length $ maybe [] UTxO.txOutputs utxoToDecommit
+              fromIntegral $ maybe 0 UTxO.size utxoToDecommit
           }
 
   utxoHash = toBuiltin $ hashUTxO @Tx utxo

hydra-tx/src/Hydra/Tx/Fanout.hs

@@ -52,7 +52,7 @@ fanoutTx scriptRegistry utxo utxoToCommit utxoToDecommit (headInput, headOutput)
   headRedeemer =
     toScriptData $
       Head.Fanout
-        { numberOfFanoutOutputs = fromIntegral $ length $ UTxO.txOutputs utxo
+        { numberOfFanoutOutputs = fromIntegral $ UTxO.size utxo
         , numberOfCommitOutputs = fromIntegral $ length orderedTxOutsToCommit
         , numberOfDecommitOutputs = fromIntegral $ length orderedTxOutsToDecommit
         }

hydra-tx/test/Hydra/Tx/Contract/ContractSpec.hs

@@ -227,7 +227,7 @@ prop_consistentHashPreSerializedCommits =
 prop_hashingCaresAboutOrderingOfTxOuts :: Property
 prop_hashingCaresAboutOrderingOfTxOuts =
   forAllShrink genUTxOWithSimplifiedAddresses shrinkUTxO $ \(utxo :: UTxO) ->
-    (length (UTxO.txOutputs utxo) > 1) ==>
+    (UTxO.size utxo > 1) ==>
       let plutusTxOuts =
             rights $
               zipWith

hydra-tx/test/Hydra/Tx/Contract/Decrement.hs

@@ -184,7 +184,7 @@ genDecrementMutation (tx, _utxo) =
             Head.DecrementRedeemer
               { signature = invalidSignature
               , snapshotNumber = fromIntegral healthySnapshotNumber
-              , numberOfDecommitOutputs = fromIntegral $ maybe 0 (length . UTxO.txOutputs) $ utxoToDecommit healthySnapshot
+              , numberOfDecommitOutputs = fromIntegral $ maybe 0 UTxO.size $ utxoToDecommit healthySnapshot
               }
     , -- Spec: Transaction is signed by a participant
       SomeMutation (pure $ toErrorCode SignerIsNotAParticipant) AlterRequiredSigner <$> do

hydra-tx/test/Hydra/Tx/Contract/FanOut.hs

@@ -67,7 +67,7 @@ healthyFanoutTx =
 
 healthyFanoutUTxO :: UTxO
 healthyFanoutUTxO =
-  UTxO.map adaOnly $ generateWith (genUTxOWithSimplifiedAddresses `suchThat` \u -> length (UTxO.txOutputs u) > 1) 42
+  UTxO.map adaOnly $ generateWith (genUTxOWithSimplifiedAddresses `suchThat` \u -> UTxO.size u > 1) 42
 
 healthySlotNo :: SlotNo
 healthySlotNo = arbitrary `generateWith` 42