1

Author: dkropachev

cassandra/policies.py

@@ -887,10 +887,10 @@ def new_scheduler(self, session) -> ShardReconnectionScheduler:
 
 class NoDelayShardReconnectionPolicy(ShardReconnectionPolicy):
     def new_scheduler(self, session) -> ShardReconnectionScheduler:
-        return NoDelayShardReconnectionScheduler(session)
+        return _NoDelayShardReconnectionScheduler(session)
 
 
-class NoDelayShardReconnectionScheduler(ShardReconnectionScheduler):
+class _NoDelayShardReconnectionScheduler(ShardReconnectionScheduler):
     def __init__(self, session):
         self.session = weakref.proxy(session)
         self.already_scheduled = {}
@@ -926,7 +926,7 @@ def __init__(self, shard_reconnection_scope, reconnection_policy):
         self.reconnection_policy = reconnection_policy
 
     def new_scheduler(self, session) -> ShardReconnectionScheduler:
-        return NoConcurrentShardReconnectionScheduler(session, self.shard_reconnection_scope, self.reconnection_policy)
+        return _NoConcurrentShardReconnectionScheduler(session, self.shard_reconnection_scope, self.reconnection_policy)
 
 
 class _ScopeBucket(object):
@@ -980,7 +980,7 @@ def run(self):
             self._schedule()
 
 
-class NoConcurrentShardReconnectionScheduler(ShardReconnectionScheduler):
+class _NoConcurrentShardReconnectionScheduler(ShardReconnectionScheduler):
     def __init__(self, session, shard_reconnection_scope, reconnection_policy):
         self.already_scheduled = {}
         self.scopes = {}

tests/unit/test_host_connection_pool.py

@@ -26,7 +26,7 @@
 from cassandra.connection import Connection
 from cassandra.pool import HostConnection, HostConnectionPool
 from cassandra.pool import Host, NoConnectionsAvailable
-from cassandra.policies import HostDistance, SimpleConvictionPolicy, NoDelayShardReconnectionScheduler
+from cassandra.policies import HostDistance, SimpleConvictionPolicy, _NoDelayShardReconnectionScheduler
 
 LOGGER = logging.getLogger(__name__)
 
@@ -41,7 +41,7 @@ def make_session(self):
         session.cluster.get_core_connections_per_host.return_value = 1
         session.cluster.get_max_requests_per_connection.return_value = 1
         session.cluster.get_max_connections_per_host.return_value = 1
-        session.shard_reconnection_scheduler = NoDelayShardReconnectionScheduler(session)
+        session.shard_reconnection_scheduler = _NoDelayShardReconnectionScheduler(session)
         session.is_shutdown = False
         return session