107 lines
3.9 KiB
Python
107 lines
3.9 KiB
Python
 
|
import pytest
|
||
|
|
from sqlalchemy import select
|
||
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
|
||
|
|
|
||
|
|
from app.bets.service import BetError, place_bet
|
||
|
|
from app.config import settings
|
||
|
|
from app.db.base import Base
|
||
|
|
from app.db.models import AuditLog, PendingTransaction, RoundParticipant, User, UtxoEvent
|
||
|
|
from app.wallet.hd import derive_user_address
|
||
|
|
|
||
|
|
|
||
|
|
class FakeElectrumClient:
|
||
|
|
def __init__(self):
|
||
|
|
self.broadcasted: list[str] = []
|
||
|
|
|
||
|
|
async def broadcast(self, raw_tx_hex: str) -> str:
|
||
|
|
self.broadcasted.append(raw_tx_hex)
|
||
|
|
return "fake-network-txid"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
async def session_factory(tmp_path, monkeypatch):
|
||
|
|
monkeypatch.setattr(settings, "master_key_path", str(tmp_path / "master.xprv.enc"))
|
||
|
|
monkeypatch.setattr(settings, "xprv_encryption_key", __import__("cryptography.fernet", fromlist=["Fernet"]).Fernet.generate_key().decode())
|
||
|
|
from app.wallet import hd
|
||
|
|
|
||
|
|
hd._account_key = None
|
||
|
|
hd.generate_master_key()
|
||
|
|
|
||
|
|
engine = create_async_engine("sqlite+aiosqlite:///:memory:")
|
||
|
|
async with engine.begin() as conn:
|
||
|
|
await conn.run_sync(Base.metadata.create_all)
|
||
|
|
yield async_sessionmaker(engine, expire_on_commit=False)
|
||
|
|
await engine.dispose()
|
||
|
|
hd._account_key = None
|
||
|
|
|
||
|
|
|
||
|
|
async def _make_funded_user(session_factory, index: int, funded_sats: int) -> int:
|
||
|
|
async with session_factory() as session:
|
||
|
|
address = derive_user_address(index)
|
||
|
|
user = User(username=f"user{index}", password_hash="x", derivation_index=index, address=address)
|
||
|
|
session.add(user)
|
||
|
|
await session.commit()
|
||
|
|
session.add(
|
||
|
|
UtxoEvent(
|
||
|
|
user_id=user.id,
|
||
|
|
txid=f"{index:02x}" * 32,
|
||
|
|
vout=0,
|
||
|
|
amount_sats=funded_sats,
|
||
|
|
confirmed_height=100,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
await session.commit()
|
||
|
|
return user.id
|
||
|
|
|
||
|
|
|
||
|
|
async def test_place_bet_broadcasts_and_records_participant(session_factory):
|
||
|
|
user_id = await _make_funded_user(session_factory, 0, 1_500_000_000)
|
||
|
|
client = FakeElectrumClient()
|
||
|
|
|
||
|
|
async with session_factory() as session:
|
||
|
|
user = await session.get(User, user_id)
|
||
|
|
participant = await place_bet(session, client, user)
|
||
|
|
|
||
|
|
assert client.broadcasted # a raw tx was broadcast
|
||
|
|
assert participant.status == "broadcast"
|
||
|
|
assert participant.bet_txid
|
||
|
|
|
||
|
|
async with session_factory() as session:
|
||
|
|
utxo = (await session.scalars(select(UtxoEvent).where(UtxoEvent.user_id == user_id))).one()
|
||
|
|
assert utxo.spent_txid == participant.bet_txid
|
||
|
|
|
||
|
|
pending = (await session.scalars(select(PendingTransaction))).one()
|
||
|
|
assert pending.kind == "bet"
|
||
|
|
|
||
|
|
audit_events = (await session.scalars(select(AuditLog))).all()
|
||
|
|
assert any(e.event_type == "bet_placed" for e in audit_events)
|
||
|
|
assert pending.current_txid == participant.bet_txid
|
||
|
|
|
||
|
|
|
||
|
|
async def test_place_bet_rejects_insufficient_balance(session_factory):
|
||
|
|
user_id = await _make_funded_user(session_factory, 1, 1_000_000) # below bet_amount_sats
|
||
|
|
client = FakeElectrumClient()
|
||
|
|
|
||
|
|
async with session_factory() as session:
|
||
|
|
user = await session.get(User, user_id)
|
||
|
|
with pytest.raises(BetError, match="insufficient balance"):
|
||
|
|
await place_bet(session, client, user)
|
||
|
|
|
||
|
|
|
||
|
|
async def test_place_bet_rejects_second_bet_same_round(session_factory):
|
||
|
|
user_id = await _make_funded_user(session_factory, 2, 3_000_000_000)
|
||
|
|
client = FakeElectrumClient()
|
||
|
|
|
||
|
|
async with session_factory() as session:
|
||
|
|
user = await session.get(User, user_id)
|
||
|
|
await place_bet(session, client, user)
|
||
|
|
|
||
|
|
async with session_factory() as session:
|
||
|
|
user = await session.get(User, user_id)
|
||
|
|
with pytest.raises(BetError, match="already"):
|
||
|
|
await place_bet(session, client, user)
|
||
|
|
|
||
|
|
async with session_factory() as session:
|
||
|
|
participants = (await session.scalars(select(RoundParticipant))).all()
|
||
|
|
assert len(participants) == 1
|