Browse Source

[tests] in zmq test, timeout if message not received

0.15
John Newbery 8 years ago
parent
commit
b1bac1cb7e
  1. 4
      test/functional/zmq_test.py

4
test/functional/zmq_test.py

@ -36,6 +36,7 @@ class ZMQTest (BitcoinTestFramework):
self.zmqContext = zmq.Context() self.zmqContext = zmq.Context()
self.zmqSubSocket = self.zmqContext.socket(zmq.SUB) self.zmqSubSocket = self.zmqContext.socket(zmq.SUB)
self.zmqSubSocket.set(zmq.RCVTIMEO, 60000)
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashblock") self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashblock")
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashtx") self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashtx")
ip_address = "tcp://127.0.0.1:28332" ip_address = "tcp://127.0.0.1:28332"
@ -94,8 +95,7 @@ class ZMQTest (BitcoinTestFramework):
msg = self.zmqSubSocket.recv_multipart() msg = self.zmqSubSocket.recv_multipart()
topic = msg[0] topic = msg[0]
body = msg[1] body = msg[1]
hashZMQ = "" assert_equal(topic, b"hashtx")
if topic == b"hashtx":
hashZMQ = bytes_to_hex_str(body) hashZMQ = bytes_to_hex_str(body)
msgSequence = struct.unpack('<I', msg[-1])[-1] msgSequence = struct.unpack('<I', msg[-1])[-1]
assert_equal(msgSequence, blockcount + 1) assert_equal(msgSequence, blockcount + 1)

Loading…
Cancel
Save