class TestConnection(TestCase):
'''Integration Tests for Connection'''
def setUp(self):
self.pool = RoundRobinPool(settings())
def tearDown(self):
self.pool.disposeAll()
def testClient(self):
'''Returns the Cassandra.Client Connection that I have'''
connection = self.pool.get()
self.assertTrue(connection is not None)
self.assertTrue(connection.client is not None)
self.assertTrue(connection.open)
def testToPool(self):
'''Return this Connection to the Pool where it came from'''
connection = self.pool.get()
poolSize = self.pool.queue.qsize()
connection.toPool()
self.assertTrue(self.pool.queue.qsize() > poolSize)
def testDispose(self):
'''Close this connection and mark it as DISPOSED'''
connection = self.pool.get()
connection.dispose()
with self.assertRaises(ConnectionDisposedError):
connection.client
def setUp(self):
'''Create the Pool'''
print "Creating a Pool with the default connections"
self.pool = RoundRobinPool(settings())
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
def setUp(self):
self.pool = RoundRobinPool(settings())
