mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-02-10 10:52:31 -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
|
block 200. node2 will reject block 102 since it's assumed valid, but it
|
||||||
isn't buried by at least two weeks' work.
|
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.key import ECKey
|
||||||
from test_framework.messages import (
|
from test_framework.messages import (
|
||||||
CBlockHeader,
|
CBlockHeader,
|
||||||
|
@ -79,24 +81,6 @@ class AssumeValidTest(BitcoinTestFramework):
|
||||||
assert not p2p_conn.is_connected
|
assert not p2p_conn.is_connected
|
||||||
break
|
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):
|
def run_test(self):
|
||||||
p2p0 = self.nodes[0].add_p2p_connection(BaseNode())
|
p2p0 = self.nodes[0].add_p2p_connection(BaseNode())
|
||||||
|
|
||||||
|
@ -177,7 +161,8 @@ class AssumeValidTest(BitcoinTestFramework):
|
||||||
|
|
||||||
# Send blocks to node0. Block 102 will be rejected.
|
# Send blocks to node0. Block 102 will be rejected.
|
||||||
self.send_blocks_until_disconnected(p2p0)
|
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.
|
# Send all blocks to node1. All blocks will be accepted.
|
||||||
for i in range(2202):
|
for i in range(2202):
|
||||||
|
@ -188,7 +173,8 @@ class AssumeValidTest(BitcoinTestFramework):
|
||||||
|
|
||||||
# Send blocks to node2. Block 102 will be rejected.
|
# Send blocks to node2. Block 102 will be rejected.
|
||||||
self.send_blocks_until_disconnected(p2p2)
|
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__':
|
if __name__ == '__main__':
|
||||||
|
|
Loading…
Add table
Reference in a new issue