mirror of
https://github.com/hanzo-ds/connect.git
synced 2026-10-30 20:31:24 +00:00
zeekay
75dd1132bd
The wire protocol was the last holdout and it was not cosmetic: the Hanzo Datastore server reads X-Datastore-User, and answers a request bearing X-ClickHouse-User with 401. Verified against the live server, both directions. So a client still sending the old headers is not merely off-brand, it cannot authenticate over HTTP. Everything else follows: types, dialects, error modules, docstrings and doc links now name Datastore, and doc URLs point at docs.hanzo.ai/datastore rather than the upstream vendor's site. Zero occurrences of the old name remain. Still cut from the pinned upstream version, so behaviour is unchanged.
167 lines
7.2 KiB
Python
167 lines
7.2 KiB
Python
import os
|
|
import time
|
|
from datetime import datetime
|
|
|
|
import pytz
|
|
|
|
from datastore_connect.driver import Client, tzutil
|
|
|
|
# We have to localize a datetime from a timezone to get a current, sensible timezone object for testing. See
|
|
# https://stackoverflow.com/questions/35462876/python-pytz-timezone-function-returns-a-timezone-that-is-off-by-9-minutes
|
|
chicago_tz = pytz.timezone('America/Chicago').localize(datetime(2020, 8, 8, 10, 5, 5)).tzinfo
|
|
|
|
|
|
def test_basic_timezones(test_client: Client):
|
|
row = test_client.query("SELECT toDateTime('2022-10-25 10:55:22', 'America/Chicago') as chicago," +
|
|
"toDateTime('2023-07-05 15:10:40') as utc",
|
|
query_tz='America/Chicago').first_row
|
|
|
|
assert row[0].tzinfo == chicago_tz
|
|
assert row[0].hour == 10
|
|
assert row[0].day == 25
|
|
|
|
assert row[1].tzinfo == chicago_tz
|
|
assert row[1].hour == 10
|
|
assert row[1].day == 5
|
|
|
|
if test_client.min_version('20'):
|
|
row = test_client.query("SELECT toDateTime64('2022-10-25 10:55:22.789123', 6, 'America/Chicago')",
|
|
query_tz='America/Chicago').first_row
|
|
assert row[0].tzinfo == chicago_tz
|
|
assert row[0].hour == 10
|
|
assert row[0].day == 25
|
|
assert row[0].microsecond == 789123
|
|
|
|
|
|
def test_server_timezone(test_client: Client):
|
|
# This test is really for manual testing since changing the timezone on the test Datastore server
|
|
# still requires a restart. Other tests will depend on https://github.com/Datastore/Datastore/pull/44149
|
|
test_client.apply_server_timezone = True
|
|
test_datetime = datetime(2023, 3, 18, 16, 4, 25)
|
|
try:
|
|
date = test_client.query('SELECT toDateTime(%s) as st', parameters=[test_datetime]).first_row[0]
|
|
if test_client.server_tz == pytz.UTC:
|
|
assert date.tzinfo is None
|
|
assert date == datetime(2023, 3, 18, 16, 4, 25, tzinfo=None)
|
|
assert date.timestamp() == 1679155465
|
|
else:
|
|
den_tz = pytz.timezone('America/Denver').localize(datetime(2020, 8, 8)).tzinfo
|
|
assert date == datetime(2023, 3, 18, 16, 4, 25, tzinfo=den_tz)
|
|
assert date.tzinfo == den_tz
|
|
assert date.timestamp() == 1679177065
|
|
finally:
|
|
test_client.apply_server_timezone = False
|
|
|
|
|
|
def test_column_timezones(test_client: Client):
|
|
date_tz64 = "toDateTime64('2023-01-02 15:44:22.7832', 6, 'Asia/Shanghai')"
|
|
if not test_client.min_version('20'):
|
|
date_tz64 = "toDateTime('2023-01-02 15:44:22', 'Asia/Shanghai')"
|
|
column_tzs = {'chicago': 'America/Chicago', 'china': 'Asia/Shanghai'}
|
|
row = test_client.query("SELECT toDateTime('2022-10-25 10:55:22', 'America/Chicago') as chicago," +
|
|
f'{date_tz64} as china,' +
|
|
"toDateTime('2023-07-05 15:10:40') as utc",
|
|
column_tzs=column_tzs).first_row
|
|
china_tz = pytz.timezone('Asia/Shanghai').localize(datetime(2024, 12, 4, 10, 5, 5)).tzinfo
|
|
assert row[0].tzinfo == chicago_tz
|
|
assert row[1].tzinfo == china_tz
|
|
assert row[2].tzinfo is None
|
|
|
|
if test_client.min_version('20'):
|
|
row = test_client.query("SELECT toDateTime('2022-10-25 10:55:22', 'America/Chicago') as chicago," +
|
|
"toDateTime64('2023-01-02 15:44:22.7832', 6, 'Asia/Shanghai') as china").first_row
|
|
if test_client.protocol_version:
|
|
assert row[0].tzinfo == chicago_tz
|
|
else:
|
|
assert row[0].tzinfo is None
|
|
assert row[1].tzinfo == china_tz # DateTime64 columns work correctly
|
|
|
|
|
|
def test_local_timezones(test_client: Client):
|
|
denver_tz = pytz.timezone('America/Denver')
|
|
tzutil.local_tz = denver_tz
|
|
test_client.apply_server_timezone = False
|
|
try:
|
|
row = test_client.query("SELECT toDateTime('2022-10-25 10:55:22'," +
|
|
"'America/Chicago') as chicago," +
|
|
"toDateTime('2023-07-05 15:10:40') as raw_utc_dst," +
|
|
"toDateTime('2023-07-05 12:44:22', 'UTC') as forced_utc," +
|
|
"toDateTime('2023-12-31 17:00:55') as raw_utc_std").first_row
|
|
if test_client.protocol_version:
|
|
assert row[0].tzinfo.tzname(None) == chicago_tz.tzname(None)
|
|
else:
|
|
assert row[0].tzinfo.tzname(None) == denver_tz.tzname(None)
|
|
assert row[1].tzinfo.tzname(None) == denver_tz.tzname(None)
|
|
assert row[2].tzinfo is None
|
|
assert row[3].tzinfo.tzname(None) == denver_tz.tzname(None)
|
|
finally:
|
|
tzutil.local_tz = pytz.UTC
|
|
test_client.apply_server_timezone = True
|
|
|
|
|
|
def test_naive_timezones(test_client: Client):
|
|
row = test_client.query("SELECT toDateTime('2022-10-25 10:55:22', 'America/Chicago') as chicago," +
|
|
"toDateTime('2023-07-05 15:10:40') as utc").first_row
|
|
|
|
if test_client.protocol_version:
|
|
assert row[0].tzinfo == chicago_tz
|
|
else:
|
|
assert row[0].tzinfo is None
|
|
assert row[1].tzinfo is None
|
|
|
|
|
|
def test_timezone_binding_client(test_client: Client):
|
|
os.environ['TZ'] = 'America/Denver'
|
|
time.tzset()
|
|
denver_tz = pytz.timezone('America/Denver')
|
|
tzutil.local_tz = denver_tz
|
|
test_client.apply_server_timezone = False
|
|
denver_time = datetime(2023, 3, 18, 16, 4, 25, tzinfo=denver_tz)
|
|
try:
|
|
server_time = test_client.query(
|
|
'SELECT toDateTime(%(dt)s) as dt', parameters={'dt': denver_time}).first_row[0]
|
|
assert server_time == denver_time
|
|
finally:
|
|
os.environ['TZ'] = 'UTC'
|
|
tzutil.local_tz = pytz.UTC
|
|
time.tzset()
|
|
test_client.apply_server_timezone = True
|
|
|
|
naive_time = datetime(2023, 3, 18, 16, 4, 25)
|
|
server_time = test_client.query(
|
|
'SELECT toDateTime(%(dt)s) as dt', parameters={'dt': naive_time}).first_row[0]
|
|
assert server_time.astimezone(pytz.UTC) == naive_time.astimezone(pytz.UTC)
|
|
|
|
utc_time = datetime(2023, 3, 18, 16, 4, 25, tzinfo=pytz.UTC)
|
|
server_time = test_client.query(
|
|
'SELECT toDateTime(%(dt)s) as dt', parameters={'dt': utc_time}).first_row[0]
|
|
assert server_time.astimezone(pytz.UTC) == utc_time
|
|
|
|
|
|
def test_timezone_binding_server(test_client: Client):
|
|
os.environ['TZ'] = 'America/Denver'
|
|
time.tzset()
|
|
denver_tz = pytz.timezone('America/Denver')
|
|
tzutil.local_tz = denver_tz
|
|
test_client.apply_server_timezone = False
|
|
denver_time = datetime(2022, 3, 18, 16, 4, 25, tzinfo=denver_tz)
|
|
try:
|
|
server_time = test_client.query(
|
|
'SELECT toDateTime({dt:DateTime}) as dt', parameters={'dt': denver_time}).first_row[0]
|
|
assert server_time == denver_time
|
|
finally:
|
|
os.environ['TZ'] = 'UTC'
|
|
time.tzset()
|
|
tzutil.local_tz = pytz.UTC
|
|
test_client.apply_server_timezone = True
|
|
|
|
naive_time = datetime(2022, 3, 18, 16, 4, 25)
|
|
server_time = test_client.query(
|
|
'SELECT toDateTime({dt:DateTime}) as dt', parameters={'dt': naive_time}).first_row[0]
|
|
assert naive_time.astimezone(pytz.UTC) == server_time.astimezone(pytz.UTC)
|
|
|
|
utc_time = datetime(2020, 3, 18, 16, 4, 25, tzinfo=pytz.UTC)
|
|
server_time = test_client.query(
|
|
'SELECT toDateTime({dt:DateTime}) as dt', parameters={'dt': utc_time}).first_row[0]
|
|
assert server_time.astimezone(pytz.UTC) == utc_time
|