mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 02:24:20 +01:00
0b2468914d
* first working version but some sats go missing * back at it * make format * restore to main * move mint database * fix some tests * make format * remove old _construct_outputs we reintroduced in merge with main * add type annotations * add wallet private key to tests * wallet: load proofs * fix tests * _generate_secrets with deterministic generation (temporary) * allow wallet initialization with custom private key * add pk to wallet api test * mint scope=module * remove private_key from test_wallet.py to see if it helps with the github tests * readd private keys to tests * workflow without env * add more private key! * readd env * ledger scope session * add default private key for testing * generate private keys if not available * testing * its working!!! * first iteration of bip32 working * get mint info and add many type annotations * tests * fix tests with bip32 * restore from multiple mints * disable profiler * make format * failed POST /mint do not increment secret counter * store derivation path in each token * fix tests * refactor migrations so private keys can be generated by the wallet with .with_db() classmethod * start fixing tests * all tests passing except those that need to set a specific private key * bip39 mnemonic to seed - with db but restore doesnt work yet with custom seed * mnemonic restore works * enter mnemonic in cli * fix tests to use different mnemonic * properly ask user for seed input * tests: dont ask for inputs * try to fix tests * fix cashu -d * fixing * bump version and add more text to mnemonic enter * add more comments * add many more comments and type annotations in the wallet * dont print generated mnemonic and dont wait for input * fix test * does this fix tests? * sigh.... * make format * do not restore from an initialized wallet * fix mnemonics * fix nitpicks * print wallet name if nonstandard wallet * fix merge error and remove comments * poetry lock and requirements * remove unused code * fix tests * mnemonic.lower() and add keyset id if not present for backwards compat * edit comment
184 lines
5.6 KiB
Python
184 lines
5.6 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
import pytest_asyncio
|
|
from fastapi.testclient import TestClient
|
|
|
|
from cashu.core.migrations import migrate_databases
|
|
from cashu.core.settings import settings
|
|
from cashu.wallet import migrations
|
|
from cashu.wallet.api.app import app
|
|
from cashu.wallet.wallet import Wallet
|
|
from tests.conftest import SERVER_ENDPOINT, mint
|
|
|
|
|
|
@pytest_asyncio.fixture(scope="function")
|
|
async def wallet(mint):
|
|
wallet = await Wallet.with_db(
|
|
url=SERVER_ENDPOINT,
|
|
db="data/test_wallet_api",
|
|
name="wallet_api",
|
|
)
|
|
await wallet.load_mint()
|
|
wallet.status()
|
|
yield wallet
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_invoice(wallet: Wallet):
|
|
with TestClient(app) as client:
|
|
response = client.post("/invoice?amount=100")
|
|
assert response.status_code == 200
|
|
if settings.lightning:
|
|
assert response.json()["invoice"]
|
|
else:
|
|
assert response.json()["amount"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_invoice_with_split(wallet: Wallet):
|
|
with TestClient(app) as client:
|
|
response = client.post("/invoice?amount=10&split=1")
|
|
assert response.status_code == 200
|
|
if settings.lightning:
|
|
assert response.json()["invoice"]
|
|
else:
|
|
assert response.json()["amount"]
|
|
# await wallet.load_proofs(reload=True)
|
|
# assert wallet.proof_amounts.count(1) >= 10
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_balance():
|
|
with TestClient(app) as client:
|
|
response = client.get("/balance")
|
|
assert response.status_code == 200
|
|
assert response.json()["balance"]
|
|
assert response.json()["keysets"]
|
|
assert response.json()["mints"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send(wallet: Wallet):
|
|
with TestClient(app) as client:
|
|
response = client.post("/send?amount=10")
|
|
assert response.status_code == 200
|
|
assert response.json()["balance"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_without_split(wallet: Wallet):
|
|
with TestClient(app) as client:
|
|
response = client.post("/send?amount=2&nosplit=true")
|
|
assert response.status_code == 200
|
|
assert response.json()["balance"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_without_split_but_wrong_amount(wallet: Wallet):
|
|
with TestClient(app) as client:
|
|
response = client.post("/send?amount=10&nosplit=true")
|
|
assert response.status_code == 400
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_pending():
|
|
with TestClient(app) as client:
|
|
response = client.get("/pending")
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_receive_all(wallet: Wallet):
|
|
with TestClient(app) as client:
|
|
response = client.post("/receive?all=true")
|
|
assert response.status_code == 200
|
|
assert response.json()["initial_balance"]
|
|
assert response.json()["balance"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_burn_all(wallet: Wallet):
|
|
with TestClient(app) as client:
|
|
response = client.post("/send?amount=20")
|
|
assert response.status_code == 200
|
|
response = client.post("/burn?all=true")
|
|
assert response.status_code == 200
|
|
assert response.json()["balance"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_pay():
|
|
with TestClient(app) as client:
|
|
invoice = (
|
|
"lnbc100n1pjzp22cpp58xvjxvagzywky9xz3vurue822aaax"
|
|
"735hzc5pj5fg307y58v5znqdq4vdshx6r4ypjx2ur0wd5hgl"
|
|
"h6ahauv24wdmac4zk478pmwfzd7sdvm8tje3dmfue3lc2g4l"
|
|
"9g40a073h39748uez9p8mxws5vqwjmkqr4wl5l7n4dlhj6z6"
|
|
"va963cqvufrs4"
|
|
)
|
|
response = client.post(f"/pay?invoice={invoice}")
|
|
if not settings.lightning:
|
|
assert response.status_code == 400
|
|
else:
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_lock():
|
|
with TestClient(app) as client:
|
|
response = client.get("/lock")
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_locks():
|
|
with TestClient(app) as client:
|
|
response = client.get("/locks")
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_invoices():
|
|
with TestClient(app) as client:
|
|
response = client.get("/invoices")
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_wallets():
|
|
with TestClient(app) as client:
|
|
response = client.get("/wallets")
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_info():
|
|
with TestClient(app) as client:
|
|
response = client.get("/info")
|
|
assert response.status_code == 200
|
|
assert response.json()["version"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_flow(wallet: Wallet):
|
|
with TestClient(app) as client:
|
|
if not settings.lightning:
|
|
response = client.get("/balance")
|
|
initial_balance = response.json()["balance"]
|
|
response = client.post("/invoice?amount=100")
|
|
response = client.get("/balance")
|
|
assert response.json()["balance"] == initial_balance + 100
|
|
response = client.post("/send?amount=50")
|
|
response = client.get("/balance")
|
|
assert response.json()["balance"] == initial_balance + 50
|
|
response = client.post("/send?amount=50")
|
|
response = client.get("/balance")
|
|
assert response.json()["balance"] == initial_balance
|
|
response = client.get("/pending")
|
|
token = response.json()["pending_token"]["0"]["token"]
|
|
amount = response.json()["pending_token"]["0"]["amount"]
|
|
response = client.post(f"/receive?token={token}")
|
|
response = client.get("/balance")
|
|
assert response.json()["balance"] == initial_balance + amount
|