lnbits-legend/tests/unit/test_db.py
dni ⚡ e607ab7a3e
test: restructure tests (#2444)
unit, api, wallets
* only run test-api for migration
2024-04-19 13:22:06 +02:00

38 lines
954 B
Python

from datetime import date
import pytest
from lnbits.core.crud import (
create_wallet,
delete_wallet,
get_wallet,
get_wallet_for_key,
)
from lnbits.db import POSTGRES
@pytest.mark.asyncio
async def test_date_conversion(db):
if db.type == POSTGRES:
row = await db.fetchone("SELECT now()::date")
assert row and isinstance(row[0], date)
# make test to create wallet and delete wallet
@pytest.mark.asyncio
async def test_create_wallet_and_delete_wallet(app, to_user):
# create wallet
wallet = await create_wallet(user_id=to_user.id, wallet_name="test_wallet_delete")
assert wallet
# delete wallet
await delete_wallet(user_id=to_user.id, wallet_id=wallet.id)
# check if wallet is deleted
del_wallet = await get_wallet(wallet.id)
assert del_wallet is not None
assert del_wallet.deleted is True
del_wallet = await get_wallet_for_key(wallet.inkey)
assert del_wallet is None