diff options
author | Dana Powers <dana.powers@gmail.com> | 2016-04-04 18:57:32 -0700 |
---|---|---|
committer | Dana Powers <dana.powers@gmail.com> | 2016-04-07 09:49:05 -0700 |
commit | d61e861b0da1647974e617e1633c429c307789be (patch) | |
tree | dad1382251212e88666baa20c7ec8d279487fa48 /test/conftest.py | |
parent | 931373478e30a9d44b89fad6491136222441e929 (diff) | |
download | kafka-python-d61e861b0da1647974e617e1633c429c307789be.tar.gz |
Consolidate conn fixture definitions
Diffstat (limited to 'test/conftest.py')
-rw-r--r-- | test/conftest.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/test/conftest.py b/test/conftest.py index f3a8947..a389480 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -31,3 +31,22 @@ def kafka_broker(version, zookeeper, request): k.close() request.addfinalizer(fin) return k + + +@pytest.fixture +def conn(mocker): + from kafka.conn import ConnectionStates + from kafka.future import Future + from kafka.protocol.metadata import MetadataResponse + conn = mocker.patch('kafka.client_async.BrokerConnection') + conn.return_value = conn + conn.state = ConnectionStates.CONNECTED + conn.send.return_value = Future().success( + MetadataResponse[0]( + [(0, 'foo', 12), (1, 'bar', 34)], # brokers + [])) # topics + conn.blacked_out.return_value = False + conn.connect.side_effect = lambda: conn.state + conn.connecting = lambda: conn.connect() is ConnectionStates.CONNECTING + conn.connected = lambda: conn.connect() is ConnectionStates.CONNECTED + return conn |