def test_singleton_pool_threadlocal(self): """Should be the same as non-threadlocal.""" listener = _TestListener() pool = SingletonThreadPool(keyspace='Keyspace1', credentials=_credentials, pool_size=5, listeners=[listener], use_threadlocal=True) # Make sure we get the same connection every time conn = pool.get() assert_equal(pool.get(), conn) for i in range(10): conn = pool.get() assert_equal(listener.connect_count, 1) pool.return_conn(conn) conn = pool.get() assert_equal(pool.get(), conn) assert_equal(listener.connect_count, 2) pool.return_conn(conn) def get_return(): conn = pool.get() print pool.status() threads = [] for i in range(5): threads.append(threading.Thread(target=get_return)) threads[-1].start() for thread in threads: thread.join() assert_equal(listener.connect_count, 7) assert_raises(NoConnectionAvailable, pool.get) assert_equal(listener.max_count, 1)
def test_singleton_thread_failover(self): listener = _TestListener() pool = SingletonThreadPool(pool_size=5, keyspace='Keyspace1', credentials=_credentials, listeners=[listener], server_list=['localhost:9160', 'localhost:9160']) conn = pool.get() cf = ColumnFamily(conn, 'Standard1') for i in range(1,5): setattr(cf.client._local.conn.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 = SingletonThreadPool(pool_size=5, keyspace='Keyspace1', credentials=_credentials, listeners=[listener], server_list=['localhost:9160', 'localhost:9160']) threads = [] args = (pool, 'key', {'col':'val'}) for j in range(0,5): threads.append(threading.Thread(target=_five_tlocal_fails, args=args)) threads[-1].start() for thread in threads: thread.join() assert_equal(listener.failure_count, 25)