django/daphne
1
1
Fork 0
mirror of https://github.com/django/daphne.git synced 2025-07-13 01:12:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed #462: Don't actually close DB connections during tests

Browse Source
This commit is contained in:
Andrew Godwin 2017-01-25 16:59:35 -08:00
parent 9942c59851
commit 9ae27cb835
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
channels/tests/base.py
View File

@ -7,6 +7,7 @@ from functools import wraps
from asgiref.inmemory import ChannelLayer as InMemoryChannelLayer from asgiref.inmemory import ChannelLayer as InMemoryChannelLayer
from django.test.testcases import TestCase, TransactionTestCase from django.test.testcases import TestCase, TransactionTestCase
from django.db import close_old_connections
from .. import DEFAULT_CHANNEL_LAYER from .. import DEFAULT_CHANNEL_LAYER
from ..asgi import ChannelLayerWrapper, channel_layers from ..asgi import ChannelLayerWrapper, channel_layers
@ -134,7 +135,10 @@ class Client(object):
consumer_started.send(sender=self.__class__) consumer_started.send(sender=self.__class__)
return consumer(message, **kwargs) return consumer(message, **kwargs)
finally: finally:
# Copy Django's workaround so we don't actually close DB conns
consumer_finished.disconnect(close_old_connections)
consumer_finished.send(sender=self.__class__) consumer_finished.send(sender=self.__class__)
consumer_finished.connect(close_old_connections)
elif fail_on_none: elif fail_on_none:
raise AssertionError("Can't find consumer for message %s" % message) raise AssertionError("Can't find consumer for message %s" % message)
elif fail_on_none: elif fail_on_none: