diff options
| author | Dana Powers <dana.powers@rd.io> | 2016-01-23 16:17:42 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2016-01-23 16:17:42 -0800 |
| commit | 962d8c08053e72ae5118ed57ecb6f5c94ac97708 (patch) | |
| tree | a9c2f4f859d95c9c50adc0198b0c10b8fdcb3337 /test | |
| parent | 318b10744cdec9bab268ed8e532fd76246cbb0df (diff) | |
| download | kafka-python-962d8c08053e72ae5118ed57ecb6f5c94ac97708.tar.gz | |
Fix client poll tests (called with sleep=False)
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_client_async.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/test/test_client_async.py b/test/test_client_async.py index b6bf0f6..2e0d9b4 100644 --- a/test/test_client_async.py +++ b/test/test_client_async.py @@ -254,22 +254,23 @@ def test_poll(mocker): metadata.return_value = 1000 tasks.return_value = 2 cli.poll() - _poll.assert_called_with(1.0) + _poll.assert_called_with(1.0, sleep=False) # user timeout wins cli.poll(250) - _poll.assert_called_with(0.25) + _poll.assert_called_with(0.25, sleep=False) # tasks timeout wins tasks.return_value = 0 cli.poll(250) - _poll.assert_called_with(0) + _poll.assert_called_with(0, sleep=False) # default is request_timeout_ms metadata.return_value = 1000000 tasks.return_value = 10000 cli.poll() - _poll.assert_called_with(cli.config['request_timeout_ms'] / 1000.0) + _poll.assert_called_with(cli.config['request_timeout_ms'] / 1000.0, + sleep=False) def test__poll(): |