mirror of
https://github.com/hanzo-ds/connect.git
synced 2026-10-30 20:31:24 +00:00
zeekay
999d38466d
Imports read `datastore_connect`, distribution `datastore-connect`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.8.17 — 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
2.3 KiB
Python
54 lines
2.3 KiB
Python
# pylint: disable=no-member
|
|
import sqlalchemy as db
|
|
from sqlalchemy.engine import Engine
|
|
|
|
from datastore_connect import common
|
|
from datastore_connect.cc_sqlalchemy.datatypes.sqltypes import UInt32, SimpleAggregateFunction, Point
|
|
|
|
|
|
def test_basic_reflection(test_engine: Engine):
|
|
common.set_setting('invalid_setting_action', 'drop')
|
|
conn = test_engine.connect()
|
|
metadata = db.MetaData(bind=test_engine, schema='system')
|
|
table = db.Table('tables', metadata, autoload_with=test_engine)
|
|
query = db.select([table.columns.create_table_query])
|
|
result = conn.execute(query)
|
|
rows = result.fetchmany(100)
|
|
assert rows
|
|
|
|
|
|
def test_full_table_reflection(test_engine: Engine, test_db: str):
|
|
common.set_setting('invalid_setting_action', 'drop')
|
|
conn = test_engine.connect()
|
|
conn.execute(f'DROP TABLE IF EXISTS {test_db}.reflect_test')
|
|
conn.execute(
|
|
f'CREATE TABLE {test_db}.reflect_test (key UInt32, value FixedString(20),'+
|
|
'agg SimpleAggregateFunction(anyLast, String))' +
|
|
'ENGINE AggregatingMergeTree ORDER BY key')
|
|
metadata = db.MetaData(bind=test_engine, schema=test_db)
|
|
table = db.Table('reflect_test', metadata, autoload_with=test_engine)
|
|
assert table.columns.key.type.__class__ == UInt32
|
|
assert table.columns.agg.type.__class__ == SimpleAggregateFunction
|
|
assert 'MergeTree' in table.engine.name
|
|
|
|
|
|
def test_types_reflection(test_engine: Engine, test_db: str):
|
|
common.set_setting('invalid_setting_action', 'drop')
|
|
conn = test_engine.connect()
|
|
conn.execute(f'DROP TABLE IF EXISTS {test_db}.sqlalchemy_types_test')
|
|
conn.execute(
|
|
f'CREATE TABLE {test_db}.sqlalchemy_types_test (key UInt32, pt Point) ' +
|
|
'ENGINE MergeTree ORDER BY key')
|
|
metadata = db.MetaData(bind=test_engine, schema=test_db)
|
|
table = db.Table('sqlalchemy_types_test', metadata, autoload_with=test_engine)
|
|
assert table.columns.key.type.__class__ == UInt32
|
|
assert table.columns.pt.type.__class__ == Point
|
|
assert 'MergeTree' in table.engine.name
|
|
|
|
|
|
def test_table_exists(test_engine: Engine):
|
|
common.set_setting('invalid_setting_action', 'drop')
|
|
conn = test_engine.connect()
|
|
assert test_engine.dialect.has_table(conn, 'columns', 'system')
|
|
assert not test_engine.dialect.has_table(conn, 'nope', 'fake_db')
|