source/projects/tentacles/test/test_store_users.py

40 lines
1.3 KiB
Python
Raw Normal View History

2023-05-13 22:58:17 +00:00
#!/usr/bin/env python3
from tentacles.db import Db
2023-05-13 22:58:17 +00:00
def test_mkuser(db: Db, username_testy, password_testy):
res = db.try_create_user(
2023-06-03 19:20:05 +00:00
username=username_testy, email=username_testy, password=password_testy
)
assert res
assert [(it.id, it.name) for it in db.list_users()] == [(res.id, username_testy)]
2023-05-13 22:58:17 +00:00
def test_mksession(db: Db, uid_testy, username_testy, password_testy, login_ttl):
assert uid_testy
res = db.try_login(username=username_testy, password=password_testy, ttl=login_ttl)
2023-06-03 19:20:05 +00:00
assert res is not None
assert [it.id for it in db.list_keys(uid=uid_testy)] == [res.id]
assert db.try_key(kid=res.id).user_id == uid_testy
2023-05-13 22:58:17 +00:00
def test_refresh_key(db: Db, sid_testy, login_ttl):
before = db.fetch_key(kid=sid_testy)
db.refresh_key(kid=sid_testy, ttl=login_ttl * 2)
after = db.fetch_key(kid=sid_testy)
2023-05-13 22:58:17 +00:00
assert before != after
def tets_mkkey(db: Db, sid_testy, uid_testy):
assert db.try_key(kid=sid_testy) == uid_testy
new_key = db.create_key(kid=sid_testy, ttl=None)
2023-05-13 22:58:17 +00:00
assert new_key is not None
assert db.try_key(kid=new_key) == uid_testy
2023-05-13 22:58:17 +00:00
def test_logout(db: Db, uid_testy, sid_testy):
assert db.try_key(kid=sid_testy)
db.delete_key(uid=uid_testy, kid=sid_testy)
assert not db.try_key(kid=sid_testy)