mock-block: more precise types

Author: nfrisby

ouroboros-consensus/src/unstable-mock-block/Ouroboros/Consensus/Mock/Ledger/Block.hs

@@ -440,7 +440,7 @@ instance TxLimits (SimpleBlock c ext) where
   type TxMeasure (SimpleBlock c ext) = SizeInBytes
 
   blockTxCapacity _cfg _st = simpleBlockTxCapacity
-  txInBlockSize   _cfg _st = SizeInBytes . txSize
+  txInBlockSize   _cfg _st = txSize
 
   txMeasureBytes _ = id
 
@@ -489,8 +489,8 @@ mkSimpleGenTx tx = SimpleGenTx
     , simpleGenTxId = Hash.hashWithSerialiser toCBOR tx
     }
 
-txSize :: GenTx (SimpleBlock c ext) -> Word32
-txSize = fromIntegral . Lazy.length . serialise
+txSize :: GenTx (SimpleBlock c ext) -> SizeInBytes
+txSize = SizeInBytes . fromIntegral . Lazy.length . serialise
 
 {-------------------------------------------------------------------------------
   Support for QueryLedger