99 lines
2.4 KiB
Python
99 lines
2.4 KiB
Python
import shutil
|
|
from dataclasses import dataclass
|
|
|
|
import pytest
|
|
|
|
from myfasthtml.core.dbmanager import DbManager, DbObject
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def session():
|
|
return {
|
|
"user_info": {
|
|
"id": "test_tenant_id",
|
|
"email": "test@email.com",
|
|
"username": "test user",
|
|
"role": [],
|
|
}
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def db_manager(session):
|
|
shutil.rmtree("TestDb", ignore_errors=True)
|
|
db_manager_instance = DbManager(session, root="TestDb", auto_register=False)
|
|
|
|
yield db_manager_instance
|
|
|
|
shutil.rmtree("TestDb", ignore_errors=True)
|
|
|
|
|
|
def simplify(res: dict) -> dict:
|
|
return {k: v for k, v in res.items() if not k.startswith("_")}
|
|
|
|
|
|
def test_i_can_init(session, db_manager):
|
|
@dataclass
|
|
class DummyObject(DbObject):
|
|
def __init__(self, sess: dict):
|
|
super().__init__(sess, "DummyObject", db_manager)
|
|
|
|
value: str = "hello"
|
|
number: int = 42
|
|
|
|
DummyObject(session)
|
|
|
|
assert simplify(db_manager.load("DummyObject")) == {"value": "hello", "number": 42}
|
|
|
|
|
|
def test_i_can_init_from_db(session, db_manager):
|
|
@dataclass
|
|
class DummyObject(DbObject):
|
|
def __init__(self, sess: dict):
|
|
super().__init__(sess, "DummyObject", db_manager)
|
|
|
|
value: str = "hello"
|
|
number: int = 42
|
|
|
|
# insert other values in db
|
|
db_manager.save("DummyObject", {"value": "other_value", "number": 34})
|
|
|
|
dummy = DummyObject(session)
|
|
|
|
assert dummy.value == "other_value"
|
|
assert dummy.number == 34
|
|
|
|
|
|
def test_db_is_updated_when_attribute_is_modified(session, db_manager):
|
|
@dataclass
|
|
class DummyObject(DbObject):
|
|
def __init__(self, sess: dict):
|
|
super().__init__(sess, "DummyObject", db_manager)
|
|
|
|
value: str = "hello"
|
|
number: int = 42
|
|
|
|
dummy = DummyObject(session)
|
|
dummy.value = "other_value"
|
|
|
|
assert simplify(db_manager.load("DummyObject")) == {"value": "other_value", "number": 42}
|
|
|
|
|
|
def test_i_do_not_save_in_db_when_value_is_the_same(session, db_manager):
|
|
@dataclass
|
|
class DummyObject(DbObject):
|
|
def __init__(self, sess: dict):
|
|
super().__init__(sess, "DummyObject", db_manager)
|
|
|
|
value: str = "hello"
|
|
number: int = 42
|
|
|
|
dummy = DummyObject(session)
|
|
dummy.value = "other_value"
|
|
in_db_1 = db_manager.load("DummyObject")
|
|
|
|
dummy.value = "other_value"
|
|
in_db_2 = db_manager.load("DummyObject")
|
|
|
|
assert in_db_1["__parent__"] == in_db_2["__parent__"]
|