dad8c67d38
64c0800 Use logging in individual tests (John Newbery) 38ad281 Use logging in test_framework/comptool.py (John Newbery) ff19073 Use logging in test_framework/blockstore.py (John Newbery) 2a9c7c7 Use logging in test_framework/util.py (John Newbery) b0dec4a Remove manual debug settings in qa tests. (John Newbery) af1363c Always enable debug log and microsecond logging for test nodes. (John Newbery) 6d0e325 Use logging in mininode.py (John Newbery) 553a976 Add logging to p2p-segwit.py (John Newbery) 0e6d23d Add logging to test_framework.py (John Newbery) Tree-SHA512: 42ee2acbf444ec32d796f930f9f6e272da03c75e93d974a126d4ea9b2dbaa77cc57ab5e63ce3fd33d609049d884eb8d9f65272c08922d10f8db69d4a60ad05a3
95 lines
3.2 KiB
Python
Executable File
95 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2015-2016 The Bitcoin Core developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
"""Test the ZMQ API."""
|
|
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import *
|
|
import zmq
|
|
import struct
|
|
|
|
class ZMQTest (BitcoinTestFramework):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.num_nodes = 4
|
|
|
|
port = 28332
|
|
|
|
def setup_nodes(self):
|
|
self.zmqContext = zmq.Context()
|
|
self.zmqSubSocket = self.zmqContext.socket(zmq.SUB)
|
|
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashblock")
|
|
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashtx")
|
|
self.zmqSubSocket.connect("tcp://127.0.0.1:%i" % self.port)
|
|
return start_nodes(self.num_nodes, self.options.tmpdir, extra_args=[
|
|
['-zmqpubhashtx=tcp://127.0.0.1:'+str(self.port), '-zmqpubhashblock=tcp://127.0.0.1:'+str(self.port)],
|
|
[],
|
|
[],
|
|
[]
|
|
])
|
|
|
|
def run_test(self):
|
|
self.sync_all()
|
|
|
|
genhashes = self.nodes[0].generate(1)
|
|
self.sync_all()
|
|
|
|
self.log.info("listen...")
|
|
msg = self.zmqSubSocket.recv_multipart()
|
|
topic = msg[0]
|
|
assert_equal(topic, b"hashtx")
|
|
body = msg[1]
|
|
nseq = msg[2]
|
|
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
|
assert_equal(msgSequence, 0) #must be sequence 0 on hashtx
|
|
|
|
msg = self.zmqSubSocket.recv_multipart()
|
|
topic = msg[0]
|
|
body = msg[1]
|
|
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
|
assert_equal(msgSequence, 0) #must be sequence 0 on hashblock
|
|
blkhash = bytes_to_hex_str(body)
|
|
|
|
assert_equal(genhashes[0], blkhash) #blockhash from generate must be equal to the hash received over zmq
|
|
|
|
n = 10
|
|
genhashes = self.nodes[1].generate(n)
|
|
self.sync_all()
|
|
|
|
zmqHashes = []
|
|
blockcount = 0
|
|
for x in range(0,n*2):
|
|
msg = self.zmqSubSocket.recv_multipart()
|
|
topic = msg[0]
|
|
body = msg[1]
|
|
if topic == b"hashblock":
|
|
zmqHashes.append(bytes_to_hex_str(body))
|
|
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
|
assert_equal(msgSequence, blockcount+1)
|
|
blockcount += 1
|
|
|
|
for x in range(0,n):
|
|
assert_equal(genhashes[x], zmqHashes[x]) #blockhash from generate must be equal to the hash received over zmq
|
|
|
|
#test tx from a second node
|
|
hashRPC = self.nodes[1].sendtoaddress(self.nodes[0].getnewaddress(), 1.0)
|
|
self.sync_all()
|
|
|
|
# now we should receive a zmq msg because the tx was broadcast
|
|
msg = self.zmqSubSocket.recv_multipart()
|
|
topic = msg[0]
|
|
body = msg[1]
|
|
hashZMQ = ""
|
|
if topic == b"hashtx":
|
|
hashZMQ = bytes_to_hex_str(body)
|
|
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
|
assert_equal(msgSequence, blockcount+1)
|
|
|
|
assert_equal(hashRPC, hashZMQ) #blockhash from generate must be equal to the hash received over zmq
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ZMQTest ().main ()
|