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)