cx-keys/tests/002-users.py

import pathlib

current = pathlib.Path(__file__).parent
root = current.parent

import sys
sys.path.append(str(root))

import assets
import sqlmodel

builder = assets.user_builder()
builder.nick = "test1"
builder.password = "qwerty"

test1 = builder.result

def drop_database() -> None:
    db = pathlib.Path("./002-test.db")

    if db.is_file():
        db.unlink()

drop_database()

connection = sqlmodel.create_engine("sqlite:///002-test.db")
sqlmodel.SQLModel.metadata.create_all(connection)

with assets.user_loader(connection) as loader:
    print("Register: " + str(loader.register(test1)))

    logged = loader.login("test1", "qwerty")
    print("Logged user apikey: " + str(logged.apikey))

    bad_psk = loader.login("test1", "qweee")
    print("Bad password user: " + str(bad_psk))

    logged.apikey = assets.apikey()
    loader.save(logged)
    print("Logged user apikey after change: " + str(logged.apikey))
    
drop_database()