plm-lottery/tests/unit/test_psbt_builder.py

import pytest
from embit import script
from embit.bip32 import HDKey

from app.wallet.plm_network import PLM_MAINNET
from app.wallet.psbt_builder import (
    InsufficientFundsError,
    Utxo,
    build_signed_transaction,
    estimate_vsize,
    select_utxos,
)


def _key(seed_byte: int) -> HDKey:
    root = HDKey.from_seed(bytes([seed_byte]) * 32, version=PLM_MAINNET["xprv"])
    return root.derive("m/84h/746h/0h/0/0")


def test_estimate_vsize_grows_with_inputs_and_outputs():
    assert estimate_vsize(1, 2) < estimate_vsize(2, 2)
    assert estimate_vsize(1, 1) < estimate_vsize(1, 2)


def test_select_utxos_picks_largest_first():
    utxos = [Utxo("a" * 64, 0, 5_000_000), Utxo("b" * 64, 0, 20_000_000), Utxo("c" * 64, 0, 1_000_000)]
    selected, total = select_utxos(utxos, target_sats=10_000_000)
    assert selected == [utxos[1]]  # the 20M UTXO alone covers 10M
    assert total == 20_000_000


def test_select_utxos_raises_when_insufficient():
    utxos = [Utxo("a" * 64, 0, 1_000_000)]
    with pytest.raises(InsufficientFundsError):
        select_utxos(utxos, target_sats=10_000_000)


def test_build_signed_transaction_deducts_fee_from_amount_not_change():
    signer = _key(1)
    from_script = script.p2wpkh(signer.to_public())
    my_address = from_script.address(network=PLM_MAINNET)
    to_address = script.p2wpkh(_key(2).to_public()).address(network=PLM_MAINNET)

    utxos = [Utxo("11" * 32, 0, 150_000_000)]
    built = build_signed_transaction(
        signing_key=signer,
        from_script=from_script,
        utxos=utxos,
        to_address=to_address,
        amount_sats=10_000_000,
        change_address=my_address,
        fee_rate_sat_vb=1,
    )

    fee = estimate_vsize(1, 2)
    assert built.fee_sats == fee
    assert built.recipient_sats == 10_000_000 - fee
    # change reflects the full amount_sats deducted from the sender, fee comes out
    # of what the recipient gets, not out of the sender's remaining balance
    assert built.change_sats == 150_000_000 - 10_000_000
    assert built.spent_utxos == utxos
    assert len(built.txid) == 64

    from embit.transaction import Transaction

    parsed = Transaction.parse(bytes.fromhex(built.raw_hex))
    assert len(parsed.vin[0].witness.items) == 2
    assert len(parsed.vout) == 2


def test_build_signed_transaction_omits_change_output_when_exact_amount():
    signer = _key(3)
    from_script = script.p2wpkh(signer.to_public())
    my_address = from_script.address(network=PLM_MAINNET)
    to_address = script.p2wpkh(_key(4).to_public()).address(network=PLM_MAINNET)

    utxos = [Utxo("22" * 32, 0, 10_000_000)]  # exactly amount_sats, zero change
    built = build_signed_transaction(
        signing_key=signer,
        from_script=from_script,
        utxos=utxos,
        to_address=to_address,
        amount_sats=10_000_000,
        change_address=my_address,
        fee_rate_sat_vb=1,
    )
    assert built.change_sats == 0

    from embit.transaction import Transaction

    parsed = Transaction.parse(bytes.fromhex(built.raw_hex))
    assert len(parsed.vout) == 1


def test_build_signed_transaction_raises_when_amount_smaller_than_fee():
    signer = _key(5)
    from_script = script.p2wpkh(signer.to_public())
    my_address = from_script.address(network=PLM_MAINNET)
    to_address = script.p2wpkh(_key(6).to_public()).address(network=PLM_MAINNET)

    small_amount = 100  # smaller than the ~141 sat fee at 1 sat/vB for 1-in-2-out
    assert small_amount < estimate_vsize(1, 2)
    utxos = [Utxo("33" * 32, 0, small_amount)]
    with pytest.raises(InsufficientFundsError):
        build_signed_transaction(
            signing_key=signer,
            from_script=from_script,
            utxos=utxos,
            to_address=to_address,
            amount_sats=small_amount,
            change_address=my_address,
            fee_rate_sat_vb=1,
        )