mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
79 lines
2.0 KiB
Python
79 lines
2.0 KiB
Python
# coding: utf-8
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
# Import 3rd-party libs
|
|
import tornado.testing
|
|
import tornado.gen
|
|
from tornado.testing import AsyncTestCase
|
|
|
|
import salt.utils.async as async
|
|
|
|
|
|
class HelperA(object):
|
|
def __init__(self, io_loop=None):
|
|
pass
|
|
|
|
@tornado.gen.coroutine
|
|
def sleep(self):
|
|
yield tornado.gen.sleep(0.5)
|
|
raise tornado.gen.Return(True)
|
|
|
|
|
|
class HelperB(object):
|
|
def __init__(self, a=None, io_loop=None):
|
|
if a is None:
|
|
a = async.SyncWrapper(HelperA)
|
|
self.a = a
|
|
|
|
@tornado.gen.coroutine
|
|
def sleep(self):
|
|
yield tornado.gen.sleep(0.5)
|
|
self.a.sleep()
|
|
raise tornado.gen.Return(False)
|
|
|
|
|
|
class TestSyncWrapper(AsyncTestCase):
|
|
@tornado.testing.gen_test
|
|
def test_helpers(self):
|
|
'''
|
|
Test that the helper classes do what we expect within a regular async env
|
|
'''
|
|
ha = HelperA()
|
|
ret = yield ha.sleep()
|
|
self.assertTrue(ret)
|
|
|
|
hb = HelperB()
|
|
ret = yield hb.sleep()
|
|
self.assertFalse(ret)
|
|
|
|
def test_basic_wrap(self):
|
|
'''
|
|
Test that we can wrap an async caller.
|
|
'''
|
|
sync = async.SyncWrapper(HelperA)
|
|
ret = sync.sleep()
|
|
self.assertTrue(ret)
|
|
|
|
def test_double(self):
|
|
'''
|
|
Test when the async wrapper object itself creates a wrap of another thing
|
|
|
|
This works fine since the second wrap is based on the first's IOLoop so we
|
|
don't have to worry about complex start/stop mechanics
|
|
'''
|
|
sync = async.SyncWrapper(HelperB)
|
|
ret = sync.sleep()
|
|
self.assertFalse(ret)
|
|
|
|
def test_double_sameloop(self):
|
|
'''
|
|
Test async wrappers initiated from the same IOLoop, to ensure that
|
|
we don't wire up both to the same IOLoop (since it causes MANY problems).
|
|
'''
|
|
a = async.SyncWrapper(HelperA)
|
|
sync = async.SyncWrapper(HelperB, (a,))
|
|
ret = sync.sleep()
|
|
self.assertFalse(ret)
|