Mock dns lookups in test_connmock_dns_test_conn

1 files changed, 8 insertions, 1 deletions

diff --git a/test/test_conn.py b/test/test_conn.py
index 953c112..66b8a0c 100644
--- a/test/test_conn.py
+++ b/test/test_conn.py
@@ -17,6 +17,13 @@ import kafka.errors as Errors
 
 
 @pytest.fixture
+def dns_lookup(mocker):
+    return mocker.patch('kafka.conn.dns_lookup',
+                        return_value=[(socket.AF_INET,
+                                       None, None, None,
+                                       ('localhost', 9092))])
+
+@pytest.fixture
 def _socket(mocker):
     socket = mocker.MagicMock()
     socket.connect_ex.return_value = 0
@@ -25,7 +32,7 @@ def _socket(mocker):
 
 
 @pytest.fixture
-def conn(_socket):
+def conn(_socket, dns_lookup):
     conn = BrokerConnection('localhost', 9092, socket.AF_INET)
     return conn