mirror of
https://github.com/hanzo-ds/pool.git
synced 2026-10-30 20:37:50 +00:00
zeekay
b958f3cd51
Imports read `datastore_pool`, distribution `datastore-pool`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.5.3 — the version this stack already ran — rather than from upstream HEAD, so it is a PURE RENAME with no behaviour change. Forking from HEAD would have dragged versions along with the name (connect alone jumped 0.8.17 -> 1.6.0, a major), which is a different and much riskier change wearing a rename's clothes. X-ClickHouse-* headers are untouched: they are wire protocol the server sends and expects, and a different token shape from the module name.
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
import pytest
|
|
from datastore_pool.pool import ChPool, TooManyConnections
|
|
|
|
|
|
def test_context():
|
|
pool = ChPool()
|
|
with pool.get_client() as client:
|
|
result = client.execute("SELECT * FROM system.numbers LIMIT 5;")
|
|
assert result == [(0,), (1,), (2,), (3,), (4,)]
|
|
assert len(pool._used) == 1
|
|
assert len(pool._used) == 0
|
|
pool.cleanup()
|
|
|
|
|
|
def test_connections_min():
|
|
pool = ChPool(connections_min=5, connections_max=10)
|
|
clients = []
|
|
for i in range(1, 6):
|
|
clients.append(pool.pull())
|
|
assert len(pool._used) == i
|
|
clients.append(pool.pull())
|
|
assert len(pool._pool) == 0
|
|
assert len(pool._used) == 6
|
|
for client in clients:
|
|
client.execute("SELECT * FROM system.numbers LIMIT 5;")
|
|
pool.push(client=client)
|
|
assert len(pool._pool) == 5
|
|
pool.cleanup()
|
|
|
|
|
|
def test_connections_max():
|
|
pool = ChPool(connections_min=1, connections_max=3)
|
|
clients = []
|
|
for i in range(3):
|
|
clients.append(pool.pull())
|
|
assert len(pool._used) == i + 1
|
|
with pytest.raises(TooManyConnections):
|
|
clients.append(pool.pull())
|
|
for client in clients:
|
|
client.execute("SELECT * FROM system.numbers LIMIT 5;")
|
|
pool.push(client=client)
|
|
pool.cleanup()
|
|
|
|
|
|
def test_connection_released_on_error():
|
|
pool = ChPool()
|
|
try:
|
|
with pool.get_client() as client:
|
|
client.execute("some_invalid_sql")
|
|
except:
|
|
pass
|
|
assert len(pool._used) == 0
|
|
pool.cleanup()
|