From 70feea2c903462f68782c6da2e69bf1c64b396d2 Mon Sep 17 00:00:00 2001 From: GregBestland Date: Thu, 15 Sep 2016 18:27:10 -0500 Subject: [PATCH] Tweaking heartbeat timeout test to not fail with protocol v2 --- tests/integration/standard/test_connection.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/standard/test_connection.py b/tests/integration/standard/test_connection.py index 8b4b0430..4beedede 100644 --- a/tests/integration/standard/test_connection.py +++ b/tests/integration/standard/test_connection.py @@ -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: