mirror of
https://github.com/django/daphne.git
synced 2024-11-13 03:26:33 +03:00
Resolve asyncio + multiprocessing problem when testing. (#247)
This commit is contained in:
parent
d3630e0925
commit
7032f8e0f8
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1,3 +1,4 @@
|
||||||
|
.idea/
|
||||||
*.egg-info
|
*.egg-info
|
||||||
*.pyc
|
*.pyc
|
||||||
__pycache__
|
__pycache__
|
||||||
|
|
37
.travis.yml
37
.travis.yml
|
@ -34,6 +34,43 @@ jobs:
|
||||||
dist: xenial
|
dist: xenial
|
||||||
sudo: required
|
sudo: required
|
||||||
|
|
||||||
|
- os: osx
|
||||||
|
language: generic
|
||||||
|
python: '3.5'
|
||||||
|
env: TWISTED="twisted==18.7.0"
|
||||||
|
before_install:
|
||||||
|
- eval "$(pyenv init -)"
|
||||||
|
- pyenv install 3.5.5
|
||||||
|
- pyenv global 3.5.5
|
||||||
|
sudo: required
|
||||||
|
- os: osx
|
||||||
|
language: generic
|
||||||
|
python: '3.5'
|
||||||
|
env: TWISTED="twisted"
|
||||||
|
before_install:
|
||||||
|
- eval "$(pyenv init -)"
|
||||||
|
- pyenv install 3.5.5
|
||||||
|
- pyenv global 3.5.5
|
||||||
|
sudo: required
|
||||||
|
- os: osx
|
||||||
|
language: generic
|
||||||
|
python: '3.6'
|
||||||
|
env: TWISTED="twisted==18.7.0"
|
||||||
|
before_install:
|
||||||
|
- eval "$(pyenv init -)"
|
||||||
|
- pyenv install 3.6.5
|
||||||
|
- pyenv global 3.6.5
|
||||||
|
sudo: required
|
||||||
|
- os: osx
|
||||||
|
language: generic
|
||||||
|
python: '3.6'
|
||||||
|
env: TWISTED="twisted"
|
||||||
|
before_install:
|
||||||
|
- eval "$(pyenv init -)"
|
||||||
|
- pyenv install 3.6.5
|
||||||
|
- pyenv global 3.6.5
|
||||||
|
sudo: required
|
||||||
|
|
||||||
- stage: lint
|
- stage: lint
|
||||||
install: pip install -U -e .[tests] black pyflakes isort
|
install: pip install -U -e .[tests] black pyflakes isort
|
||||||
script:
|
script:
|
||||||
|
|
|
@ -6,11 +6,6 @@ import tempfile
|
||||||
import traceback
|
import traceback
|
||||||
from concurrent.futures import CancelledError
|
from concurrent.futures import CancelledError
|
||||||
|
|
||||||
from twisted.internet import reactor
|
|
||||||
|
|
||||||
from .endpoints import build_endpoint_description_strings
|
|
||||||
from .server import Server
|
|
||||||
|
|
||||||
|
|
||||||
class DaphneTestingInstance:
|
class DaphneTestingInstance:
|
||||||
"""
|
"""
|
||||||
|
@ -121,6 +116,17 @@ class DaphneProcess(multiprocessing.Process):
|
||||||
self.errors = multiprocessing.Queue()
|
self.errors = multiprocessing.Queue()
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
|
# OK, now we are in a forked child process, and want to use the reactor.
|
||||||
|
# However, FreeBSD systems like MacOS do not fork the underlying Kqueue,
|
||||||
|
# which asyncio (hence asyncioreactor) is built on.
|
||||||
|
# Therefore, we should uninstall the broken reactor and install a new one.
|
||||||
|
_reinstall_reactor()
|
||||||
|
|
||||||
|
from twisted.internet import reactor
|
||||||
|
|
||||||
|
from .server import Server
|
||||||
|
from .endpoints import build_endpoint_description_strings
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# Create the server class
|
# Create the server class
|
||||||
endpoints = build_endpoint_description_strings(host=self.host, port=0)
|
endpoints = build_endpoint_description_strings(host=self.host, port=0)
|
||||||
|
@ -143,6 +149,8 @@ class DaphneProcess(multiprocessing.Process):
|
||||||
self.errors.put((e, traceback.format_exc()))
|
self.errors.put((e, traceback.format_exc()))
|
||||||
|
|
||||||
def resolve_port(self):
|
def resolve_port(self):
|
||||||
|
from twisted.internet import reactor
|
||||||
|
|
||||||
if self.server.listening_addresses:
|
if self.server.listening_addresses:
|
||||||
self.port.value = self.server.listening_addresses[0][1]
|
self.port.value = self.server.listening_addresses[0][1]
|
||||||
self.ready.set()
|
self.ready.set()
|
||||||
|
@ -249,3 +257,24 @@ class TestApplication:
|
||||||
os.unlink(cls.result_storage)
|
os.unlink(cls.result_storage)
|
||||||
except OSError:
|
except OSError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def _reinstall_reactor():
|
||||||
|
import sys
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
from twisted.internet import asyncioreactor
|
||||||
|
|
||||||
|
# Uninstall the reactor.
|
||||||
|
if "twisted.internet.reactor" in sys.modules:
|
||||||
|
del sys.modules["twisted.internet.reactor"]
|
||||||
|
|
||||||
|
# The daphne.server module may have already installed the reactor.
|
||||||
|
# If so, using this module will use uninstalled one, thus we should
|
||||||
|
# reimport this module too.
|
||||||
|
if "daphne.server" in sys.modules:
|
||||||
|
del sys.modules["daphne.server"]
|
||||||
|
|
||||||
|
event_loop = asyncio.new_event_loop()
|
||||||
|
asyncioreactor.install(event_loop)
|
||||||
|
asyncio.set_event_loop(event_loop)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user