def test_null_pool(self):
listener = _TestListener()
pool = NullPool(keyspace='Keyspace1', credentials=_credentials,
listeners=[listener])
conn = pool.get()
pool.return_conn(conn)
assert_equal(listener.checkout_count, 1)
assert_equal(listener.checkin_count, 1)
def test_null_pool_failover(self):
listener = _TestListener()
pool = NullPool(keyspace='Keyspace1', credentials=_credentials,
listeners=[listener], use_threadlocal=False,
server_list=['localhost:9160', 'localhost:9160'])
conn = pool.get()
cf = ColumnFamily(conn, 'Standard1')
for i in range(1,5):
setattr(cf.client._connection.client, 'batch_mutate', _timeout)
# The first insert attempt should fail, but failover should occur
# and the insert should succeed
cf.insert('key', {'col': 'val'})
assert_equal(listener.failure_count, i)
cf.get('key')
pool.dispose()
listener.reset()
pool = NullPool(keyspace='Keyspace1', credentials=_credentials,
listeners=[listener], use_threadlocal=False,
server_list=['localhost:9160', 'localhost:9160'])
threads = []
args = (pool, 'key', {'col':'val'})
for i in range(0, 5):
threads.append(threading.Thread(target=_five_fails, args=args))
threads[-1].start()
for thread in threads:
thread.join()
assert_equal(listener.failure_count, 25)
pool.dispose()
def test_server_list_func(self):
listener = _TestListener()
pool = NullPool(keyspace='Keyspace1', server_list=_get_list,
listeners=[listener])
assert_equal(listener.serv_list, ['foo:bar'])
assert_equal(listener.list_count, 1)
def test_pool_connection_failure(self):
listener = _TestListener()
def get_extra():
"""Make failure count adjustments based on whether or not
the permuted list starts with a good host:port"""
if listener.serv_list[0] == 'localhost:9160':
return 0
else:
return 1
pool = QueuePool(pool_size=5, max_overflow=5, recycle=10000,
prefill=True,
keyspace='Keyspace1', credentials=_credentials,
timeout=0.05,
listeners=[listener], use_threadlocal=False,
server_list=['localhost:9160', 'foobar:1'])
assert_equal(listener.failure_count, 4 + get_extra())
for i in range(0,7):
pool.get()
assert_equal(listener.failure_count, 6 + get_extra())
pool.dispose()
listener.reset()
pool = QueuePool(pool_size=5, max_overflow=5, recycle=10000,
prefill=True,
keyspace='Keyspace1', credentials=_credentials,
timeout=0.05,
listeners=[listener], use_threadlocal=True,
server_list=['localhost:9160', 'foobar:1'])
assert_equal(listener.failure_count, 4 + get_extra())
threads = []
for i in range(0, 7):
threads.append(threading.Thread(target=pool.get))
threads[-1].start()
for thread in threads:
thread.join()
assert_equal(listener.failure_count, 6 + get_extra())
pool.dispose()
listener.reset()
pool = SingletonThreadPool(pool_size=5, keyspace='Keyspace1',
credentials=_credentials, listeners=[listener],
timeout=0.05,
server_list=['localhost:9160', 'foobar:1'])
threads = []
for i in range(0, 5):
threads.append(threading.Thread(target=pool.get))
threads[-1].start()
for thread in threads:
thread.join()
assert_equal(listener.failure_count, 4 + get_extra())
pool.dispose()
listener.reset()
pool = StaticPool(keyspace='Keyspace1', credentials=_credentials,
timeout=0.05, listeners=[listener],
server_list=['localhost:9160', 'foobar:1'])
pool.get()
assert_equal(listener.failure_count, 0 + get_extra())
pool.dispose()
listener.reset()
pool = NullPool(keyspace='Keyspace1', credentials=_credentials,
timeout=0.05,
listeners=[listener], use_threadlocal=False,
server_list=['localhost:9160', 'foobar:1'])
for i in range(0, 5):
pool.get()
assert_equal(listener.failure_count, 4 + get_extra())
pool.dispose()
listener.reset()
pool = AssertionPool(keyspace='Keyspace1', credentials=_credentials,
timeout=0.05,
listeners=[listener], use_threadlocal=False,
server_list=['localhost:9160', 'foobar:1'])
while True:
if listener.serv_list[0] == 'foobar:1':
break
pool.set_server_list(['localhost:9160', 'foobar:1'])
conn = pool.get()
assert_equal(listener.failure_count, 1)
pool.dispose()
listener.reset()
pool = AssertionPool(keyspace='Keyspace1', credentials=_credentials,
timeout=0.05,
listeners=[listener], use_threadlocal=True,
server_list=['localhost:9160', 'foobar:1'])
while True:
if listener.serv_list[0] == 'foobar:1':
break
pool.set_server_list(['localhost:9160', 'foobar:1'])
conn = pool.get()
assert_equal(listener.failure_count, 1)
pool.dispose()
