mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-01-19 05:33:47 +01:00
bda054415a
* [FEAT] improve update_admin_settings while working on the push notification pr i found it very hard just to update 2 settings inside the db, so i improved upon update_admin_settings. now you just need to provide a dict with key/values you want to update inside db. also debugging the endpoints for update_settings i found despite the type of `EditableSettings` fastapi did in fact pass a dict. * t * use `EditableSettings` as param in update_settings * fix settings model validation we previously overrode the pydantic validation with our own method * make `LnbitsSettings` a `BaseModel` and only add `BaseSettings` later this allows us to instantiate `EditableSettings` without the environment values being loaded in * add test * forbid extra fields in update api * fixup * add test * test datadir * move UpdateSettings * fix compat * fixup webpush --------- Co-authored-by: jacksn <jkranawetter05@gmail.com>
174 lines
4.3 KiB
Python
174 lines
4.3 KiB
Python
import asyncio
|
|
|
|
import uvloop
|
|
|
|
uvloop.install() # noqa
|
|
|
|
import pytest
|
|
import pytest_asyncio
|
|
from fastapi.testclient import TestClient
|
|
from httpx import AsyncClient
|
|
|
|
from lnbits.app import create_app
|
|
from lnbits.core.crud import create_account, create_wallet, get_user
|
|
from lnbits.core.models import CreateInvoice
|
|
from lnbits.core.services import update_wallet_balance
|
|
from lnbits.core.views.api import api_payments_create_invoice
|
|
from lnbits.db import Database
|
|
from lnbits.settings import settings
|
|
from tests.helpers import get_hold_invoice, get_random_invoice_data, get_real_invoice
|
|
|
|
# override settings for tests
|
|
settings.lnbits_admin_extensions = []
|
|
settings.lnbits_data_folder = "./tests/data"
|
|
settings.lnbits_admin_ui = True
|
|
settings.lnbits_extensions_default_install = []
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
def event_loop():
|
|
loop = asyncio.get_event_loop()
|
|
yield loop
|
|
loop.close()
|
|
|
|
|
|
# use session scope to run once before and once after all tests
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def app():
|
|
app = create_app()
|
|
await app.router.startup()
|
|
yield app
|
|
await app.router.shutdown()
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def client(app):
|
|
client = AsyncClient(app=app, base_url=f"http://{settings.host}:{settings.port}")
|
|
yield client
|
|
await client.aclose()
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def test_client(app):
|
|
return TestClient(app)
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def db():
|
|
yield Database("database")
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def from_user():
|
|
user = await create_account()
|
|
yield user
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def from_wallet(from_user):
|
|
user = from_user
|
|
wallet = await create_wallet(user_id=user.id, wallet_name="test_wallet_from")
|
|
await update_wallet_balance(
|
|
wallet_id=wallet.id,
|
|
amount=999999999,
|
|
)
|
|
yield wallet
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def from_wallet_ws(from_wallet, test_client):
|
|
# wait a bit in order to avoid receiving topup notification
|
|
await asyncio.sleep(0.1)
|
|
with test_client.websocket_connect(f"/api/v1/ws/{from_wallet.id}") as ws:
|
|
yield ws
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def to_user():
|
|
user = await create_account()
|
|
yield user
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def superuser():
|
|
user = await get_user(settings.super_user)
|
|
yield user
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def to_wallet(to_user):
|
|
user = to_user
|
|
wallet = await create_wallet(user_id=user.id, wallet_name="test_wallet_to")
|
|
await update_wallet_balance(
|
|
wallet_id=wallet.id,
|
|
amount=999999999,
|
|
)
|
|
yield wallet
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def to_wallet_ws(to_wallet, test_client):
|
|
# wait a bit in order to avoid receiving topup notification
|
|
await asyncio.sleep(0.1)
|
|
with test_client.websocket_connect(f"/api/v1/ws/{to_wallet.id}") as ws:
|
|
yield ws
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def inkey_headers_from(from_wallet):
|
|
wallet = from_wallet
|
|
yield {
|
|
"X-Api-Key": wallet.inkey,
|
|
"Content-type": "application/json",
|
|
}
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def adminkey_headers_from(from_wallet):
|
|
wallet = from_wallet
|
|
yield {
|
|
"X-Api-Key": wallet.adminkey,
|
|
"Content-type": "application/json",
|
|
}
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def inkey_headers_to(to_wallet):
|
|
wallet = to_wallet
|
|
yield {
|
|
"X-Api-Key": wallet.inkey,
|
|
"Content-type": "application/json",
|
|
}
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def adminkey_headers_to(to_wallet):
|
|
wallet = to_wallet
|
|
yield {
|
|
"X-Api-Key": wallet.adminkey,
|
|
"Content-type": "application/json",
|
|
}
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="session")
|
|
async def invoice(to_wallet):
|
|
data = await get_random_invoice_data()
|
|
invoiceData = CreateInvoice(**data)
|
|
invoice = await api_payments_create_invoice(invoiceData, to_wallet)
|
|
yield invoice
|
|
del invoice
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="function")
|
|
async def real_invoice():
|
|
invoice = get_real_invoice(100)
|
|
yield {"bolt11": invoice["payment_request"]}
|
|
del invoice
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="function")
|
|
async def hold_invoice():
|
|
invoice = get_hold_invoice(100)
|
|
yield invoice
|
|
del invoice
|