Make ClickHouse version configurable

Author: ods

tests/conftest.py

@@ -1,6 +1,7 @@
 import asyncio
 from datetime import datetime
 from decimal import Decimal
+import os
 
 import clickhouse_sqlalchemy
 import pytest
@@ -17,13 +18,34 @@ def pytest_collection_modifyitems(items):
             item.add_marker('asyncio')
 
 
+def pytest_addoption(parser):
+    parser.addoption(
+        '--clickhouse-version', default=None,
+        help='use specified version of ClickHouse',
+    )
+
+
 @pytest.fixture(scope='session')
-def dsn(docker_services):
+def dsn(request, docker_services):
+    ch_version = request.config.getoption('--clickhouse-version')
+    if ch_version is not None:
+        os.environ['CLICKHOUSE_VERSION'] = ch_version
     docker_services.start('clickhouse-server')
     public_port = docker_services.wait_for_service('clickhouse-server', 8123)
     return f'clickhouse://{docker_services.docker_ip}:{public_port}'
 
 
+@pytest.fixture
+async def clickhouse_version(dsn):
+    if not hasattr(pytest, 'clickhouse_version'):
+        async with aiochsa.connect(dsn) as conn:
+            version = await conn.fetchval('SELECT version()')
+            pytest.clickhouse_version = tuple(
+                int(num) for num in version.split('.')
+            )
+    return pytest.clickhouse_version
+
+
 @pytest.fixture
 async def pool(dsn):
     async with aiochsa.create_pool(dsn) as pool:

tests/docker-compose.yml

@@ -3,6 +3,6 @@ version: '2.1'
 services:
 
   clickhouse-server:
-    image: yandex/clickhouse-server
+    image: "yandex/clickhouse-server:${CLICKHOUSE_VERSION:-latest}"
     ports:
       - 8123

tests/test_types.py

@@ -175,7 +175,9 @@ async def test_as_is_round(conn, value):
     ],
     ids = parametrized_id,
 )
-async def test_zero_dates(conn, sa_type, value):
+async def test_zero_dates(clickhouse_version, conn, sa_type, value):
+    if clickhouse_version >= (20, 7):
+        pytest.skip('Feature is dropped in 20.7')
     result = await conn.fetchval(
         sa.select([sa.func.cast(value, sa_type)])
     )