From 02bc28017b9ad665d58baab424d279bc1b3dd1c1 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Wed, 13 Mar 2019 18:35:26 -0700 Subject: Mock dns lookups in test_conn --- test/test_conn.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'test/test_conn.py') 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 @@ -16,6 +16,13 @@ from kafka.protocol.produce import ProduceRequest 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() @@ -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 -- cgit v1.2.1