mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-06 05:14:38 +08:00
fawce
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
"""
|
|
Dummy simulator backported from Qexec for development on Zipline.
|
|
"""
|
|
|
|
import logging
|
|
from multiprocessing import Process
|
|
from unittest2 import TestCase
|
|
|
|
from zipline.monitor import Controller
|
|
|
|
import gevent
|
|
from gevent_zeromq import zmq
|
|
|
|
ctx = zmq.Context()
|
|
|
|
#TODO: disabled by prefixing the test methods with a d
|
|
class TestControlProtocol(TestCase):
|
|
|
|
def setUpController(self):
|
|
self.controller.run()
|
|
|
|
def setUp(self):
|
|
self.controller = Controller(
|
|
'tcp://127.0.0.1:5000',
|
|
'tcp://127.0.0.1:5001',
|
|
)
|
|
self.control_proc = Process(target=self.setUpController)
|
|
self.control_proc.start()
|
|
|
|
def tearDown(self):
|
|
self.control_proc.terminate()
|
|
ctx.destroy()
|
|
|
|
def asyncMessage(self, socket):
|
|
return socket.recv()
|
|
|
|
def send_and_receive(self, push, sub, message='\x01'):
|
|
msg = gevent.spawn(sub.recv)
|
|
push.send(message)
|
|
gevent.sleep(0) # explicit gevent yield
|
|
msg.join()
|
|
self.assertEqual(msg.value, message)
|
|
|
|
def dtest_control_message(self):
|
|
|
|
sub = self.controller.message_listener(context=ctx)
|
|
message = gevent.spawn(self.asyncMessage, sub)
|
|
|
|
push = self.controller.message_sender(context=ctx)
|
|
|
|
# Don't like introducing time constants but because of
|
|
# the way gevent scheduler works zmq will often send all
|
|
# the messages off before the other thread even gets to
|
|
# listening.
|
|
self.send_and_receive(push, sub)
|
|
sub.close()
|
|
push.close()
|
|
|
|
def dtest_control_delivery(self):
|
|
# Assert that the number of messages sent on the wire is
|
|
# the number of messages received, ie we don't drop any.
|
|
# This is of course depenendent on the topology of the
|
|
# listeners being fixed. Which normally it isn't.
|
|
|
|
sub = self.controller.message_listener(context=ctx)
|
|
message = gevent.spawn(self.asyncMessage, sub)
|
|
|
|
push = self.controller.message_sender(context=ctx)
|
|
|
|
# Don't like introducing time constants but because of
|
|
# the way gevent scheduler works zmq will often send all
|
|
# the messages off before the other thread even gets to
|
|
# listening.
|
|
for i in xrange(25):
|
|
self.send_and_receive(push, sub)
|
|
|
|
sub.close()
|
|
push.close()
|