# Copyright (C) PyZMQ Developers # Distributed under the terms of the Modified BSD License. import time import os import threading import pytest import zmq from zmq.tests import BaseZMQTestCase, have_gevent try: from tornado.ioloop import IOLoop as BaseIOLoop from zmq.eventloop import ioloop _tornado = True except ImportError: _tornado = False def printer(): os.system("say hello") raise Exception print (time.time()) class Delay(threading.Thread): def __init__(self, f, delay=1): self.f=f self.delay=delay self.aborted=False self.cond=threading.Condition() super(Delay, self).__init__() def run(self): self.cond.acquire() self.cond.wait(self.delay) self.cond.release() if not self.aborted: self.f() def abort(self): self.aborted=True self.cond.acquire() self.cond.notify() self.cond.release() class TestIOLoop(BaseZMQTestCase): def setUp(self): if not _tornado: pytest.skip("tornado required") super(TestIOLoop, self).setUp() def tearDown(self): super(TestIOLoop, self).tearDown() BaseIOLoop.clear_current() BaseIOLoop.clear_instance() def test_simple(self): """simple IOLoop creation test""" loop = ioloop.IOLoop() loop.make_current() dc = ioloop.PeriodicCallback(loop.stop, 200) pc = ioloop.PeriodicCallback(lambda : None, 10) pc.start() dc.start() t = Delay(loop.stop,1) t.start() loop.start() if t.isAlive(): t.abort() else: self.fail("IOLoop failed to exit") def test_instance(self): """Green IOLoop.instance returns the right object""" loop = ioloop.IOLoop.instance() assert isinstance(loop, ioloop.IOLoop) base_loop = BaseIOLoop.instance() assert base_loop is loop def test_current(self): """Green IOLoop.current returns the right object""" loop = ioloop.IOLoop.current() assert isinstance(loop, ioloop.IOLoop) base_loop = BaseIOLoop.current() assert base_loop is loop def test_close_all(self): """Test close(all_fds=True)""" loop = ioloop.IOLoop.instance() req,rep = self.create_bound_pair(zmq.REQ, zmq.REP) loop.add_handler(req, lambda msg: msg, ioloop.IOLoop.READ) loop.add_handler(rep, lambda msg: msg, ioloop.IOLoop.READ) self.assertEqual(req.closed, False) self.assertEqual(rep.closed, False) loop.close(all_fds=True) self.assertEqual(req.closed, True) self.assertEqual(rep.closed, True) if have_gevent and _tornado: import zmq.green.eventloop.ioloop as green_ioloop class TestIOLoopGreen(BaseZMQTestCase): def test_instance(self): """Green IOLoop.instance returns the right object""" loop = green_ioloop.IOLoop.instance() assert isinstance(loop, green_ioloop.IOLoop) base_loop = BaseIOLoop.instance() assert base_loop is loop def test_current(self): """Green IOLoop.current returns the right object""" loop = green_ioloop.IOLoop.current() assert isinstance(loop, green_ioloop.IOLoop) base_loop = BaseIOLoop.current() assert base_loop is loop