class TestRoundRobinPool(TestCase):
'''Tests a RoundRobin Pool...'''
def setUp(self):
'''Create the Pool'''
print "Creating a Pool with the default connections"
self.pool = RoundRobinPool(settings())
def tearDown(self):
'''Dispose all alive connections in the Pool'''
self.pool.disposeAll()
def testGet(self):
'''Yields a valid connection to this Keyspace'''
connection = self.pool.get()
self.assertTrue(connection is not None)
self.assertTrue(connection.client is not None)
self.assertTrue(connection.open)
def testPut(self):
"""Returns a Connection to the pool"""
connection = self.pool.get()
self.pool.put(connection)
assert self.pool.queue.qsize() == 1
def testDisposeAll(self):
'''Disposes all the Connections in the Pool, typically called at System Exit'''
cons = []
for i in range(5):
conn = self.pool.get()
cons.append(conn)
for i in cons:
self.pool.put(i)
print self.pool.queue.qsize()
self.pool.disposeAll()
assert self.pool.queue.qsize() == 0
@skip("Takes to Long to Run..")
def testEviction(self):
'''Checks if Idle connections are eventually evicted from the Connection Pool'''
cons = []
for i in range(15):
conn = self.pool.get()
cons.append(conn)
for i in cons:
self.pool.put(i)
time.sleep(45)
print self.pool.queue.qsize()
assert self.pool.queue.qsize() == self.pool.maxIdle
