1

Author: dkropachev

tests/integration/experiments/test_tablets.py

@@ -3,7 +3,8 @@
 import pytest
 
 from cassandra.cluster import Cluster
-from cassandra.policies import ConstantReconnectionPolicy, RoundRobinPolicy, TokenAwarePolicy
+from cassandra.policies import ConstantReconnectionPolicy, RoundRobinPolicy, TokenAwarePolicy, \
+    NoDelayShardReconnectionPolicy
 
 from tests.integration import PROTOCOL_VERSION, use_cluster
 from tests.unit.test_host_connection_pool import LOGGER
@@ -21,7 +22,9 @@ class TestTabletsIntegration:
     def setup_class(cls):
         cls.cluster = Cluster(contact_points=["127.0.0.1", "127.0.0.2", "127.0.0.3"], protocol_version=PROTOCOL_VERSION,
                               load_balancing_policy=TokenAwarePolicy(RoundRobinPolicy()),
-                              reconnection_policy=ConstantReconnectionPolicy(1))
+                              reconnection_policy=ConstantReconnectionPolicy(1),
+                              shard_reconnection_policy=NoDelayShardReconnectionPolicy(),
+                              )
         cls.session = cls.cluster.connect()
         cls.create_ks_and_cf(cls.session)
         cls.create_data(cls.session)