aboutsummaryrefslogtreecommitdiff
path: root/test/functional/node_network_limited.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/functional/node_network_limited.py')
-rwxr-xr-xtest/functional/node_network_limited.py12
1 files changed, 4 insertions, 8 deletions
diff --git a/test/functional/node_network_limited.py b/test/functional/node_network_limited.py
index 5229dfc99b..1e7c61b44c 100755
--- a/test/functional/node_network_limited.py
+++ b/test/functional/node_network_limited.py
@@ -3,7 +3,7 @@
# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
from test_framework.messages import CInv, msg_getdata, msg_verack
-from test_framework.mininode import NetworkThread, P2PInterface
+from test_framework.mininode import NODE_BLOOM, NODE_NETWORK_LIMITED, NODE_WITNESS, NetworkThread, P2PInterface
from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import assert_equal
@@ -47,11 +47,7 @@ class NodeNetworkLimitedTest(BitcoinTestFramework):
def run_test(self):
# NODE_BLOOM & NODE_WITNESS & NODE_NETWORK_LIMITED must now be signaled
- assert_equal(self.get_signalled_service_flags(), 1036) # 1036 == 0x40C == 0100 0000 1100
-# | ||
-# | |^--- NODE_BLOOM
-# | ^---- NODE_WITNESS
-# ^-- NODE_NETWORK_LIMITED
+ assert_equal(self.get_signalled_service_flags(), NODE_BLOOM | NODE_WITNESS | NODE_NETWORK_LIMITED)
# Now mine some blocks over the NODE_NETWORK_LIMITED + 2(racy buffer ext.) target
firstblock = self.nodes[0].generate(1)[0]
@@ -66,10 +62,10 @@ class NodeNetworkLimitedTest(BitcoinTestFramework):
# NODE_NETWORK_LIMITED must still be signaled after restart
self.restart_node(0)
- assert_equal(self.get_signalled_service_flags(), 1036)
+ assert_equal(self.get_signalled_service_flags(), NODE_BLOOM | NODE_WITNESS | NODE_NETWORK_LIMITED)
# Test the RPC service flags
- assert_equal(self.nodes[0].getnetworkinfo()['localservices'], "000000000000040c")
+ assert_equal(int(self.nodes[0].getnetworkinfo()['localservices'], 16), NODE_BLOOM | NODE_WITNESS | NODE_NETWORK_LIMITED)
# getdata a block above the NODE_NETWORK_LIMITED threshold must be possible
self.try_get_block_via_getdata(block_within_limited_range, False)