def test_close(self): """ You can close all the connections """ c1 = MagicMock() c2 = MagicMock() pool = ConnectionPool() pool.add(c1) pool.add(c2) d = pool.close() d.addCallback(lambda _: c1.close.assert_called_once_with()) d.addCallback(lambda _: c2.close.assert_called_once_with()) return d