85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
from tests import LimitedTestCase, main
|
|
import time
|
|
from eventlet import api
|
|
from eventlet import hubs
|
|
from eventlet.green import socket
|
|
|
|
DELAY = 0.001
|
|
|
|
class TestScheduleCall(LimitedTestCase):
|
|
def test_local(self):
|
|
lst = [1]
|
|
api.spawn(hubs.get_hub().schedule_call_local, DELAY, lst.pop)
|
|
api.sleep(0)
|
|
api.sleep(DELAY*2)
|
|
assert lst == [1], lst
|
|
|
|
def test_global(self):
|
|
lst = [1]
|
|
api.spawn(hubs.get_hub().schedule_call_global, DELAY, lst.pop)
|
|
api.sleep(0)
|
|
api.sleep(DELAY*2)
|
|
assert lst == [], lst
|
|
|
|
def test_ordering(self):
|
|
lst = []
|
|
hubs.get_hub().schedule_call_global(DELAY*2, lst.append, 3)
|
|
hubs.get_hub().schedule_call_global(DELAY, lst.append, 1)
|
|
hubs.get_hub().schedule_call_global(DELAY, lst.append, 2)
|
|
while len(lst) < 3:
|
|
api.sleep(DELAY)
|
|
self.assertEquals(lst, [1,2,3])
|
|
|
|
|
|
class TestDebug(LimitedTestCase):
|
|
def test_debug_listeners(self):
|
|
hubs.get_hub().set_debug_listeners(True)
|
|
hubs.get_hub().set_debug_listeners(False)
|
|
|
|
def test_timer_exceptions(self):
|
|
hubs.get_hub().set_timer_exceptions(True)
|
|
hubs.get_hub().set_timer_exceptions(False)
|
|
|
|
|
|
class TestExceptionInMainloop(LimitedTestCase):
|
|
def test_sleep(self):
|
|
# even if there was an error in the mainloop, the hub should continue to work
|
|
start = time.time()
|
|
api.sleep(DELAY)
|
|
delay = time.time() - start
|
|
|
|
assert delay >= DELAY*0.9, 'sleep returned after %s seconds (was scheduled for %s)' % (delay, DELAY)
|
|
|
|
def fail():
|
|
1/0
|
|
|
|
hubs.get_hub().schedule_call_global(0, fail)
|
|
|
|
start = time.time()
|
|
api.sleep(DELAY)
|
|
delay = time.time() - start
|
|
|
|
assert delay >= DELAY*0.9, 'sleep returned after %s seconds (was scheduled for %s)' % (delay, DELAY)
|
|
|
|
|
|
class TestHubSelection(LimitedTestCase):
|
|
def test_explicit_hub(self):
|
|
if getattr(hubs.get_hub(), 'uses_twisted_reactor', None):
|
|
# doesn't work with twisted
|
|
return
|
|
oldhub = hubs.get_hub()
|
|
try:
|
|
hubs.use_hub(Foo)
|
|
self.assert_(isinstance(hubs.get_hub(), Foo), hubs.get_hub())
|
|
finally:
|
|
hubs._threadlocal.hub = oldhub
|
|
|
|
|
|
|
|
class Foo(object):
|
|
pass
|
|
|
|
if __name__=='__main__':
|
|
main()
|
|
|