lnbits-legend/tests/unit/test_helpers_query.py

import pytest

from lnbits.helpers import (
    insert_query,
    update_query,
)
from tests.helpers import DbTestModel

test = DbTestModel(id=1, name="test", value="yes")


@pytest.mark.asyncio
async def test_helpers_insert_query():
    q = insert_query("test_helpers_query", test)
    assert q == "INSERT INTO test_helpers_query (id, name, value) VALUES (?, ?, ?)"


@pytest.mark.asyncio
async def test_helpers_update_query():
    q = update_query("test_helpers_query", test)
    assert q == "UPDATE test_helpers_query SET id = ?, name = ?, value = ? WHERE id = ?"