cx-library-manager/server_tests/006-user_manager.py

import sys
import pathlib

test_file = pathlib.Path(__file__)
project = test_file.parent.parent

sys.path.append(str(project))

import server_source as source
from test import test

import asyncio
import tortoise

async def main():
    modules = {
        source.model.Meta.app: [ "server_source" ]
    }

    await tortoise.Tortoise.init(
        db_url = "sqlite://:memory:", 
        modules = modules
    )

    await tortoise.Tortoise.generate_schemas()

    created = await source.users_manager.register("test", "sample")
    test(created.nick, "test")

    by_apikey = await source.users_manager.get_by_apikey(created.apikey)
    test(by_apikey.nick, "test")

    loged = await source.users_manager.login("test", "sample")
    test(loged.nick, "test")

    not_loged = await source.users_manager.login("test", "not_working")
    test(not_loged, None)
    await tortoise.Tortoise.close_connections()

asyncio.run(main())