mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-26 15:42:30 +01:00
85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
import json
|
|
|
|
import pytest
|
|
|
|
from lnbits.db import (
|
|
dict_to_model,
|
|
insert_query,
|
|
model_to_dict,
|
|
update_query,
|
|
)
|
|
from tests.helpers import DbTestModel, DbTestModel2, DbTestModel3
|
|
|
|
test_data = DbTestModel3(
|
|
id=1,
|
|
user="userid",
|
|
child=DbTestModel2(
|
|
id=2,
|
|
label="test",
|
|
description="mydesc",
|
|
child=DbTestModel(id=3, name="myname", value="myvalue"),
|
|
child_list=[DbTestModel(id=6, name="myname", value="myvalue")],
|
|
),
|
|
children=[DbTestModel(id=4, name="myname", value="myvalue")],
|
|
children_ids=[4, 1, 3],
|
|
active=True,
|
|
)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_helpers_insert_query():
|
|
q = insert_query("test_helpers_query", test_data)
|
|
assert q == (
|
|
"INSERT INTO test_helpers_query "
|
|
"""("id", "user", "child", "active", "children", "children_ids") """
|
|
"VALUES (:id, :user, :child, :active, :children, :children_ids)"
|
|
)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_helpers_update_query():
|
|
q = update_query("test_helpers_query", test_data)
|
|
assert q == (
|
|
"""UPDATE test_helpers_query SET "id" = :id, "user" = """
|
|
""":user, "child" = :child, "active" = :active, "children" = """
|
|
""":children, "children_ids" = :children_ids WHERE id = :id"""
|
|
)
|
|
|
|
|
|
child_json = json.dumps(
|
|
{
|
|
"id": 2,
|
|
"label": "test",
|
|
"description": "mydesc",
|
|
"child": {"id": 3, "name": "myname", "value": "myvalue"},
|
|
"child_list": [{"id": 6, "name": "myname", "value": "myvalue"}],
|
|
}
|
|
)
|
|
test_dict = {
|
|
"id": 1,
|
|
"user": "userid",
|
|
"child": child_json,
|
|
"active": True,
|
|
"children": '[{"id": 4, "name": "myname", "value": "myvalue"}]',
|
|
"children_ids": "[4, 1, 3]",
|
|
}
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_helpers_model_to_dict():
|
|
d = model_to_dict(test_data)
|
|
assert d.get("id") == test_data.id
|
|
assert d.get("active") == test_data.active
|
|
assert d.get("child") == child_json
|
|
assert d.get("user") == test_data.user
|
|
assert d == test_dict
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_helpers_dict_to_model():
|
|
m = dict_to_model(test_dict, DbTestModel3)
|
|
assert m == test_data
|
|
assert type(m) is DbTestModel3
|
|
assert m.active is True
|
|
assert type(m.child) is DbTestModel2
|
|
assert type(m.child.child) is DbTestModel
|