mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-24 22:58:46 +01:00
dbf687c47a
* feat: add useful query helpers for insert and update i saw this bits of code all over the codebase, this `helpers.py` tries to abstract that away and make it more readable.
27 lines
581 B
Python
27 lines
581 B
Python
import pytest
|
|
from pydantic import BaseModel
|
|
|
|
from lnbits.helpers import (
|
|
insert_query,
|
|
update_query,
|
|
)
|
|
|
|
|
|
class TestModel(BaseModel):
|
|
id: int
|
|
name: str
|
|
|
|
|
|
test = TestModel(id=1, name="test")
|
|
|
|
|
|
@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) 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 = ? WHERE id = ?"
|