Merge #10552: [Test] Tests for zmqpubrawtx and zmqpubrawblock
d3677ab75
Tests for zmqpubrawtx and zmqpubrawblock (Andrew Chow)
Pull request description:
Tree-SHA512: 9e367fd8936514bfb567ef3f3d83770d374287354b59c9187e844056dd086e8aa2de32ce55d35486cecd706e7c93cd1c1e2709ee82d3dddb805827be8d2bcb14
This commit is contained in:
commit
9c3c9cdae3
2 changed files with 54 additions and 6 deletions
|
@ -7,6 +7,7 @@
|
|||
from base64 import b64encode
|
||||
from binascii import hexlify, unhexlify
|
||||
from decimal import Decimal, ROUND_DOWN
|
||||
import hashlib
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
|
@ -173,6 +174,13 @@ def count_bytes(hex_string):
|
|||
def bytes_to_hex_str(byte_str):
|
||||
return hexlify(byte_str).decode('ascii')
|
||||
|
||||
def hash256(byte_str):
|
||||
sha256 = hashlib.sha256()
|
||||
sha256.update(byte_str)
|
||||
sha256d = hashlib.sha256()
|
||||
sha256d.update(sha256.digest())
|
||||
return sha256d.digest()[::-1]
|
||||
|
||||
def hex_str_to_bytes(hex_str):
|
||||
return unhexlify(hex_str.encode('ascii'))
|
||||
|
||||
|
|
|
@ -10,6 +10,7 @@ import struct
|
|||
from test_framework.test_framework import BitcoinTestFramework, SkipTest
|
||||
from test_framework.util import (assert_equal,
|
||||
bytes_to_hex_str,
|
||||
hash256,
|
||||
)
|
||||
|
||||
class ZMQTest (BitcoinTestFramework):
|
||||
|
@ -37,9 +38,12 @@ class ZMQTest (BitcoinTestFramework):
|
|||
self.zmqSubSocket.set(zmq.RCVTIMEO, 60000)
|
||||
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashblock")
|
||||
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashtx")
|
||||
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"rawblock")
|
||||
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"rawtx")
|
||||
ip_address = "tcp://127.0.0.1:28332"
|
||||
self.zmqSubSocket.connect(ip_address)
|
||||
self.extra_args = [['-zmqpubhashtx=%s' % ip_address, '-zmqpubhashblock=%s' % ip_address], []]
|
||||
self.extra_args = [['-zmqpubhashblock=%s' % ip_address, '-zmqpubhashtx=%s' % ip_address,
|
||||
'-zmqpubrawblock=%s' % ip_address, '-zmqpubrawtx=%s' % ip_address], []]
|
||||
self.add_nodes(self.num_nodes, self.extra_args)
|
||||
self.start_nodes()
|
||||
|
||||
|
@ -59,28 +63,51 @@ class ZMQTest (BitcoinTestFramework):
|
|||
msg = self.zmqSubSocket.recv_multipart()
|
||||
topic = msg[0]
|
||||
assert_equal(topic, b"hashtx")
|
||||
body = msg[1]
|
||||
txhash = msg[1]
|
||||
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
||||
assert_equal(msgSequence, 0) # must be sequence 0 on hashtx
|
||||
|
||||
# rawtx
|
||||
msg = self.zmqSubSocket.recv_multipart()
|
||||
topic = msg[0]
|
||||
assert_equal(topic, b"rawtx")
|
||||
body = msg[1]
|
||||
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
||||
assert_equal(msgSequence, 0) # must be sequence 0 on rawtx
|
||||
|
||||
# Check that the rawtx hashes to the hashtx
|
||||
assert_equal(hash256(body), txhash)
|
||||
|
||||
self.log.info("Wait for block")
|
||||
msg = self.zmqSubSocket.recv_multipart()
|
||||
topic = msg[0]
|
||||
assert_equal(topic, b"hashblock")
|
||||
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
|
||||
|
||||
# rawblock
|
||||
msg = self.zmqSubSocket.recv_multipart()
|
||||
topic = msg[0]
|
||||
assert_equal(topic, b"rawblock")
|
||||
body = msg[1]
|
||||
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
||||
assert_equal(msgSequence, 0) #must be sequence 0 on rawblock
|
||||
|
||||
# Check the hash of the rawblock's header matches generate
|
||||
assert_equal(genhashes[0], bytes_to_hex_str(hash256(body[:80])))
|
||||
|
||||
self.log.info("Generate 10 blocks (and 10 coinbase txes)")
|
||||
n = 10
|
||||
genhashes = self.nodes[1].generate(n)
|
||||
self.sync_all()
|
||||
|
||||
zmqHashes = []
|
||||
zmqRawHashed = []
|
||||
blockcount = 0
|
||||
for x in range(n * 2):
|
||||
for x in range(n * 4):
|
||||
msg = self.zmqSubSocket.recv_multipart()
|
||||
topic = msg[0]
|
||||
body = msg[1]
|
||||
|
@ -89,9 +116,14 @@ class ZMQTest (BitcoinTestFramework):
|
|||
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
||||
assert_equal(msgSequence, blockcount + 1)
|
||||
blockcount += 1
|
||||
if topic == b"rawblock":
|
||||
zmqRawHashed.append(bytes_to_hex_str(hash256(body[:80])))
|
||||
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
||||
assert_equal(msgSequence, blockcount)
|
||||
|
||||
for x in range(n):
|
||||
assert_equal(genhashes[x], zmqHashes[x]) # blockhash from generate must be equal to the hash received over zmq
|
||||
assert_equal(genhashes[x], zmqRawHashed[x])
|
||||
|
||||
self.log.info("Wait for tx from second node")
|
||||
# test tx from a second node
|
||||
|
@ -101,13 +133,21 @@ class ZMQTest (BitcoinTestFramework):
|
|||
# now we should receive a zmq msg because the tx was broadcast
|
||||
msg = self.zmqSubSocket.recv_multipart()
|
||||
topic = msg[0]
|
||||
body = msg[1]
|
||||
assert_equal(topic, b"hashtx")
|
||||
body = msg[1]
|
||||
hashZMQ = bytes_to_hex_str(body)
|
||||
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
||||
assert_equal(msgSequence, blockcount + 1)
|
||||
|
||||
msg = self.zmqSubSocket.recv_multipart()
|
||||
topic = msg[0]
|
||||
assert_equal(topic, b"rawtx")
|
||||
body = msg[1]
|
||||
hashedZMQ = bytes_to_hex_str(hash256(body))
|
||||
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
||||
assert_equal(msgSequence, blockcount+1)
|
||||
assert_equal(hashRPC, hashZMQ) # txid from sendtoaddress must be equal to the hash received over zmq
|
||||
assert_equal(hashRPC, hashedZMQ)
|
||||
|
||||
if __name__ == '__main__':
|
||||
ZMQTest().main()
|
||||
|
|
Loading…
Reference in a new issue