Skip to content

Commit

Permalink
[tests] Use network_thread_start() in tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
jnewbery committed Dec 11, 2017
1 parent 5fc6e71 commit 74e64f2
Show file tree
Hide file tree
Showing 22 changed files with 37 additions and 34 deletions.
2 changes: 1 addition & 1 deletion test/functional/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ contains the higher level logic for processing P2P payloads and connecting to
the Bitcoin Core node application logic. For custom behaviour, subclass the
P2PInterface object and override the callback methods.

- Call `NetworkThread.start()` after all `P2PInterface` objects are created to
- Call `network_thread_start()` after all `P2PInterface` objects are created to
start the networking thread. (Continue with the test logic in your existing
thread.)

Expand Down
6 changes: 3 additions & 3 deletions test/functional/assumevalid.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,10 +38,10 @@
CTransaction,
CTxIn,
CTxOut,
NetworkThread,
P2PInterface,
msg_block,
msg_headers)
msg_headers,
network_thread_start)
from test_framework.script import (CScript, OP_TRUE)
from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import assert_equal
Expand Down Expand Up @@ -98,7 +98,7 @@ def run_test(self):
# Connect to node0
p2p0 = self.nodes[0].add_p2p_connection(BaseNode())

NetworkThread().start() # Start up network handling in another thread
network_thread_start()
self.nodes[0].p2p.wait_for_verack()

# Build the blockchain
Expand Down
2 changes: 1 addition & 1 deletion test/functional/bip65-cltv-p2p.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ def set_test_params(self):
def run_test(self):
self.nodes[0].add_p2p_connection(P2PInterface())

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

# wait_for_verack ensures that the P2P connection is fully up.
self.nodes[0].p2p.wait_for_verack()
Expand Down
4 changes: 2 additions & 2 deletions test/functional/bip68-112-113-p2p.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@

from test_framework.test_framework import ComparisonTestFramework
from test_framework.util import *
from test_framework.mininode import ToHex, CTransaction, NetworkThread
from test_framework.mininode import ToHex, CTransaction, network_thread_start
from test_framework.blocktools import create_coinbase, create_block
from test_framework.comptool import TestInstance, TestManager
from test_framework.script import *
Expand Down Expand Up @@ -100,7 +100,7 @@ def set_test_params(self):
def run_test(self):
test = TestManager(self, self.options.tmpdir)
test.add_all_connections(self.nodes)
NetworkThread().start() # Start up network handling in another thread
network_thread_start()
test.run()

def send_generic_input_tx(self, node, coinbases):
Expand Down
6 changes: 3 additions & 3 deletions test/functional/bip9-softforks.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@

from test_framework.test_framework import ComparisonTestFramework
from test_framework.util import *
from test_framework.mininode import CTransaction, NetworkThread
from test_framework.mininode import CTransaction, network_thread_start
from test_framework.blocktools import create_coinbase, create_block
from test_framework.comptool import TestInstance, TestManager
from test_framework.script import CScript, OP_1NEGATE, OP_CHECKSEQUENCEVERIFY, OP_DROP
Expand All @@ -36,7 +36,7 @@ def set_test_params(self):
def run_test(self):
self.test = TestManager(self, self.options.tmpdir)
self.test.add_all_connections(self.nodes)
NetworkThread().start() # Start up network handling in another thread
network_thread_start()
self.test.run()

