mirror of
https://github.com/dragonflydb/dragonfly.git
synced 2025-05-11 18:35:46 +02:00
c96f637f73
1. pytest extensions and fixes - allows running them with the existing local server by providing its port (--existing <port>). 2. Extend "DEBUG WATCHED" command to provide more information about watched state. 3. Improve debug/vlog printings around the code. This noisy PR is a preparation before BRPOP fix that will follow later. Signed-off-by: Roman Gershman <roman@dragonflydb.io>
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
from threading import Thread
|
|
|
|
import pytest
|
|
import redis
|
|
|
|
|
|
class BLPopWorkerThread:
|
|
def __init__(self):
|
|
self.result = None
|
|
self.thread = None
|
|
|
|
def async_blpop(self, client: redis.Redis):
|
|
self.result = None
|
|
|
|
def blpop_task(self, client):
|
|
self.result = client.blpop(
|
|
['list1{t}', 'list2{t}', 'list2{t}', 'list1{t}'], 0.5)
|
|
|
|
self.thread = Thread(target=blpop_task, args=(self, client))
|
|
self.thread.start()
|
|
|
|
def wait(self, timeout):
|
|
self.thread.join(timeout)
|
|
return not self.thread.is_alive()
|
|
|
|
|
|
@pytest.mark.parametrize('index', range(50))
|
|
class TestBlPop:
|
|
def test_blpop_multiple_keys(self, client, index):
|
|
wt_blpop = BLPopWorkerThread()
|
|
wt_blpop.async_blpop(client)
|
|
|
|
client.lpush('list1{t}', 'a')
|
|
assert wt_blpop.wait(2)
|
|
assert wt_blpop.result[1] == 'a'
|
|
watched = client.execute_command('DEBUG WATCHED')
|
|
assert watched == ['awaked', [], 'watched', []]
|
|
|
|
wt_blpop.async_blpop(client)
|
|
client.lpush('list2{t}', 'b')
|
|
assert wt_blpop.wait(2)
|
|
assert wt_blpop.result[1] == 'b'
|