innov
/
dragonfly
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

test: flush all records between pytest tests #199 (#205) 

test: flush all records between pytest tests
This commit is contained in:
Shmulik Klein 2022-07-13 11:45:29 +02:00 committed by GitHub
parent d29425fb2e
commit d2b7987ac3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/pytest/test_dragonfly.py
View File

@ -8,6 +8,7 @@ from threading import Thread
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def dragonfly_db(): def dragonfly_db():
""" Starts a single DragonflyDB process, runs only once. """
dragonfly_path = os.environ.get("DRAGONFLY_HOME", '../../build-dbg/dragonfly') dragonfly_path = os.environ.get("DRAGONFLY_HOME", '../../build-dbg/dragonfly')
print("Starting DragonflyDB [{}]".format(dragonfly_path)) print("Starting DragonflyDB [{}]".format(dragonfly_path))
# TODO: parse arguments and pass them over # TODO: parse arguments and pass them over
@ -19,7 +20,7 @@ def dragonfly_db():
yield yield
print("Terminating DragonflyDB process [{}]".format(p.id)) print("Terminating DragonflyDB process [{}]".format(p.pid))
try: try:
p.terminate() p.terminate()
outs, errs = p.communicate(timeout=15) outs, errs = p.communicate(timeout=15)
@ -31,12 +32,19 @@ def dragonfly_db():
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def client(dragonfly_db): def connection(dragonfly_db):
pool = redis.ConnectionPool(decode_responses=True) pool = redis.ConnectionPool(decode_responses=True)
client = redis.Redis(connection_pool=pool) client = redis.Redis(connection_pool=pool)
return client return client
@pytest.fixture
def client(connection):
""" Flushes all the records, runs before each test. """
connection.flushall()
return connection
class BLPopWorkerThread: class BLPopWorkerThread:
def __init__(self): def __init__(self):
self.result = None self.result = None