示例#1
0
    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