Merge pull request #2454 from lukeburden/master

Fixes issue #2453
This commit is contained in:
Ionel Cristian Mărieș 2015-01-24 10:38:46 +02:00
Родитель 3f713bed7c 3c25f3abdd
Коммит f56461b6f1
2 изменённых файлов: 5 добавлений и 2 удалений

Просмотреть файл

@ -227,7 +227,7 @@ class DjangoWorkerFixup(object):
def _close_database(self):
try:
funs = [conn.close for conn in self._db.connections]
funs = [conn.close for conn in self._db.connections.all()]
except AttributeError:
if hasattr(self._db, 'close_old_connections'): # django 1.6
funs = [self._db.close_old_connections]

Просмотреть файл

@ -205,10 +205,13 @@ class test_DjangoWorkerFixup(FixupCase):
def test__close_database(self):
with self.fixup_context(self.app) as (f, _, _):
conns = f._db.connections = [Mock(), Mock(), Mock()]
conns = [Mock(), Mock(), Mock()]
conns[1].close.side_effect = KeyError('already closed')
f.database_errors = (KeyError, )
f._db.connections = Mock() # ConnectionHandler
f._db.connections.all.side_effect = lambda: conns
f._close_database()
conns[0].close.assert_called_with()
conns[1].close.assert_called_with()