summaryrefslogtreecommitdiff
path: root/test/test_client_async.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_client_async.py')
-rw-r--r--test/test_client_async.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_client_async.py b/test/test_client_async.py
index 9191c5e..b6bf0f6 100644
--- a/test/test_client_async.py
+++ b/test/test_client_async.py
@@ -252,7 +252,7 @@ def test_poll(mocker):
# metadata timeout wins
metadata.return_value = 1000
- tasks.return_value = time.time() + 2 # 2 seconds from now
+ tasks.return_value = 2
cli.poll()
_poll.assert_called_with(1.0)
@@ -261,13 +261,13 @@ def test_poll(mocker):
_poll.assert_called_with(0.25)
# tasks timeout wins
- tasks.return_value = time.time() # next task is now
+ tasks.return_value = 0
cli.poll(250)
_poll.assert_called_with(0)
# default is request_timeout_ms
metadata.return_value = 1000000
- tasks.return_value = time.time() + 10000
+ tasks.return_value = 10000
cli.poll()
_poll.assert_called_with(cli.config['request_timeout_ms'] / 1000.0)