33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
import logging
|
|
from lbry.wallet.testcase import IntegrationTestCase
|
|
|
|
|
|
class BlockchainReorganizationTests(IntegrationTestCase):
|
|
|
|
VERBOSITY = logging.WARN
|
|
|
|
async def assertBlockHash(self, height):
|
|
self.assertEqual(
|
|
self.ledger.headers.hash(height).decode(),
|
|
await self.blockchain.get_block_hash(height)
|
|
)
|
|
|
|
async def test_reorg(self):
|
|
# invalidate current block, move forward 2
|
|
self.assertEqual(self.ledger.headers.height, 200)
|
|
await self.assertBlockHash(200)
|
|
await self.blockchain.invalidate_block(self.ledger.headers.hash(200).decode())
|
|
await self.blockchain.generate(2)
|
|
await self.ledger.on_header.where(lambda e: e.height == 201)
|
|
self.assertEqual(self.ledger.headers.height, 201)
|
|
await self.assertBlockHash(200)
|
|
await self.assertBlockHash(201)
|
|
|
|
# invalidate current block, move forward 3
|
|
await self.blockchain.invalidate_block(self.ledger.headers.hash(200).decode())
|
|
await self.blockchain.generate(3)
|
|
await self.ledger.on_header.where(lambda e: e.height == 202)
|
|
self.assertEqual(self.ledger.headers.height, 202)
|
|
await self.assertBlockHash(200)
|
|
await self.assertBlockHash(201)
|
|
await self.assertBlockHash(202)
|