mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 10:34:20 +01:00
e846acf946
* add websockets for quote updates * add test (not working) * wip: emit events to everyone * wip: emit events to everyone * wip, lots of things broken but invoice callback works * wip * add wip files * tests almost passing * add task * refactor nut constants * startup fix * works with old mints * wip cli * fix mypy * remove automatic invoice test now with websockets * remove comment * better logging * send back response * add rate limiter to websocket * add rate limiter to subscriptions * refactor websocket ratelimit * websocket tests * subscription kinds * doesnt start * remove circular import * update * fix mypy * move test file in test because it fails if it runs later... dunno why * adjust websocket NUT-06 settings * local import and small fix * disable websockets in CLI if "no_check" is selected * move subscription test to where it was * check proof state with callback, add tests * tests: run mint fixture per module instead of per session * subscription command name fix * test per session again * update test race conditions * fix tests * clean up * tmp * fix db issues and remove cached secrets * fix tests * blindly try pipeline * remove comments * comments
119 lines
3.7 KiB
Python
119 lines
3.7 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
import pytest_asyncio
|
|
|
|
from cashu.core.base import Method, 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)
|
|
pay_if_regtest(invoice.bolt11)
|
|
wait = settings.fakewallet_delay_incoming_payment or 2
|
|
await asyncio.sleep(wait + 2)
|
|
|
|
# TODO: check for pending and paid states according to: https://github.com/cashubtc/nuts/pull/136
|
|
# TODO: we have three messages here, but the value "paid" only changes once
|
|
# the mint sends an update when the quote is pending but the API does not express that yet
|
|
|
|
# first we expect the issued=False state to arrive
|
|
|
|
assert triggered
|
|
assert len(msg_stack) == 3
|
|
|
|
assert msg_stack[0].payload["paid"] is False
|
|
|
|
assert msg_stack[1].payload["paid"] is True
|
|
|
|
assert msg_stack[2].payload["paid"] is True
|
|
|
|
|
|
@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)
|
|
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.value == "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.value == "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.value == "SPENT"
|