mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-02-02 09:46:52 -05:00
test: remove assert_blockchain_height
This commit is contained in:
parent
b401b09355
commit
fa0a4d6c60
1 changed files with 8 additions and 22 deletions
|
@ -29,9 +29,11 @@ Start three nodes:
|
|||
block 200. node2 will reject block 102 since it's assumed valid, but it
|
||||
isn't buried by at least two weeks' work.
|
||||
"""
|
||||
import time
|
||||
|
||||
from test_framework.blocktools import (create_block, create_coinbase)
|
||||
from test_framework.blocktools import (
|
||||
create_block,
|
||||
create_coinbase,
|
||||
)
|
||||
from test_framework.key import ECKey
|
||||
from test_framework.messages import (
|
||||
CBlockHeader,
|
||||
|
@ -79,24 +81,6 @@ class AssumeValidTest(BitcoinTestFramework):
|
|||
assert not p2p_conn.is_connected
|
||||
break
|
||||
|
||||
def assert_blockchain_height(self, node, height):
|
||||
"""Wait until the blockchain is no longer advancing and verify it's reached the expected height."""
|
||||
last_height = node.getblock(node.getbestblockhash())['height']
|
||||
timeout = 10
|
||||
while True:
|
||||
time.sleep(0.25)
|
||||
current_height = node.getblock(node.getbestblockhash())['height']
|
||||
if current_height != last_height:
|
||||
last_height = current_height
|
||||
if timeout < 0:
|
||||
assert False, "blockchain too short after timeout: %d" % current_height
|
||||
timeout -= 0.25
|
||||
continue
|
||||
elif current_height > height:
|
||||
assert False, "blockchain too long: %d" % current_height
|
||||
elif current_height == height:
|
||||
break
|
||||
|
||||
def run_test(self):
|
||||
p2p0 = self.nodes[0].add_p2p_connection(BaseNode())
|
||||
|
||||
|
@ -177,7 +161,8 @@ class AssumeValidTest(BitcoinTestFramework):
|
|||
|
||||
# Send blocks to node0. Block 102 will be rejected.
|
||||
self.send_blocks_until_disconnected(p2p0)
|
||||
self.assert_blockchain_height(self.nodes[0], 101)
|
||||
self.wait_until(lambda: self.nodes[0].getblockcount() >= 101)
|
||||
assert_equal(self.nodes[0].getblockcount(), 101)
|
||||
|
||||
# Send all blocks to node1. All blocks will be accepted.
|
||||
for i in range(2202):
|
||||
|
@ -188,7 +173,8 @@ class AssumeValidTest(BitcoinTestFramework):
|
|||
|
||||
# Send blocks to node2. Block 102 will be rejected.
|
||||
self.send_blocks_until_disconnected(p2p2)
|
||||
self.assert_blockchain_height(self.nodes[2], 101)
|
||||
self.wait_until(lambda: self.nodes[2].getblockcount() >= 101)
|
||||
assert_equal(self.nodes[2].getblockcount(), 101)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
Loading…
Add table
Reference in a new issue