From 18fc2eff3120be9beff948c9293de5aa9050cb63 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Chojnowski?= Date: Wed, 18 Mar 2026 20:28:23 +0100 Subject: [PATCH] test: add a missing reconnect_driver in test_sstable_compression_dictionaries_upgrade.py Need to work around https://github.com/scylladb/python-driver/issues/295, lest a CQL query fail spuriously after the cluster restart. Fixes: SCYLLADB-1114 Closes scylladb/scylladb#29118 (cherry picked from commit 6b18d95dec968e4f1b87589719adf11aa30c701e) Closes scylladb/scylladb#29146 Closes scylladb/scylladb#29366 --- test/cluster/test_sstable_compression_dictionaries_upgrade.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/cluster/test_sstable_compression_dictionaries_upgrade.py b/test/cluster/test_sstable_compression_dictionaries_upgrade.py index c6218f2fb8..5ad9b351b7 100644 --- a/test/cluster/test_sstable_compression_dictionaries_upgrade.py +++ b/test/cluster/test_sstable_compression_dictionaries_upgrade.py @@ -15,6 +15,7 @@ from test.pylib.manager_client import ManagerClient, ServerInfo from test.pylib.rest_client import read_barrier, HTTPError from test.pylib.scylla_cluster import ScyllaVersionDescription from test.pylib.util import wait_for_cql_and_get_hosts, wait_for_feature +from test.cluster.util import reconnect_driver from cassandra.cluster import ConsistencyLevel from cassandra.policies import FallthroughRetryPolicy from cassandra.protocol import ServerError @@ -162,6 +163,7 @@ async def test_upgrade_and_rollback(manager: ManagerClient, scylla_2025_1: Scyll ) logger.info("Waiting for SSTABLE_COMPRESSION_DICTS cluster feature") + cql = await reconnect_driver(manager) hosts = await wait_for_cql_and_get_hosts(cql, servers, time.time() + 60) await asyncio.gather(*(wait_for_feature("SSTABLE_COMPRESSION_DICTS", cql, h, time.time() + 60) for h in hosts))