lnbits-legend/tests/unit/test_services_wallet_limit.py

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
        )