39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
|
import time
|
||
|
|
|
||
|
|
import zmq
|
||
|
|
|
||
|
|
from .test_process import TestProcessProto
|
||
|
|
|
||
|
|
|
||
|
|
class TestProcessQueue(TestProcessProto):
|
||
|
|
_process_cmd = ['minode', '--msg-queue']
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
super().setUpClass()
|
||
|
|
context = zmq.Context()
|
||
|
|
cls.socket = context.socket(zmq.SUB)
|
||
|
|
cls.socket.connect('tcp://localhost:5556')
|
||
|
|
cls.socket.setsockopt(zmq.RCVTIMEO, 120000)
|
||
|
|
cls.socket.setsockopt(zmq.SUBSCRIBE, b'obj')
|
||
|
|
|
||
|
|
def test_receive_msg(self):
|
||
|
|
"""wait a couple of messages"""
|
||
|
|
timeout = 240
|
||
|
|
start = time.time()
|
||
|
|
c = 0
|
||
|
|
while time.time() - start < timeout:
|
||
|
|
if c > 1:
|
||
|
|
return
|
||
|
|
try:
|
||
|
|
tag, data = self.socket.recv().split(b'\x00', 1)
|
||
|
|
except zmq.error.Again:
|
||
|
|
time.sleep(2)
|
||
|
|
continue
|
||
|
|
else:
|
||
|
|
c += 1
|
||
|
|
self.assertEqual(tag, b'obj')
|
||
|
|
|
||
|
|
if c == 0:
|
||
|
|
self.fail('No messages received in %ss' % timeout)
|