Merge pull request apache#327 from thoslin/thoslin-patch-1

aholmberg · aholmberg · commit ccd7e460e28c · 2015-05-26T08:23:08.000-05:00
Check if max_attempts is None for ConstantReconnectionPolicy
diff --git a/cassandra/policies.py b/cassandra/policies.py
@@ -509,14 +509,16 @@ def __init__(self, delay, max_attempts=64):
         """
         if delay < 0:
             raise ValueError("delay must not be negative")
-        if max_attempts < 0:
+        if max_attempts is not None and max_attempts < 0:
             raise ValueError("max_attempts must not be negative")
 
         self.delay = delay
         self.max_attempts = max_attempts
 
     def new_schedule(self):
-        return repeat(self.delay, self.max_attempts)
+        if self.max_attempts:
+            return repeat(self.delay, self.max_attempts)
+        return repeat(self.delay)
 
 
 class ExponentialReconnectionPolicy(ReconnectionPolicy):
