highperfocused/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-06-26 17:02:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

test: Add IPv6 test to zmq

Browse Source
This commit is contained in:
nthumann 2021-05-26 19:38:38 +02:00
parent ded449b726
commit 8abe5703a9
No known key found for this signature in database
GPG Key ID: 91AE26F52DE54B76
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
test/functional/interface_zmq.py
View File

@ -13,6 +13,7 @@ from test_framework.util import (
assert_equal, assert_equal,
assert_raises_rpc_error, assert_raises_rpc_error,
) )
from test_framework.netutil import test_ipv6_local
from io import BytesIO from io import BytesIO
from time import sleep from time import sleep
@ -108,6 +109,7 @@ class ZMQTest (BitcoinTestFramework):
self.test_mempool_sync() self.test_mempool_sync()
self.test_reorg() self.test_reorg()
self.test_multiple_interfaces() self.test_multiple_interfaces()
self.test_ipv6()
finally: finally:
# Destroy the ZMQ context. # Destroy the ZMQ context.
self.log.debug("Destroying ZMQ context") self.log.debug("Destroying ZMQ context")
@ -115,10 +117,12 @@ class ZMQTest (BitcoinTestFramework):
# Restart node with the specified zmq notifications enabled, subscribe to # Restart node with the specified zmq notifications enabled, subscribe to
# all of them and return the corresponding ZMQSubscriber objects. # all of them and return the corresponding ZMQSubscriber objects.
def setup_zmq_test(self, services, *, recv_timeout=60, sync_blocks=True): def setup_zmq_test(self, services, *, recv_timeout=60, sync_blocks=True, ipv6=False):
subscribers = [] subscribers = []
for topic, address in services: for topic, address in services:
socket = self.ctx.socket(zmq.SUB) socket = self.ctx.socket(zmq.SUB)
if ipv6:
socket.setsockopt(zmq.IPV6, 1)
subscribers.append(ZMQSubscriber(socket, topic.encode())) subscribers.append(ZMQSubscriber(socket, topic.encode()))
self.restart_node(0, ["-zmqpub%s=%s" % (topic, address) for topic, address in services] + self.restart_node(0, ["-zmqpub%s=%s" % (topic, address) for topic, address in services] +
@ -557,5 +561,22 @@ class ZMQTest (BitcoinTestFramework):
assert_equal(self.nodes[0].getbestblockhash(), subscribers[0].receive().hex()) assert_equal(self.nodes[0].getbestblockhash(), subscribers[0].receive().hex())
assert_equal(self.nodes[0].getbestblockhash(), subscribers[1].receive().hex()) assert_equal(self.nodes[0].getbestblockhash(), subscribers[1].receive().hex())
def test_ipv6(self):
if not test_ipv6_local():
self.log.info("Skipping IPv6 test, because IPv6 is not supported.")
return
self.log.info("Testing IPv6")
# Set up subscriber using IPv6 loopback address
subscribers = self.setup_zmq_test([
("hashblock", "tcp://[::1]:28332")
], ipv6=True)
# Generate 1 block in nodes[0]
self.nodes[0].generatetoaddress(1, ADDRESS_BCRT1_UNSPENDABLE)
# Should receive the same block hash
assert_equal(self.nodes[0].getbestblockhash(), subscribers[0].receive().hex())
if __name__ == '__main__': if __name__ == '__main__':
ZMQTest().main() ZMQTest().main()