diff --git a/test/unit/Cardano/Wallet/BlockSyncerSpec.hs b/test/unit/Cardano/Wallet/BlockSyncerSpec.hs index 6c4acd9e2b2..79190201ed6 100644 --- a/test/unit/Cardano/Wallet/BlockSyncerSpec.hs +++ b/test/unit/Cardano/Wallet/BlockSyncerSpec.hs @@ -104,20 +104,11 @@ mkConsecutiveTestBlocks -- ^ number of consecutive blocks to create -> [(Hash "BlockHeader", Block)] -- ^ returns block paired with generated hashes starting from the oldest -mkConsecutiveTestBlocks = - let - prev = Hash "initial block" - h = BlockHeader 1 0 prev - in - loop [(blockHeaderHash h, Block h mempty)] +mkConsecutiveTestBlocks n = + reverse $ take n $ iterate next (blockHeaderHash h0, Block h0 mempty) where - loop - :: [(Hash "BlockHeader", Block)] - -> Int - -> [(Hash "BlockHeader", Block)] - loop (block : blocks) n - | n <= 0 = reverse (block : blocks) - | otherwise = loop (next block : block : blocks) (n -1) + h0 :: BlockHeader + h0 = BlockHeader 1 0 (Hash "initial block") next :: (Hash "BlockHeader", Block) @@ -132,6 +123,7 @@ mkConsecutiveTestBlocks = + blockHeaderHash :: BlockHeader -> Hash "BlockHeader" blockHeaderHash = Hash . CBOR.toStrictByteString . encodeBlockHeader