diff options
| -rw-r--r-- | kafka/conn.py | 2 | ||||
| -rw-r--r-- | test/test_conn.py | 2 |
2 files changed, 3 insertions, 1 deletions
diff --git a/kafka/conn.py b/kafka/conn.py index 1243bdb..f30d987 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -278,7 +278,7 @@ class BrokerConnection(object): def connect(self): """Attempt to connect and return ConnectionState""" - if self.state is ConnectionStates.DISCONNECTED: + if self.state is ConnectionStates.DISCONNECTED and not self.blacked_out(): self.last_attempt = time.time() next_lookup = self._next_afi_host_port() if not next_lookup: diff --git a/test/test_conn.py b/test/test_conn.py index ef7925a..5698519 100644 --- a/test/test_conn.py +++ b/test/test_conn.py @@ -263,6 +263,7 @@ def test_lookup_on_connect(): ] with mock.patch("socket.getaddrinfo", return_value=mock_return2) as m: + conn.last_attempt = 0 conn.connect() m.assert_called_once_with(hostname, port, 0, 1) conn.close() @@ -288,6 +289,7 @@ def test_relookup_on_failure(): ] with mock.patch("socket.getaddrinfo", return_value=mock_return2) as m: + conn.last_attempt = 0 conn.connect() m.assert_called_once_with(hostname, port, 0, 1) conn.close() |