mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-01-21 07:55:00 +00:00
daabeccb60
See db.py for rationale. Tests: - Run functional tests and unit tests. Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org> Reviewed-by: James Valleroy <jvalleroy@mailbox.org>
69 lines
1.4 KiB
Python
69 lines
1.4 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
"""
|
|
Tests for database utilities.
|
|
"""
|
|
import threading
|
|
import time
|
|
|
|
from .. import db
|
|
|
|
|
|
def test_db_lock_no_wait():
|
|
"""Test that lock is immediately by first user."""
|
|
lock = db.DBLock()
|
|
|
|
start_time = time.time()
|
|
with lock:
|
|
pass
|
|
|
|
end_time = time.time()
|
|
assert end_time - start_time < 0.1
|
|
|
|
|
|
def test_db_lock_max_wait():
|
|
"""Test that lock waits only for timeout period."""
|
|
event = threading.Event()
|
|
lock = db.DBLock()
|
|
lock.timeout = 0.25
|
|
|
|
def thread_func():
|
|
with lock:
|
|
event.set()
|
|
time.sleep(0.3)
|
|
|
|
thread = threading.Thread(target=thread_func)
|
|
thread.start()
|
|
|
|
event.wait()
|
|
start_time = time.time()
|
|
with lock as return_value:
|
|
pass
|
|
|
|
end_time = time.time()
|
|
assert end_time - start_time < 0.27
|
|
assert not return_value
|
|
|
|
|
|
def test_db_lock_release():
|
|
"""Test that lock is available after release."""
|
|
event = threading.Event()
|
|
lock = db.DBLock()
|
|
lock.timeout = 0.25
|
|
|
|
def thread_func():
|
|
with lock:
|
|
event.set()
|
|
time.sleep(0.2)
|
|
|
|
thread = threading.Thread(target=thread_func)
|
|
thread.start()
|
|
|
|
event.wait()
|
|
start_time = time.time()
|
|
with lock as return_value:
|
|
pass
|
|
|
|
end_time = time.time()
|
|
assert return_value
|
|
assert end_time - start_time <= 0.23
|