lnbits-legend/tests/unit/test_helpers_query.py

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