Skip to content

Commit 4b54a84

Browse files
committed
fix ScyllaCloudConfigTests to get all available connection classes
Prior commit made `get_all_supported_connections_classes` available, let's use it instead of custom implementation.
1 parent 7452376 commit 4b54a84

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

tests/integration/standard/test_scylla_cloud.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,9 @@
55
from ccmlib.utils.sni_proxy import refresh_certs, get_cluster_info, start_sni_proxy, create_cloud_config
66

77
from tests.integration import use_cluster
8-
from cassandra.cluster import Cluster, TwistedConnection
8+
from cassandra.cluster import Cluster, get_all_supported_connections_classes
99

10-
11-
from cassandra.io.libevreactor import LibevConnection
12-
supported_connection_classes = [LibevConnection, TwistedConnection]
13-
try:
14-
from cassandra.io.asyncorereactor import AsyncoreConnection
15-
supported_connection_classes += [AsyncoreConnection]
16-
except ImportError:
17-
pass
10+
supported_connection_classes, _ = get_all_supported_connections_classes()
1811

1912
#from cassandra.io.geventreactor import GeventConnection
2013
#from cassandra.io.eventletreactor import EventletConnection

0 commit comments

Comments
 (0)