mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-02-08 10:31:50 -05:00
[test] Add P2P_SERVICES to p2p.py
The messages.py module should contain code and helpers for [de]serializing p2p messages. Specific usage of those messages should be in p2p.py. Therefore specify the nServices value in the calling code, not in the messages.py module.
This commit is contained in:
parent
010542614d
commit
9ce4c3c4c1
4 changed files with 8 additions and 2 deletions
|
@ -21,6 +21,7 @@ from test_framework.messages import (
|
|||
)
|
||||
from test_framework.p2p import (
|
||||
P2PInterface,
|
||||
P2P_SERVICES,
|
||||
P2P_SUBVERSION,
|
||||
P2P_VERSION,
|
||||
p2p_lock,
|
||||
|
@ -225,6 +226,7 @@ class FilterTest(BitcoinTestFramework):
|
|||
version_without_fRelay = msg_version()
|
||||
version_without_fRelay.nVersion = P2P_VERSION
|
||||
version_without_fRelay.strSubVer = P2P_SUBVERSION
|
||||
version_without_fRelay.nServices = P2P_SERVICES
|
||||
version_without_fRelay.relay = 0
|
||||
filter_peer_without_nrelay.send_message(version_without_fRelay)
|
||||
filter_peer_without_nrelay.wait_for_verack()
|
||||
|
|
|
@ -20,6 +20,7 @@ from test_framework.messages import (
|
|||
from test_framework.p2p import (
|
||||
P2PInterface,
|
||||
P2P_SUBVERSION,
|
||||
P2P_SERVICES,
|
||||
P2P_VERSION_RELAY,
|
||||
)
|
||||
from test_framework.test_framework import BitcoinTestFramework
|
||||
|
@ -136,6 +137,7 @@ class P2PLeakTest(BitcoinTestFramework):
|
|||
old_version_msg = msg_version()
|
||||
old_version_msg.nVersion = 31799
|
||||
old_version_msg.strSubVer = P2P_SUBVERSION
|
||||
old_version_msg.nServices = P2P_SERVICES
|
||||
old_version_msg.relay = P2P_VERSION_RELAY
|
||||
with self.nodes[0].assert_debug_log(['peer=3 using obsolete version 31799; disconnecting']):
|
||||
p2p_old_peer.send_message(old_version_msg)
|
||||
|
|
|
@ -1022,7 +1022,7 @@ class msg_version:
|
|||
|
||||
def __init__(self):
|
||||
self.nVersion = 0
|
||||
self.nServices = NODE_NETWORK | NODE_WITNESS
|
||||
self.nServices = 0
|
||||
self.nTime = int(time.time())
|
||||
self.addrTo = CAddress()
|
||||
self.addrFrom = CAddress()
|
||||
|
|
|
@ -83,6 +83,8 @@ MIN_P2P_VERSION_SUPPORTED = 60001
|
|||
# The P2P version that this test framework implements and sends in its `version` message
|
||||
# Version 70016 supports wtxid relay
|
||||
P2P_VERSION = 70016
|
||||
# The services that this test framework offers in its `version` message
|
||||
P2P_SERVICES = NODE_NETWORK | NODE_WITNESS
|
||||
# The P2P user agent string that this test framework sends in its `version` message
|
||||
P2P_SUBVERSION = "/python-p2p-tester:0.0.3/"
|
||||
# Value for relay that this test framework sends in its `version` message
|
||||
|
@ -346,7 +348,7 @@ class P2PInterface(P2PConnection):
|
|||
vt.addrFrom.port = 0
|
||||
self.on_connection_send_msg = vt # Will be sent in connection_made callback
|
||||
|
||||
def peer_connect(self, *args, services=NODE_NETWORK | NODE_WITNESS, send_version=True, **kwargs):
|
||||
def peer_connect(self, *args, services=P2P_SERVICES, send_version=True, **kwargs):
|
||||
create_conn = super().peer_connect(*args, **kwargs)
|
||||
|
||||
if send_version:
|
||||
|
|
Loading…
Add table
Reference in a new issue