def test_simple_pool_of_one():
pool = ThriftConnectionPool(ENDPOINTS, EzPz.Client, size=1)
conn = pool.get_connection()
try:
nt.assert_equal('pz', conn.ez())
finally:
pool.return_connection(conn)
def test_simple_pool_of_one():
pool = ThriftConnectionPool(ENDPOINTS, EzPz.Client, size=1, use_ssl=True,
ca_certs=clientcapath, cert=clientcertpath, key=clientprivpath)
conn = pool.get_connection()
try:
nt.assert_equal('pz', conn.ez())
finally:
pool.return_connection(conn)
def test_queuing_behavior():
pool = ThriftConnectionPool(ENDPOINTS, EzPz.Client, 1)
conn = pool.get_connection()
nt.assert_equal(0, pool._connection_queue.qsize())
try:
nt.assert_equal('pz', conn.ez())
finally:
pool.return_connection(conn)
nt.assert_equal(1, pool._connection_queue.qsize())
def test_semaphore_behavior():
pool = ThriftConnectionPool(ENDPOINTS, EzPz.Client, 1)
conn = pool.get_connection()
nt.assert_equal(0, pool._semaphore._Semaphore__value)
try:
nt.assert_equal('pz', conn.ez())
finally:
pool.return_connection(conn)
nt.assert_equal(1, pool._semaphore._Semaphore__value)
def test_queuing_behavior():
pool = ThriftConnectionPool(ENDPOINTS, EzPz.Client, 1, use_ssl=True,
ca_certs=clientcapath, cert=clientcertpath, key=clientprivpath)
conn = pool.get_connection()
nt.assert_equal(0, pool._connection_queue.qsize())
try:
nt.assert_equal('pz', conn.ez())
finally:
pool.return_connection(conn)
nt.assert_equal(1, pool._connection_queue.qsize())
