mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 10:34:20 +01:00
6a0a370ba5
* clean up db * db: table lock * db.table_with_schema * fix encrypt.py * postgres nowait * add timeout to lock * melt quote state in db * kinda working * kinda working with postgres * remove dispose * getting there * porperly clean up db for tests * faster tests * configure connection pooling * try github with connection pool * invoice dispatcher does not lock db * fakewallet: pay_if_regtest waits * pay fakewallet invoices * add more * faster * slower * pay_if_regtest async * do not lock the invoice dispatcher * test: do I get disk I/O errors if we disable the invoice_callback_dispatcher? * fix fake so it workss without a callback dispatchert * test on github * readd tasks * refactor * increase time for lock invoice disatcher * try avoiding a race * remove task * github actions: test regtest with postgres * mint per module * no connection pool for testing * enable pool * do not resend paid event * reuse connection * close db connections * sessions * enable debug * dispose engine * disable connection pool for tests * enable connection pool for postgres only * clean up shutdown routine * remove wait for lightning fakewallet lightning invoice * cancel invoice listener tasks on shutdown * fakewallet conftest: decrease outgoing delay * delay payment and set postgres only if needed * disable fail fast for regtest * clean up regtest.yml * change order of tests_db.py * row-specific mint_quote locking * refactor * fix lock statement * refactor swap * refactor * remove psycopg2 * add connection string example to .env.example * remove unnecessary pay * shorter sleep in test_wallet_subscription_swap
116 lines
3.6 KiB
Python
116 lines
3.6 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
import pytest_asyncio
|
|
|
|
from cashu.core.base import Method, MintQuoteState, ProofSpentState, ProofState
|
|
from cashu.core.json_rpc.base import JSONRPCNotficationParams
|
|
from cashu.core.nuts import WEBSOCKETS_NUT
|
|
from cashu.core.settings import settings
|
|
from cashu.wallet.wallet import Wallet
|
|
from tests.conftest import SERVER_ENDPOINT
|
|
from tests.helpers import (
|
|
is_fake,
|
|
pay_if_regtest,
|
|
)
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="function")
|
|
async def wallet(mint):
|
|
wallet1 = await Wallet.with_db(
|
|
url=SERVER_ENDPOINT,
|
|
db="test_data/wallet_subscriptions",
|
|
name="wallet_subscriptions",
|
|
)
|
|
await wallet1.load_mint()
|
|
yield wallet1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(is_fake, reason="only regtest")
|
|
async def test_wallet_subscription_mint(wallet: Wallet):
|
|
if not wallet.mint_info.supports_nut(WEBSOCKETS_NUT):
|
|
pytest.skip("No websocket support")
|
|
|
|
if not wallet.mint_info.supports_websocket_mint_quote(
|
|
Method["bolt11"], wallet.unit
|
|
):
|
|
pytest.skip("No websocket support for bolt11_mint_quote")
|
|
|
|
triggered = False
|
|
msg_stack: list[JSONRPCNotficationParams] = []
|
|
|
|
def callback(msg: JSONRPCNotficationParams):
|
|
nonlocal triggered, msg_stack
|
|
triggered = True
|
|
msg_stack.append(msg)
|
|
asyncio.run(wallet.mint(int(invoice.amount), id=invoice.id))
|
|
|
|
invoice, sub = await wallet.request_mint_with_callback(128, callback=callback)
|
|
await pay_if_regtest(invoice.bolt11)
|
|
wait = settings.fakewallet_delay_incoming_payment or 2
|
|
await asyncio.sleep(wait + 2)
|
|
|
|
assert triggered
|
|
assert len(msg_stack) == 3
|
|
|
|
assert msg_stack[0].payload["paid"] is False
|
|
assert msg_stack[0].payload["state"] == MintQuoteState.unpaid.value
|
|
|
|
assert msg_stack[1].payload["paid"] is True
|
|
assert msg_stack[1].payload["state"] == MintQuoteState.paid.value
|
|
|
|
assert msg_stack[2].payload["paid"] is True
|
|
assert msg_stack[2].payload["state"] == MintQuoteState.issued.value
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_wallet_subscription_swap(wallet: Wallet):
|
|
if not wallet.mint_info.supports_nut(WEBSOCKETS_NUT):
|
|
pytest.skip("No websocket support")
|
|
|
|
invoice = await wallet.request_mint(64)
|
|
await pay_if_regtest(invoice.bolt11)
|
|
await wallet.mint(64, id=invoice.id)
|
|
|
|
triggered = False
|
|
msg_stack: list[JSONRPCNotficationParams] = []
|
|
|
|
def callback(msg: JSONRPCNotficationParams):
|
|
nonlocal triggered, msg_stack
|
|
triggered = True
|
|
msg_stack.append(msg)
|
|
|
|
n_subscriptions = len(wallet.proofs)
|
|
state, sub = await wallet.check_proof_state_with_callback(
|
|
wallet.proofs, callback=callback
|
|
)
|
|
|
|
_ = await wallet.split_to_send(wallet.proofs, 64)
|
|
|
|
wait = 1
|
|
await asyncio.sleep(wait)
|
|
assert triggered
|
|
|
|
# we receive 3 messages for each subscription:
|
|
# initial state (UNSPENT), pending state (PENDING), spent state (SPENT)
|
|
assert len(msg_stack) == n_subscriptions * 3
|
|
|
|
# the first one is the UNSPENT state
|
|
pending_stack = msg_stack[:n_subscriptions]
|
|
for msg in pending_stack:
|
|
proof_state = ProofState.parse_obj(msg.payload)
|
|
assert proof_state.state == ProofSpentState.unspent
|
|
|
|
# the second one is the PENDING state
|
|
spent_stack = msg_stack[n_subscriptions : n_subscriptions * 2]
|
|
for msg in spent_stack:
|
|
proof_state = ProofState.parse_obj(msg.payload)
|
|
assert proof_state.state == ProofSpentState.pending
|
|
|
|
# the third one is the SPENT state
|
|
spent_stack = msg_stack[n_subscriptions * 2 :]
|
|
for msg in spent_stack:
|
|
proof_state = ProofState.parse_obj(msg.payload)
|
|
assert proof_state.state == ProofSpentState.spent
|