summaryrefslogtreecommitdiff
path: root/test/conftest.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2018-03-09 14:54:06 -0500
committerGitHub <noreply@github.com>2018-03-09 14:54:06 -0500
commit1ffdd5caf7f10fb5372780cb9a5ac4a906cac342 (patch)
tree927d74134f29bab100d5170711b339ca31484132 /test/conftest.py
parent4cbeb2e591447ba25271c4924393e602ba49b324 (diff)
downloadkafka-python-1ffdd5caf7f10fb5372780cb9a5ac4a906cac342.tar.gz
Add BrokerConnection.connect_blocking() (#1411)
Diffstat (limited to 'test/conftest.py')
-rw-r--r--test/conftest.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/test/conftest.py b/test/conftest.py
index d53ff23..52ebfb4 100644
--- a/test/conftest.py
+++ b/test/conftest.py
@@ -128,6 +128,7 @@ def conn(mocker):
return state
conn._set_conn_state = _set_conn_state
conn.connect.side_effect = lambda: conn.state
+ conn.connect_blocking.return_value = True
conn.connecting = lambda: conn.state in (ConnectionStates.CONNECTING,
ConnectionStates.HANDSHAKE)
conn.connected = lambda: conn.state is ConnectionStates.CONNECTED