Align style of generate() with generate_and_wait().

This commit is contained in:
Jonathan Moody 2022-09-08 08:52:20 -05:00 committed by Lex Berezhny
parent 45a255e7a2
commit 78a28de2aa

View file

@ -327,9 +327,9 @@ class IntegrationTestCase(AsyncioTestCase):
async def generate(self, blocks):
""" Ask lbrycrd to generate some blocks and wait until ledger has them. """
prepare = self.ledger.on_header.where(self.blockchain.is_expected_block)
height = self.blockchain.block_expected + blocks
self.conductor.spv_node.server.synchronized.clear()
await self.blockchain.generate(blocks)
height = self.blockchain.block_expected
await prepare # no guarantee that it didn't happen already, so start waiting from before calling generate
while True:
await self.conductor.spv_node.server.synchronized.wait()