mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 18:44:20 +01:00
d8d3037cc5
* `PaymentResult` * ledger: rely on PaymentResult instead of paid flag. Double check for payments marked pending. * `None` is `PENDING` * make format * reflected changes API tests where `PaymentStatus` is used + reflected changes in lnbits * reflect changes in blink backend and tests * fix lnbits get_payment_status * remove paid flag * fix mypy * remove more paid flags * fix strike mypy * green * shorten all state checks * fix * fix some tests * gimme ✅ * fix............ * fix lnbits * fix error * lightning refactor * add more regtest tests * add tests for pending state and failure * shorten checks * use match case for startup check - and remember modified checking_id from pay_invoice * fix strike pending return * new tests? * refactor startup routine into get_melt_quote * test with purge * refactor blink * cleanup responses * blink: return checking_id on failure * fix lndgrpc try except * add more testing for melt branches * speed things up a bit * remove comments * remove comments * block pending melt quotes * remove comments --------- Co-authored-by: lollerfirst <lollerfirst@gmail.com>
148 lines
4.7 KiB
Python
148 lines
4.7 KiB
Python
from typing import List, Union
|
|
|
|
import bolt11
|
|
import pytest
|
|
import pytest_asyncio
|
|
|
|
from cashu.core.base import Proof
|
|
from cashu.core.errors import CashuError
|
|
from cashu.wallet.lightning import LightningWallet
|
|
from tests.conftest import SERVER_ENDPOINT
|
|
from tests.helpers import (
|
|
get_real_invoice,
|
|
is_deprecated_api_only,
|
|
is_fake,
|
|
is_regtest,
|
|
pay_if_regtest,
|
|
)
|
|
|
|
|
|
async def assert_err(f, msg: Union[str, CashuError]):
|
|
"""Compute f() and expect an error message 'msg'."""
|
|
try:
|
|
await f
|
|
except Exception as exc:
|
|
error_message: str = str(exc.args[0])
|
|
if isinstance(msg, CashuError):
|
|
if msg.detail not in error_message:
|
|
raise Exception(
|
|
f"CashuError. Expected error: {msg.detail}, got: {error_message}"
|
|
)
|
|
return
|
|
if msg not in error_message:
|
|
raise Exception(f"Expected error: {msg}, got: {error_message}")
|
|
return
|
|
raise Exception(f"Expected error: {msg}, got no error")
|
|
|
|
|
|
def assert_amt(proofs: List[Proof], expected: int):
|
|
"""Assert amounts the proofs contain."""
|
|
assert [p.amount for p in proofs] == expected
|
|
|
|
|
|
async def reset_wallet_db(wallet: LightningWallet):
|
|
await wallet.db.execute("DELETE FROM proofs")
|
|
await wallet.db.execute("DELETE FROM proofs_used")
|
|
await wallet.db.execute("DELETE FROM keysets")
|
|
await wallet.load_mint()
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="function")
|
|
async def wallet():
|
|
wallet = await LightningWallet.with_db(
|
|
url=SERVER_ENDPOINT,
|
|
db="test_data/wallet1",
|
|
name="wallet1",
|
|
)
|
|
await wallet.async_init()
|
|
yield wallet
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_invoice(wallet: LightningWallet):
|
|
invoice = await wallet.create_invoice(64)
|
|
assert invoice.payment_request
|
|
assert invoice.payment_request.startswith("ln")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(is_deprecated_api_only, reason="only works with v1 API")
|
|
async def test_create_invoice_with_description(wallet: LightningWallet):
|
|
invoice = await wallet.create_invoice(64, "test description")
|
|
assert invoice.payment_request
|
|
assert invoice.payment_request.startswith("ln")
|
|
invoiceObj = bolt11.decode(invoice.payment_request)
|
|
assert invoiceObj.description == "test description"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(is_regtest, reason="only works with FakeWallet")
|
|
async def test_check_invoice_internal(wallet: LightningWallet):
|
|
# fill wallet
|
|
invoice = await wallet.create_invoice(64)
|
|
assert invoice.payment_request
|
|
assert invoice.checking_id
|
|
status = await wallet.get_invoice_status(invoice.checking_id)
|
|
assert status.settled
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(is_fake, reason="only works with Regtest")
|
|
async def test_check_invoice_external(wallet: LightningWallet):
|
|
# fill wallet
|
|
invoice = await wallet.create_invoice(64)
|
|
assert invoice.payment_request
|
|
assert invoice.checking_id
|
|
status = await wallet.get_invoice_status(invoice.checking_id)
|
|
assert not status.settled
|
|
await pay_if_regtest(invoice.payment_request)
|
|
status = await wallet.get_invoice_status(invoice.checking_id)
|
|
assert status.settled
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(is_regtest, reason="only works with FakeWallet")
|
|
async def test_pay_invoice_internal(wallet: LightningWallet):
|
|
# fill wallet
|
|
invoice = await wallet.create_invoice(64)
|
|
assert invoice.payment_request
|
|
assert invoice.checking_id
|
|
await wallet.get_invoice_status(invoice.checking_id)
|
|
assert wallet.available_balance >= 64
|
|
|
|
# pay invoice
|
|
invoice2 = await wallet.create_invoice(16)
|
|
assert invoice2.payment_request
|
|
status = await wallet.pay_invoice(invoice2.payment_request)
|
|
|
|
assert status.settled
|
|
|
|
# check payment
|
|
assert invoice2.checking_id
|
|
status = await wallet.get_payment_status(invoice2.checking_id)
|
|
assert status.settled
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(is_fake, reason="only works with Regtest")
|
|
async def test_pay_invoice_external(wallet: LightningWallet):
|
|
# fill wallet
|
|
invoice = await wallet.create_invoice(64)
|
|
assert invoice.payment_request
|
|
assert invoice.checking_id
|
|
await pay_if_regtest(invoice.payment_request)
|
|
status = await wallet.get_invoice_status(invoice.checking_id)
|
|
assert status.settled
|
|
assert wallet.available_balance >= 64
|
|
|
|
# pay invoice
|
|
invoice_real = get_real_invoice(16)
|
|
status = await wallet.pay_invoice(invoice_real["payment_request"])
|
|
|
|
assert status.settled
|
|
|
|
# check payment
|
|
assert status.checking_id
|
|
status = await wallet.get_payment_status(status.checking_id)
|
|
assert status.settled
|