mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-28 20:22:12 +08:00
Stephen Diehl
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
import gevent
|
|
from logbook.compat import LoggingHandler
|
|
from unittest2 import TestCase, skip
|
|
|
|
from zipline.core.monitor import Controller
|
|
|
|
|
|
class TestMonitor(TestCase):
|
|
def setUp(self):
|
|
self.log_handler = LoggingHandler()
|
|
self.log_handler.push_application()
|
|
|
|
def tearDown(self):
|
|
self.log_handler.pop_application()
|
|
|
|
def test_init(self):
|
|
pub_socket = 'tcp://127.0.0.1:5000'
|
|
route_socket = 'tcp://127.0.0.1:5001'
|
|
|
|
con = Controller(pub_socket, route_socket)
|
|
con.manage([])
|
|
|
|
def test_init_topology(self):
|
|
pub_socket = 'tcp://127.0.0.1:5000'
|
|
route_socket = 'tcp://127.0.0.1:5001'
|
|
|
|
con = Controller(pub_socket, route_socket, )
|
|
con.manage([ 'a', 'b', 'c', 'd' ])
|
|
|
|
@skip
|
|
def test_poll(self):
|
|
from mock_zmq import zmq_synthetic
|
|
pub_socket = 'tcp://127.0.0.1:5000'
|
|
route_socket = 'tcp://127.0.0.1:5001'
|
|
cancel_socket = 'tcp://127.0.0.1:5002'
|
|
|
|
con = Controller(pub_socket, route_socket, cancel_socket)
|
|
con.manage([ 'a', 'b', 'c', 'd' ])
|
|
con.zmq = zmq_synthetic
|
|
con.zmq_flavor = 'green'
|
|
|
|
con.period = 0.00001
|
|
gevent.spawn(con.run).join(timeout=con.period)
|