37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
|
from datetime import timedelta, datetime
|
||
|
|
from .checkin_bot import CheckinDB
|
||
|
|
|
||
|
|
TEST_DB_NAME = "test.db"
|
||
|
|
|
||
|
|
checkin_db = CheckinDB(TEST_DB_NAME)
|
||
|
|
checkin_db.clear_rows()
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_present_count() -> None:
|
||
|
|
checkin_db.update("Alice", True)
|
||
|
|
checkin_db.update("Bob", True)
|
||
|
|
checkin_db.update("Bob", False)
|
||
|
|
checkin_db.update("Bob", True)
|
||
|
|
assert checkin_db.get_checkin_count() == 2
|
||
|
|
|
||
|
|
|
||
|
|
def test_present_names() -> None:
|
||
|
|
assert checkin_db.get_checkin_names() == ["Alice", "Bob"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_updated_timestamp() -> None:
|
||
|
|
assert checkin_db.get_updated_for_name(
|
||
|
|
"Alice"
|
||
|
|
) >= datetime.now() - timedelta(minutes=1)
|
||
|
|
|
||
|
|
|
||
|
|
def test_auto_checkout() -> None:
|
||
|
|
assert "Alice" in checkin_db.get_checkin_names()
|
||
|
|
old_timestamp = (datetime.now() - timedelta(hours=6)).timestamp()
|
||
|
|
with checkin_db.con:
|
||
|
|
checkin_db.con.execute(
|
||
|
|
f"UPDATE checkin SET updated = {old_timestamp} WHERE name = 'Alice'"
|
||
|
|
)
|
||
|
|
checkin_db.auto_checkout()
|
||
|
|
assert "Alice" not in checkin_db.get_checkin_names()
|