lnbits-legend/tests/unit/test_services_wallet_limit.py
2024-12-11 11:39:28 +02:00

36 lines
1.1 KiB
Python

import pytest
from lnbits.core.services.payments import check_wallet_daily_withdraw_limit
from lnbits.settings import Settings
@pytest.mark.anyio
async def test_no_wallet_limit(settings: Settings):
settings.lnbits_wallet_limit_daily_max_withdraw = 0
result = await check_wallet_daily_withdraw_limit(
conn=None, wallet_id="333333", amount_msat=0
)
assert result is None, "No limit set."
@pytest.mark.anyio
async def test_wallet_limit_but_no_payments(settings: Settings):
settings.lnbits_wallet_limit_daily_max_withdraw = 5
result = await check_wallet_daily_withdraw_limit(
conn=None, wallet_id="333333", amount_msat=0
)
assert result is None, "Limit not reqached."
@pytest.mark.anyio
async def test_no_wallet_spend_allowed(settings: Settings):
settings.lnbits_wallet_limit_daily_max_withdraw = -1
with pytest.raises(
ValueError, match="It is not allowed to spend funds from this server."
):
await check_wallet_daily_withdraw_limit(
conn=None, wallet_id="333333", amount_msat=0
)