78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
from mock import AsyncMock
|
|
|
|
from lnbits import bolt11
|
|
from lnbits.wallets.base import PaymentResponse, PaymentStatus, StatusResponse
|
|
from lnbits.wallets.fake import FakeWallet
|
|
|
|
from .helpers import WALLET, get_random_string, is_fake
|
|
|
|
|
|
# generates an invoice with FakeWallet
|
|
async def generate_mock_invoice(**x):
|
|
invoice = await FakeWallet.create_invoice(
|
|
FakeWallet(), amount=10, memo=f"mock invoice {get_random_string()}"
|
|
)
|
|
return invoice
|
|
|
|
|
|
if is_fake:
|
|
WALLET.status = AsyncMock(
|
|
return_value=StatusResponse(
|
|
"", # no error
|
|
1000000, # msats
|
|
)
|
|
)
|
|
|
|
# Note: if this line is uncommented, invoices will always be generated by FakeWallet
|
|
# WALLET.create_invoice = generate_mock_invoice
|
|
|
|
# NOTE: This mock fails since it yields the same invoice multiple
|
|
# times which makes the db throw an error due to uniqueness contraints
|
|
# on the checking ID
|
|
|
|
# # primitive event loop for generate_mock_invoice()
|
|
# def drive(c):
|
|
# while True:
|
|
# try:
|
|
# c.send(None)
|
|
# except StopIteration as e:
|
|
# return e.value
|
|
|
|
# # finally we await it
|
|
# invoice = drive(generate_mock_invoice())
|
|
|
|
# WALLET.create_invoice = AsyncMock(
|
|
# return_value=InvoiceResponse(
|
|
# True, # ok
|
|
# invoice.checking_id, # checking_id (i.e. payment_hash)
|
|
# invoice.payment_request, # payment_request
|
|
# "", # no error
|
|
# )
|
|
# )
|
|
|
|
|
|
if is_fake:
|
|
|
|
def pay_invoice_side_effect(
|
|
payment_request: str, fee_limit_msat: int
|
|
) -> PaymentResponse:
|
|
invoice = bolt11.decode(payment_request)
|
|
return PaymentResponse(
|
|
True, # ok
|
|
invoice.payment_hash, # checking_id (i.e. payment_hash)
|
|
0, # fee_msat
|
|
"", # no error
|
|
)
|
|
|
|
WALLET.pay_invoice = AsyncMock(side_effect=pay_invoice_side_effect)
|
|
WALLET.get_invoice_status = AsyncMock(
|
|
return_value=PaymentStatus(
|
|
True, # paid
|
|
)
|
|
)
|
|
WALLET.get_payment_status = AsyncMock(
|
|
return_value=PaymentStatus(
|
|
True, # paid
|
|
)
|
|
)
|