Skip to content

Commit c3a8e09

Browse files
committed
Merge bitcoin#16991: qa: Fix service flag comparison check in rpc_net test (luke-jr)
9c23ebd qa: Fix service flag comparison check in rpc_net test (Luke Dashjr) Pull request description: Rebase of bitcoin#16936 ACKs for top commit: darosior: ACK 9c23ebd Tree-SHA512: 74f287740403da1040ab1e235ef6eba4e304f3ee5d57a3b25d1e2e1f2f982d256528d398a4d6cb24ba393798e680a8f46cd7dae54ed84ab2c747e96288f1f884
2 parents 2fe6c18 + 9c23ebd commit c3a8e09

File tree

1 file changed

+7
-14
lines changed

1 file changed

+7
-14
lines changed

test/functional/rpc_net.py

+7-14
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,12 @@
2020
wait_until,
2121
)
2222
from test_framework.mininode import P2PInterface
23+
import test_framework.messages
2324
from test_framework.messages import (
2425
CAddress,
2526
msg_addr,
2627
NODE_NETWORK,
2728
NODE_WITNESS,
28-
NODE_GETUTXO,NODE_BLOOM,
29-
NODE_NETWORK_LIMITED,
3029
)
3130

3231
def assert_net_servicesnames(servicesflag, servicenames):
@@ -36,16 +35,10 @@ def assert_net_servicesnames(servicesflag, servicenames):
3635
:param servicesflag: The services as an integer.
3736
:param servicesnames: The list of decoded services names, as strings.
3837
"""
39-
if servicesflag & NODE_NETWORK:
40-
assert "NETWORK" in servicenames
41-
if servicesflag & NODE_GETUTXO:
42-
assert "GETUTXO" in servicenames
43-
if servicesflag & NODE_BLOOM:
44-
assert "BLOOM" in servicenames
45-
if servicesflag & NODE_WITNESS:
46-
assert "WITNESS" in servicenames
47-
if servicesflag & NODE_NETWORK_LIMITED:
48-
assert "NETWORK_LIMITED" in servicenames
38+
servicesflag_generated = 0
39+
for servicename in servicenames:
40+
servicesflag_generated |= getattr(test_framework.messages, 'NODE_' + servicename)
41+
assert servicesflag_generated == servicesflag
4942

5043
class NetTest(BitcoinTestFramework):
5144
def set_test_params(self):
@@ -119,7 +112,7 @@ def _test_getnetworkinfo(self):
119112
# check the `servicesnames` field
120113
network_info = [node.getnetworkinfo() for node in self.nodes]
121114
for info in network_info:
122-
assert_net_servicesnames(int(info["localservices"]), info["localservicesnames"])
115+
assert_net_servicesnames(int(info["localservices"], 0x10), info["localservicesnames"])
123116

124117
def _test_getaddednodeinfo(self):
125118
assert_equal(self.nodes[0].getaddednodeinfo(), [])
@@ -143,7 +136,7 @@ def _test_getpeerinfo(self):
143136
assert_equal(peer_info[1][0]['minfeefilter'], Decimal("0.00001000"))
144137
# check the `servicesnames` field
145138
for info in peer_info:
146-
assert_net_servicesnames(int(info[0]["services"]), info[0]["servicesnames"])
139+
assert_net_servicesnames(int(info[0]["services"], 0x10), info[0]["servicesnames"])
147140

148141
def _test_getnodeaddresses(self):
149142
self.nodes[0].add_p2p_connection(P2PInterface())

0 commit comments

Comments
 (0)