|
@@ -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()
|
|
|
+ f._db.connections.all.side_effect = lambda: conns
|
|
|
+
|
|
|
f._close_database()
|
|
|
conns[0].close.assert_called_with()
|
|
|
conns[1].close.assert_called_with()
|