mirror of https://github.com/scrapy/scrapy.git
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import asyncio
|
|
import warnings
|
|
from unittest import TestCase
|
|
|
|
from pytest import mark
|
|
|
|
from scrapy.utils.reactor import (
|
|
install_reactor,
|
|
is_asyncio_reactor_installed,
|
|
set_asyncio_event_loop,
|
|
)
|
|
|
|
|
|
@mark.usefixtures("reactor_pytest")
|
|
class AsyncioTest(TestCase):
|
|
def test_is_asyncio_reactor_installed(self):
|
|
# the result should depend only on the pytest --reactor argument
|
|
self.assertEqual(
|
|
is_asyncio_reactor_installed(), self.reactor_pytest == "asyncio"
|
|
)
|
|
|
|
def test_install_asyncio_reactor(self):
|
|
from twisted.internet import reactor as original_reactor
|
|
|
|
with warnings.catch_warnings(record=True) as w:
|
|
install_reactor("twisted.internet.asyncioreactor.AsyncioSelectorReactor")
|
|
self.assertEqual(len(w), 0)
|
|
from twisted.internet import reactor
|
|
|
|
assert original_reactor == reactor
|
|
|
|
async def test_set_asyncio_event_loop(self):
|
|
install_reactor("twisted.internet.asyncioreactor.AsyncioSelectorReactor")
|
|
assert set_asyncio_event_loop() is asyncio.get_running_loop()
|