refactoring #8 | define single block generator from previous block

cardano-foundation · Mar 11, 2019 · 2ac72be · 2ac72be
1 parent 842775d
commit 2ac72be
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/test/unit/Cardano/Wallet/BlockSyncerSpec.hs b/test/unit/Cardano/Wallet/BlockSyncerSpec.hs
@@ -105,10 +105,28 @@ mkConsecutiveTestBlocks blockNum =
     loop blockNum []
   where
     bytelistGenerator = pack <$> vector 10 :: Gen ByteString
+
+    fromPreviousBlock
+        :: (Hash "BlockHeader", Block)
+        -> Gen (Hash "BlockHeader", Block)
+    fromPreviousBlock (h, b) = do
+        h' <- Hash <$> bytelistGenerator
+        return
+            ( h'
+            , Block
+                { header = BlockHeader
+                    { epochIndex = epochIndex (header b)
+                    , slotNumber = slotNumber (header b) + 1
+                    , prevBlockHash = h
+                    }
+                , transactions = mempty
+                }
+            )
+
     loop
         :: Int
-        -> [((Hash "BlockHeader"),Block)]
-        -> IO [((Hash "BlockHeader"),Block)]
+        -> [(Hash "BlockHeader", Block)]
+        -> IO [(Hash "BlockHeader", Block)]
     loop n res
         | n <= 0 = return $ reverse res
         | res == mempty = do