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()