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 6b18d95dec)

Closes scylladb/scylladb#29146

Closes scylladb/scylladb#29366
This commit is contained in:
Michał Chojnowski
2026-03-18 20:28:23 +01:00
committed by Botond Dénes
parent 340369a4d4
commit 18fc2eff31

View File

@@ -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))