Use correct timeout consts in tests

Author: Pijukatel

tests/unit/test_client_timeouts.py

@@ -106,19 +106,19 @@ def assert_timeout(expected_timeout: int, request: Request) -> Response:
     (KeyValueStoreClient, 'set_record', DEFAULT_TIMEOUT, {'key': 'some_key', 'value': 'some_value'}),
     (KeyValueStoreClient, 'delete_record', kvs._SMALL_TIMEOUT, {'key': 'some_key'}),
     (RequestQueueClient, 'get', request_queue._SMALL_TIMEOUT, {}),
-    (RequestQueueClient, 'update', kvs._SMALL_TIMEOUT, {}),
-    (RequestQueueClient, 'delete', kvs._SMALL_TIMEOUT, {}),
-    (RequestQueueClient, 'list_head', kvs._SMALL_TIMEOUT, {}),
-    (RequestQueueClient, 'list_and_lock_head', kvs._MEDIUM_TIMEOUT, {'lock_secs': 1}),
-    (RequestQueueClient, 'add_request', kvs._SMALL_TIMEOUT, {'request': {}}),
-    (RequestQueueClient, 'get_request', kvs._SMALL_TIMEOUT, {'request_id': 'some_id'}),
-    (RequestQueueClient, 'update_request', kvs._MEDIUM_TIMEOUT, {'request': {'id': 123}}),
-    (RequestQueueClient, 'delete_request', kvs._SMALL_TIMEOUT, {'request_id': 123}),
-    (RequestQueueClient, 'prolong_request_lock', kvs._MEDIUM_TIMEOUT, {'request_id': 123, 'lock_secs': 1}),
-    (RequestQueueClient, 'delete_request_lock', kvs._SMALL_TIMEOUT, {'request_id': 123}),
-    (RequestQueueClient, 'batch_add_requests', kvs._MEDIUM_TIMEOUT, {'requests': [{}]}),
-    (RequestQueueClient, 'batch_delete_requests', kvs._SMALL_TIMEOUT, {'requests': [{}]}),
-    (RequestQueueClient, 'list_requests', kvs._MEDIUM_TIMEOUT, {}),
+    (RequestQueueClient, 'update', request_queue._SMALL_TIMEOUT, {}),
+    (RequestQueueClient, 'delete', request_queue._SMALL_TIMEOUT, {}),
+    (RequestQueueClient, 'list_head', request_queue._SMALL_TIMEOUT, {}),
+    (RequestQueueClient, 'list_and_lock_head', request_queue._MEDIUM_TIMEOUT, {'lock_secs': 1}),
+    (RequestQueueClient, 'add_request', request_queue._SMALL_TIMEOUT, {'request': {}}),
+    (RequestQueueClient, 'get_request', request_queue._SMALL_TIMEOUT, {'request_id': 'some_id'}),
+    (RequestQueueClient, 'update_request', request_queue._MEDIUM_TIMEOUT, {'request': {'id': 123}}),
+    (RequestQueueClient, 'delete_request', request_queue._SMALL_TIMEOUT, {'request_id': 123}),
+    (RequestQueueClient, 'prolong_request_lock', request_queue._MEDIUM_TIMEOUT, {'request_id': 123, 'lock_secs': 1}),
+    (RequestQueueClient, 'delete_request_lock', request_queue._SMALL_TIMEOUT, {'request_id': 123}),
+    (RequestQueueClient, 'batch_add_requests', request_queue._MEDIUM_TIMEOUT, {'requests': [{}]}),
+    (RequestQueueClient, 'batch_delete_requests', request_queue._SMALL_TIMEOUT, {'requests': [{}]}),
+    (RequestQueueClient, 'list_requests', request_queue._MEDIUM_TIMEOUT, {}),
 ]