mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-12 14:13:38 +00:00
70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||
|
|
# not use this file except in compliance with the License. You may obtain
|
||
|
|
# a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
|
# License for the specific language governing permissions and limitations
|
||
|
|
# under the License.
|
||
|
|
|
||
|
|
from __future__ import absolute_import, division, print_function, with_statement
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import textwrap
|
||
|
|
|
||
|
|
from tornado import gen
|
||
|
|
from tornado.testing import AsyncTestCase, gen_test
|
||
|
|
from tornado.test.util import unittest
|
||
|
|
|
||
|
|
try:
|
||
|
|
from tornado.platform.asyncio import asyncio, AsyncIOLoop
|
||
|
|
except ImportError:
|
||
|
|
asyncio = None
|
||
|
|
|
||
|
|
skipIfNoSingleDispatch = unittest.skipIf(
|
||
|
|
gen.singledispatch is None, "singledispatch module not present")
|
||
|
|
|
||
|
|
|
||
|
|
@unittest.skipIf(asyncio is None, "asyncio module not present")
|
||
|
|
class AsyncIOLoopTest(AsyncTestCase):
|
||
|
|
def get_new_ioloop(self):
|
||
|
|
io_loop = AsyncIOLoop()
|
||
|
|
asyncio.set_event_loop(io_loop.asyncio_loop)
|
||
|
|
return io_loop
|
||
|
|
|
||
|
|
def test_asyncio_callback(self):
|
||
|
|
# Basic test that the asyncio loop is set up correctly.
|
||
|
|
asyncio.get_event_loop().call_soon(self.stop)
|
||
|
|
self.wait()
|
||
|
|
|
||
|
|
@skipIfNoSingleDispatch
|
||
|
|
@gen_test
|
||
|
|
def test_asyncio_future(self):
|
||
|
|
# Test that we can yield an asyncio future from a tornado coroutine.
|
||
|
|
# Without 'yield from', we must wrap coroutines in asyncio.async.
|
||
|
|
x = yield asyncio.async(
|
||
|
|
asyncio.get_event_loop().run_in_executor(None, lambda: 42))
|
||
|
|
self.assertEqual(x, 42)
|
||
|
|
|
||
|
|
@unittest.skipIf(sys.version_info < (3, 3),
|
||
|
|
'PEP 380 not available')
|
||
|
|
@skipIfNoSingleDispatch
|
||
|
|
@gen_test
|
||
|
|
def test_asyncio_yield_from(self):
|
||
|
|
# Test that we can use asyncio coroutines with 'yield from'
|
||
|
|
# instead of asyncio.async(). This requires python 3.3 syntax.
|
||
|
|
global_namespace = dict(globals(), **locals())
|
||
|
|
local_namespace = {}
|
||
|
|
exec(textwrap.dedent("""
|
||
|
|
@gen.coroutine
|
||
|
|
def f():
|
||
|
|
event_loop = asyncio.get_event_loop()
|
||
|
|
x = yield from event_loop.run_in_executor(None, lambda: 42)
|
||
|
|
return x
|
||
|
|
"""), global_namespace, local_namespace)
|
||
|
|
result = yield local_namespace['f']()
|
||
|
|
self.assertEqual(result, 42)
|