lnbits/tests/extensions/bleskomat/test_lnurl_api.py
calle 0f97f8f18b
Fee reserve for lightning backends (#557)
* preparing fees

* fee_limit_msat

* await resp result

* clightning

* fix tests

* fix test

* add fee to test

* mypy

* invoice_status

* checking id fix

* fee reserve error message

* only for external payments
2022-03-16 07:20:15 +01:00

134 lines
5.2 KiB
Python

import pytest
import secrets
from lnbits.core.crud import get_wallet
from lnbits.settings import HOST, PORT
from lnbits.extensions.bleskomat.crud import get_bleskomat_lnurl
from lnbits.extensions.bleskomat.helpers import (
generate_bleskomat_lnurl_signature,
query_to_signing_payload,
)
from tests.conftest import client
from tests.helpers import credit_wallet
from tests.extensions.bleskomat.conftest import bleskomat, lnurl
from tests.mocks import WALLET
@pytest.mark.asyncio
async def test_bleskomat_lnurl_api_missing_secret(client):
response = await client.get("/bleskomat/u")
assert response.status_code == 200
assert response.json() == {"status": "ERROR", "reason": "Missing secret"}
@pytest.mark.asyncio
async def test_bleskomat_lnurl_api_invalid_secret(client):
response = await client.get("/bleskomat/u?k1=invalid-secret")
assert response.status_code == 200
assert response.json() == {"status": "ERROR", "reason": "Invalid secret"}
@pytest.mark.asyncio
async def test_bleskomat_lnurl_api_unknown_api_key(client):
query = {
"id": "does-not-exist",
"nonce": secrets.token_hex(10),
"tag": "withdrawRequest",
"minWithdrawable": "1",
"maxWithdrawable": "1",
"defaultDescription": "",
"f": "EUR",
}
payload = query_to_signing_payload(query)
signature = "xxx" # not checked, so doesn't matter
response = await client.get(f"/bleskomat/u?{payload}&signature={signature}")
assert response.status_code == 200
assert response.json() == {"status": "ERROR", "reason": "Unknown API key"}
@pytest.mark.asyncio
async def test_bleskomat_lnurl_api_invalid_signature(client, bleskomat):
query = {
"id": bleskomat.api_key_id,
"nonce": secrets.token_hex(10),
"tag": "withdrawRequest",
"minWithdrawable": "1",
"maxWithdrawable": "1",
"defaultDescription": "",
"f": "EUR",
}
payload = query_to_signing_payload(query)
signature = "invalid"
response = await client.get(f"/bleskomat/u?{payload}&signature={signature}")
assert response.status_code == 200
assert response.json() == {"status": "ERROR", "reason": "Invalid API key signature"}
@pytest.mark.asyncio
async def test_bleskomat_lnurl_api_valid_signature(client, bleskomat):
query = {
"id": bleskomat.api_key_id,
"nonce": secrets.token_hex(10),
"tag": "withdrawRequest",
"minWithdrawable": "1",
"maxWithdrawable": "1",
"defaultDescription": "test valid sig",
"f": "EUR", # tests use the dummy exchange rate provider
}
payload = query_to_signing_payload(query)
signature = generate_bleskomat_lnurl_signature(
payload=payload,
api_key_secret=bleskomat.api_key_secret,
api_key_encoding=bleskomat.api_key_encoding,
)
response = await client.get(f"/bleskomat/u?{payload}&signature={signature}")
assert response.status_code == 200
data = response.json()
assert data["tag"] == "withdrawRequest"
assert data["minWithdrawable"] == 1000
assert data["maxWithdrawable"] == 1000
assert data["defaultDescription"] == "test valid sig"
assert data["callback"] == f"http://{HOST}:{PORT}/bleskomat/u"
k1 = data["k1"]
lnurl = await get_bleskomat_lnurl(secret=k1)
assert lnurl
@pytest.mark.asyncio
async def test_bleskomat_lnurl_api_action_insufficient_balance(client, lnurl):
bleskomat = lnurl["bleskomat"]
secret = lnurl["secret"]
pr = "lntb500n1pseq44upp5xqd38rgad72lnlh4gl339njlrsl3ykep82j6gj4g02dkule7k54qdqqcqzpgxqyz5vqsp5h0zgewuxdxcl2rnlumh6g520t4fr05rgudakpxm789xgjekha75s9qyyssq5vhwsy9knhfeqg0wn6hcnppwmum8fs3g3jxkgw45havgfl6evchjsz3s8e8kr6eyacz02szdhs7v5lg0m7wehd5rpf6yg8480cddjlqpae52xu"
WALLET.pay_invoice.reset_mock()
response = await client.get(f"/bleskomat/u?k1={secret}&pr={pr}")
assert response.status_code == 200
assert response.json()["status"] == "ERROR"
assert ("Insufficient balance" in response.json()["reason"]) or (
"fee" in response.json()["reason"]
)
wallet = await get_wallet(bleskomat.wallet)
assert wallet.balance_msat == 0
bleskomat_lnurl = await get_bleskomat_lnurl(secret)
assert bleskomat_lnurl.has_uses_remaining() == True
WALLET.pay_invoice.assert_not_called()
@pytest.mark.asyncio
async def test_bleskomat_lnurl_api_action_success(client, lnurl):
bleskomat = lnurl["bleskomat"]
secret = lnurl["secret"]
pr = "lntb500n1pseq44upp5xqd38rgad72lnlh4gl339njlrsl3ykep82j6gj4g02dkule7k54qdqqcqzpgxqyz5vqsp5h0zgewuxdxcl2rnlumh6g520t4fr05rgudakpxm789xgjekha75s9qyyssq5vhwsy9knhfeqg0wn6hcnppwmum8fs3g3jxkgw45havgfl6evchjsz3s8e8kr6eyacz02szdhs7v5lg0m7wehd5rpf6yg8480cddjlqpae52xu"
await credit_wallet(
wallet_id=bleskomat.wallet,
amount=100000,
)
wallet = await get_wallet(bleskomat.wallet)
assert wallet.balance_msat == 100000
WALLET.pay_invoice.reset_mock()
response = await client.get(f"/bleskomat/u?k1={secret}&pr={pr}")
assert response.json() == {"status": "OK"}
wallet = await get_wallet(bleskomat.wallet)
assert wallet.balance_msat == 50000
bleskomat_lnurl = await get_bleskomat_lnurl(secret)
assert bleskomat_lnurl.has_uses_remaining() == False
WALLET.pay_invoice.assert_called_once_with(pr, 2000)