cx-library-manager/server_tests/014-attachments_manager.py

import sys
import pathlib

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

sys.path.append(str(project))

import server_source as source
from test import test


import asyncio
import tortoise

async def main():
    global test_dir

    modules = {
        source.model.Meta.app: [ "server_source" ]
    }

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

    await tortoise.Tortoise.generate_schemas()
    
    directory = source.attachments_directory(test_dir)
    manager = source.attachments_manager(directory)
    
    first_id = await manager.upload(
        "IyBVd1UKICogRmlyc3QgcG9pbnQKICogU2Vjb25kIHBvaW50Cg==",
        "txt",
        "Sample",
        "That is example attachment."
    )

    first = await manager.get_by_id(first_id)

    print("Result after creation:")
    print(repr(first))
    print()

    await manager.edit(first_id, "new_name", "That is new description.")

    first = await manager.get_by_id(first_id)

    print("Result after edit:")
    print(repr(first))
    print()

    second_id = await manager.upload(
        "IyBVd1UKICogRmlyc3QgcG9pbnQKICogU2Vjb25kIHBvaW50Cg==",
        "md",
        "second_item",
        "That is second item."
    )

    alls = await manager.get_all()
    
    print("Created second. All items:")
    print(repr(alls))
    print()

    await tortoise.Tortoise.close_connections()

asyncio.run(main())