Коммит
f56461b6f1
|
@ -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()
|
||||
|
|
Загрузка…
Ссылка в новой задаче