highperfocused/lnbits
1
0
Fork 0
mirror of https://github.com/lnbits/lnbits.git synced 2025-05-04 00:30:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

[TEST] proper credit_wallet function from services (#1862)

Browse Source
This commit is contained in:
dni ⚡ 2023-08-07 21:49:37 +02:00 committed by GitHub
parent 03acf5e5ec
commit f096b51f70
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/conftest.py
View File

@ -8,15 +8,11 @@ from httpx import AsyncClient
from lnbits.app import create_app from lnbits.app import create_app
from lnbits.commands import migrate_databases from lnbits.commands import migrate_databases
from lnbits.core.crud import create_account, create_wallet from lnbits.core.crud import create_account, create_wallet
from lnbits.core.services import update_wallet_balance
from lnbits.core.views.api import CreateInvoiceData, api_payments_create_invoice from lnbits.core.views.api import CreateInvoiceData, api_payments_create_invoice
from lnbits.db import Database from lnbits.db import Database
from lnbits.settings import settings from lnbits.settings import settings
from tests.helpers import ( from tests.helpers import get_hold_invoice, get_random_invoice_data, get_real_invoice
credit_wallet,
get_hold_invoice,
get_random_invoice_data,
get_real_invoice,
)
@pytest_asyncio.fixture(scope="session") @pytest_asyncio.fixture(scope="session")
@ -68,7 +64,7 @@ async def from_user():
async def from_wallet(from_user): async def from_wallet(from_user):
user = from_user user = from_user
wallet = await create_wallet(user_id=user.id, wallet_name="test_wallet_from") wallet = await create_wallet(user_id=user.id, wallet_name="test_wallet_from")
await credit_wallet( await update_wallet_balance(
wallet_id=wallet.id, wallet_id=wallet.id,
amount=999999999, amount=999999999,
) )
@ -91,7 +87,7 @@ async def to_user():
async def to_wallet(to_user): async def to_wallet(to_user):
user = to_user user = to_user
wallet = await create_wallet(user_id=user.id, wallet_name="test_wallet_to") wallet = await create_wallet(user_id=user.id, wallet_name="test_wallet_to")
await credit_wallet( await update_wallet_balance(
wallet_id=wallet.id, wallet_id=wallet.id,
amount=999999999, amount=999999999,
) )

19
tests/helpers.py
View File

@ -2,32 +2,13 @@ import hashlib
import json import json
import os import os
import random import random
import secrets
import string import string
from subprocess import PIPE, Popen, run from subprocess import PIPE, Popen, run
from typing import Tuple from typing import Tuple
from lnbits.core.crud import create_payment
from lnbits.wallets import get_wallet_class, set_wallet_class from lnbits.wallets import get_wallet_class, set_wallet_class
async def credit_wallet(wallet_id: str, amount: int):
preimage = secrets.token_hex(32)
m = hashlib.sha256()
m.update(f"{preimage}".encode())
payment_hash = m.hexdigest()
await create_payment(
wallet_id=wallet_id,
payment_request="",
payment_hash=payment_hash,
checking_id=payment_hash,
preimage=preimage,
memo=f"funding_test_{get_random_string(5)}",
amount=amount, # msat
pending=False, # not pending, so it will increase the wallet's balance
)
def get_random_string(N: int = 10): def get_random_string(N: int = 10):
return "".join( return "".join(
random.SystemRandom().choice(string.ascii_uppercase + string.digits) random.SystemRandom().choice(string.ascii_uppercase + string.digits)