def create_transaction(self, node, coinbase, to_address, amount):
Expand Down Expand Up @@ -245,7 +245,7 @@ def test_BIP(self, bipName, activated_version, invalidate, invalidatePostSignatu
self.setup_chain()
self.setup_network()
self.test.add_all_connections(self.nodes)
NetworkThread().start()
network_thread_start()
self.test.p2p_connections[0].wait_for_verack()

def get_tests(self):
Expand Down
2 changes: 1 addition & 1 deletion test/functional/bipdersig-p2p.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def set_test_params(self):
def run_test(self):
self.nodes[0].add_p2p_connection(P2PInterface())

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

# wait_for_verack ensures that the P2P connection is fully up.
self.nodes[0].p2p.wait_for_verack()
Expand Down
4 changes: 2 additions & 2 deletions test/functional/example_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,11 +17,11 @@
from test_framework.blocktools import (create_block, create_coinbase)
from test_framework.mininode import (
CInv,
NetworkThread,
P2PInterface,
mininode_lock,
msg_block,
msg_getdata,
network_thread_start,
)
from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import (
Expand Down Expand Up @@ -136,7 +136,7 @@ def run_test(self):

# Start up network handling in another thread. This needs to be called
# after the P2P connections have been created.
NetworkThread().start()
network_thread_start()
# wait_for_verack ensures that the P2P connection is fully up.
self.nodes[0].p2p.wait_for_verack()

Expand Down
3 changes: 2 additions & 1 deletion test/functional/invalidblockrequest.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
from test_framework.util import *
from test_framework.comptool import TestManager, TestInstance, RejectResult
from test_framework.blocktools import *
from test_framework.mininode import network_thread_start
import copy
import time

Expand All @@ -32,7 +33,7 @@ def run_test(self):
test.add_all_connections(self.nodes)
self.tip = None
self.block_time = None
NetworkThread().start() # Start up network handling in another thread
network_thread_start()
test.run()

def get_tests(self):
Expand Down
2 changes: 1 addition & 1 deletion test/functional/invalidtxrequest.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def run_test(self):
test.add_all_connections(self.nodes)
self.tip = None
self.block_time = None
NetworkThread().start() # Start up network handling in another thread
network_thread_start()
test.run()

def get_tests(self):
Expand Down
4 changes: 2 additions & 2 deletions test/functional/maxuploadtarget.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ def run_test(self):
for _ in range(3):
p2p_conns.append(self.nodes[0].add_p2p_connection(TestNode()))

NetworkThread().start() # Start up network handling in another thread
network_thread_start()
for p2pc in p2p_conns:
p2pc.wait_for_verack()

Expand Down Expand Up @@ -149,7 +149,7 @@ def run_test(self):
# Reconnect to self.nodes[0]
self.nodes[0].add_p2p_connection(TestNode())

NetworkThread().start() # Start up network handling in another thread
network_thread_start()
self.nodes[0].p2p.wait_for_verack()

#retrieve 20 blocks which should be enough to break the 1MB limit
Expand Down
4 changes: 2 additions & 2 deletions test/functional/nulldummy.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@

from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import *
from test_framework.mininode import CTransaction, NetworkThread
from test_framework.mininode import CTransaction, network_thread_start
from test_framework.blocktools import create_coinbase, create_block, add_witness_commitment
from test_framework.script import CScript
from io import BytesIO
Expand Down Expand Up @@ -50,7 +50,7 @@ def run_test(self):
self.wit_address = self.nodes[0].addwitnessaddress(self.address)
self.wit_ms_address = self.nodes[0].addwitnessaddress(self.ms_address)

NetworkThread().start() # Start up network handling in another thread
network_thread_start()
self.coinbase_blocks = self.nodes[0].generate(2) # Block 2
coinbase_txid = []
for i in self.coinbase_blocks:
Expand Down
4 changes: 2 additions & 2 deletions test/functional/p2p-acceptblock.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def run_test(self):
# min_work_node connects to node1 (whitelisted)
min_work_node = self.nodes[1].add_p2p_connection(P2PInterface())

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

# Test logic begins here
test_node.wait_for_verack()
Expand Down Expand Up @@ -294,7 +294,7 @@ def run_test(self):
self.nodes[0].disconnect_p2ps()
test_node = self.nodes[0].add_p2p_connection(P2PInterface())

NetworkThread().start() # Start up network handling in another thread
network_thread_start()
test_node.wait_for_verack()

# We should have failed reorg and switched back to 290 (but have block 291)
Expand Down
2 changes: 1 addition & 1 deletion test/functional/p2p-compactblocks.py
Original file line number Diff line number Diff line change
Expand Up @@ -792,7 +792,7 @@ def run_test(self):
self.segwit_node = self.nodes[1].add_p2p_connection(TestNode(), services=NODE_NETWORK|NODE_WITNESS)
self.old_node = self.nodes[1].add_p2p_connection(TestNode(), services=NODE_NETWORK)

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

self.test_node.wait_for_verack()

Expand Down
2 changes: 1 addition & 1 deletion test/functional/p2p-feefilter.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def run_test(self):

# Setup the p2p connections and start up the network thread.
self.nodes[0].add_p2p_connection(TestNode())
NetworkThread().start()
network_thread_start()
self.nodes[0].p2p.wait_for_verack()

# Test that invs are received for all txs at feerate of 20 sat/byte
Expand Down
4 changes: 2 additions & 2 deletions test/functional/p2p-fingerprint.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,12 +13,12 @@
from test_framework.blocktools import (create_block, create_coinbase)
from test_framework.mininode import (
CInv,
NetworkThread,
P2PInterface,
msg_headers,
msg_block,
msg_getdata,
msg_getheaders,
network_thread_start,
wait_until,
)
from test_framework.test_framework import BitcoinTestFramework
Expand Down Expand Up @@ -77,7 +77,7 @@ def last_header_equals(self, expected_hash, node):
def run_test(self):
node0 = self.nodes[0].add_p2p_connection(P2PInterface())

NetworkThread().start()
network_thread_start()
node0.wait_for_verack()

# Set node time to 60 days ago
Expand Down
3 changes: 2 additions & 1 deletion test/functional/p2p-fullblocktest.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
import time
from test_framework.key import CECKey
from test_framework.script import *
from test_framework.mininode import network_thread_start
import struct

class PreviousSpendableOutput():
Expand Down Expand Up @@ -68,7 +69,7 @@ def add_options(self, parser):
def run_test(self):
self.test = TestManager(self, self.options.tmpdir)
self.test.add_all_connections(self.nodes)
NetworkThread().start() # Start up network handling in another thread
network_thread_start()
self.test.run()

def add_transactions_to_block(self, block, tx_list):
Expand Down
5 changes: 3 additions & 2 deletions test/functional/p2p-leaktests.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,7 @@ def run_test(self):
unsupported_service_bit5_node = self.nodes[0].add_p2p_connection(CLazyNode(), services=NODE_NETWORK|NODE_UNSUPPORTED_SERVICE_BIT_5)
unsupported_service_bit7_node = self.nodes[0].add_p2p_connection(CLazyNode(), services=NODE_NETWORK|NODE_UNSUPPORTED_SERVICE_BIT_7)

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

wait_until(lambda: no_version_bannode.ever_connected, timeout=10, lock=mininode_lock)
wait_until(lambda: no_version_idlenode.ever_connected, timeout=10, lock=mininode_lock)
Expand Down Expand Up @@ -142,7 +142,8 @@ def run_test(self):
allowed_service_bit5_node = self.nodes[0].add_p2p_connection(P2PInterface(), services=NODE_NETWORK|NODE_UNSUPPORTED_SERVICE_BIT_5)
allowed_service_bit7_node = self.nodes[0].add_p2p_connection(P2PInterface(), services=NODE_NETWORK|NODE_UNSUPPORTED_SERVICE_BIT_7)

NetworkThread().start() # Network thread stopped when all previous P2PInterfaces disconnected. Restart it
# Network thread stopped when all previous P2PInterfaces disconnected. Restart it
network_thread_start()

wait_until(lambda: allowed_service_bit5_node.message_count["verack"], lock=mininode_lock)
wait_until(lambda: allowed_service_bit7_node.message_count["verack"], lock=mininode_lock)
Expand Down
2 changes: 1 addition & 1 deletion test/functional/p2p-mempool.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ def set_test_params(self):
def run_test(self):
# Add a p2p connection
self.nodes[0].add_p2p_connection(P2PInterface())
NetworkThread().start()
network_thread_start()
self.nodes[0].p2p.wait_for_verack()

#request mempool
Expand Down
2 changes: 1 addition & 1 deletion test/functional/p2p-segwit.py
Original file line number Diff line number Diff line change
Expand Up @@ -1882,7 +1882,7 @@ def run_test(self):
# self.std_node is for testing node1 (fRequireStandard=true)
self.std_node = self.nodes[1].add_p2p_connection(TestNode(), services=NODE_NETWORK|NODE_WITNESS)

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

# Keep a place to store utxo's that can be used in later tests
self.utxo = []
Expand Down
2 changes: 1 addition & 1 deletion test/functional/p2p-timeouts.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ def run_test(self):
no_version_node = self.nodes[0].add_p2p_connection(TestNode(), send_version=False)
no_send_node = self.nodes[0].add_p2p_connection(TestNode(), send_version=False)

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

sleep(1)

Expand Down
2 changes: 1 addition & 1 deletion test/functional/p2p-versionbits-warning.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def run_test(self):
# Setup the p2p connection and start up the network thread.
self.nodes[0].add_p2p_connection(TestNode())

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

# Test logic begins here
self.nodes[0].p2p.wait_for_verack()
Expand Down
4 changes: 2 additions & 2 deletions test/functional/sendheaders.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@
CBlockHeader,
CInv,
NODE_WITNESS,
NetworkThread,
network_thread_start,
P2PInterface,
mininode_lock,
msg_block,
Expand Down Expand Up @@ -238,7 +238,7 @@ def run_test(self):
# will occur outside of direct fetching
test_node = self.nodes[0].add_p2p_connection(BaseNode(), services=NODE_WITNESS)

NetworkThread().start() # Start up network handling in another thread
network_thread_start()

# Test logic begins here
inv_node.wait_for_verack()
Expand Down

0 comments on commit 74e64f2

Please sign in to comment.