Tweaking heartbeat timeout test to not fail with protocol v2

This commit is contained in:
GregBestland
2016-09-15 18:27:10 -05:00
parent 4fba5514ab
commit 70feea2c90

View File

@@ -143,7 +143,7 @@ class HeartbeatTest(unittest.TestCase):
for conn in holders:
if host == str(getattr(conn, 'host', '')):
if isinstance(conn, HostConnectionPool):
if conn._connections is not None:
if conn._connections is not None and len(conn._connections) > 0:
connections.append(conn._connections)
else:
if conn._connection is not None:
@@ -162,7 +162,7 @@ class HeartbeatTest(unittest.TestCase):
def wait_for_no_connections(self, host, cluster):
retry = 0
while(retry < 200):
while(retry < 100):
retry += 1
connections = self.fetch_connections(host, cluster)
if len(connections) is 0: