def test_drop(self): pool = Pool(Mock()) conn1 = pool.acquire() pool.release(conn1) ok_(conn1 in pool._pool) pool.drop(conn1) ok_(conn1 not in pool._pool) conn1.close.assert_called_once_with()
def test_drop_closed_connection(self): pool = Pool(Mock()) conn1 = pool.acquire() conn1.is_connected = Mock() conn1.is_connected.return_value = False pool.release(conn1) ok_(conn1 in pool._pool) pool.drop(conn1) ok_(conn1 not in pool._pool) ok_(not conn1.close.called)
def test_drop_invalid_connection(self): pool = Pool(Mock()) pool.drop(Mock())
def test_drop_using_connection(self): pool = Pool(Mock()) conn1 = pool.acquire() pool.drop(conn1)