Skip to content

Commit c33eef4

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 dde7d2a commit c33eef4

File tree

1 file changed

+3
-15
lines changed

1 file changed

+3
-15
lines changed

tests/integration/standard/test_scylla_cloud.py

Lines changed: 3 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -8,22 +8,10 @@
88
from cassandra.policies import TokenAwarePolicy, RoundRobinPolicy, ConstantReconnectionPolicy
99
from tests.integration import use_cluster, PROTOCOL_VERSION
1010
from cassandra.cluster import Cluster, TwistedConnection
11+
from tests.integration import use_cluster
12+
from cassandra.cluster import Cluster, get_all_supported_connections_classes
1113

12-
13-
supported_connection_classes = [TwistedConnection]
14-
15-
try:
16-
from cassandra.io.libevreactor import LibevConnection
17-
supported_connection_classes += [LibevConnection]
18-
except ImportError:
19-
pass
20-
21-
22-
try:
23-
from cassandra.io.asyncorereactor import AsyncoreConnection
24-
supported_connection_classes += [AsyncoreConnection]
25-
except ImportError:
26-
pass
14+
supported_connection_classes, _ = get_all_supported_connections_classes()
2715

2816
#from cassandra.io.geventreactor import GeventConnection
2917
#from cassandra.io.eventletreactor import EventletConnection

0 commit comments

Comments
 (0)