forked from bitcoinafterlife/bal-electrum-plugin
kaibot
6d568bf304
When deleting old transactions and clicking Check, task_phase1 would fail with 'NoneType object has no attribute get' error in WillItem.normalize_locktime. Root cause: In Will.normalize_will (will.py:150), the parameter others_inputs defaults to None. At line 151, a local variable 'others_input' is created with a safe default value (empty dict). However, lines 179 and 184 still used the original 'others_inputs' parameter instead of the safe local variable, causing the error when calling .get() on None. Fix: Use the safe 'others_input' variable instead of the raw parameter in lines 179 and 184. Also added: - Traceback logging to on_error_phase1 so future errors include full call stack - Debug logging in update_all to log willitems state before processing - Comprehensive test (test_e5_build_with_real_wallet_heirs_and_utxos) that reproduces the exact scenario from the user's karen7 wallet
652 lines
22 KiB
Python
652 lines
22 KiB
Python
"""
|
|
Group E - Mock tests built around a fake wallet named "karen7".
|
|
|
|
These tests exercise the four behaviour areas requested for Group E using a
|
|
single, self-contained fake wallet (``karen7``). No real Electrum wallet
|
|
file is needed: everything is driven by lightweight mocks/stubs, exactly like
|
|
the other ``test_group_*`` and ``test_core_*`` suites.
|
|
|
|
The four sections are:
|
|
|
|
* E1 - calendar / .ics: reminder-offset distribution, separate-VEVENT shape,
|
|
description escaping and temporary .ics file creation.
|
|
* E2 - inheritance / states: WillItem status transitions and heir-change
|
|
detection for karen7's inheritance.
|
|
* E3 - connectivity: parallel pinging of karen7's will-executor servers
|
|
(concurrency, per-server callback, write-back of results).
|
|
* E4 - will-executor: selection flag and the filtering rule that decides which
|
|
transactions are pushed to a will-executor.
|
|
|
|
Run:
|
|
QT_QPA_PLATFORM=offscreen PYTHONPATH=electrum-src \
|
|
python3 -m pytest tests/test_group_e_mock_karen7.py -q
|
|
"""
|
|
|
|
import copy
|
|
import json
|
|
import os
|
|
import sys
|
|
import time
|
|
import traceback
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
# Make the plugin package importable when run directly (tests/ is one level
|
|
# below the repo root that contains the ``bal`` package).
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), os.pardir))
|
|
|
|
from bal.core.will import WillItem, Will, HeirNotFoundException
|
|
from bal.core.heirs import Heirs
|
|
from bal.core.willexecutors import Willexecutors
|
|
from bal.gui.qt.calendar import BalCalendar
|
|
from bal.gui.qt.widgets import compute_reminder_offsets
|
|
|
|
|
|
# A valid serialized Bitcoin transaction hex (1 input + 1 P2PKH output,
|
|
# version 2). Reused from test_core_will.py so WillItem can parse a real tx.
|
|
_VALID_TX_HEX = (
|
|
"01000000012a5c9a94fcde98f5581cd00162c60a13936ceb75389ea65b"
|
|
"f38633b424eb4031000000006c493046022100a82bbc57a0136751e543"
|
|
"3f41cf000b3f1a99c6744775e76ec764fb78c54ee100022100f9e80b7d"
|
|
"e89de861dc6fb0c1429d5da72c2b6b2ee2406bc9bfb1beedd729d98501"
|
|
"2102e61d176da16edd1d258a200ad9759ef63adf8e14cd97f53227bae3"
|
|
"5cdb84d2f6ffffffff0140420f00000000001976a914230ac37834073a"
|
|
"42146f11ef8414ae929feaafc388ac00000000"
|
|
)
|
|
|
|
|
|
# ------------------------------------------------------------------ #
|
|
# The fake "karen7" wallet
|
|
# ------------------------------------------------------------------ #
|
|
|
|
class FakeDB:
|
|
"""Minimal stand-in for an Electrum wallet database (a dict with
|
|
get/put), used so the Heirs model can read/write the "heirs" key."""
|
|
|
|
def __init__(self, data=None):
|
|
"""Store the initial key/value mapping (empty by default)."""
|
|
self._data = data or {}
|
|
|
|
def get(self, key, default=None):
|
|
"""Return the stored value for ``key`` or ``default`` if missing."""
|
|
return self._data.get(key, default)
|
|
|
|
def put(self, key, value):
|
|
"""Store ``value`` under ``key``."""
|
|
self._data[key] = value
|
|
|
|
|
|
class Karen7Wallet:
|
|
"""A fake wallet named "karen7".
|
|
|
|
It mimics just enough of an Electrum wallet for the Group E tests:
|
|
|
|
* ``str(wallet)`` returns the wallet name ("karen7"), which is what the
|
|
plugin uses to build calendar UIDs and labels.
|
|
* ``db`` is a :class:`FakeDB` pre-loaded with two heirs so the Heirs model
|
|
can be constructed without a live wallet.
|
|
"""
|
|
|
|
#: Wallet name, exposed via ``str(wallet)``.
|
|
NAME = "karen7"
|
|
|
|
def __init__(self, heirs=None):
|
|
"""Build the wallet with an optional custom heirs mapping.
|
|
|
|
Args:
|
|
heirs: optional ``{name: [address, amount, locktime]}`` mapping.
|
|
Defaults to two heirs (alice, bob).
|
|
"""
|
|
if heirs is None:
|
|
heirs = {
|
|
"alice": ["addr_alice", "50%", "30d"],
|
|
"bob": ["addr_bob", "10000", "90d"],
|
|
}
|
|
self.db = FakeDB({"heirs": heirs})
|
|
|
|
def __str__(self):
|
|
"""Return the wallet name so it can be used in labels/UIDs."""
|
|
return self.NAME
|
|
|
|
|
|
def make_karen7_wallet(heirs=None):
|
|
"""Factory returning a fresh "karen7" fake wallet for each test."""
|
|
return Karen7Wallet(heirs=heirs)
|
|
|
|
|
|
def _make_willitem(**overrides):
|
|
"""Create a WillItem for karen7 from a minimal, valid dict.
|
|
|
|
The status block is reset to the clean defaults so each test starts from a
|
|
deterministic state (VALID=True, everything else False).
|
|
"""
|
|
d = {
|
|
"tx": _VALID_TX_HEX,
|
|
"heirs": {"alice": ["addr_alice", 5000, "30d"]},
|
|
"willexecutor": None,
|
|
"status": "",
|
|
"description": "",
|
|
"time": 0,
|
|
"change": "",
|
|
"baltx_fees": 100,
|
|
}
|
|
d.update(overrides)
|
|
item = WillItem(d)
|
|
item.STATUS = copy.deepcopy(WillItem.STATUS_DEFAULT)
|
|
return item
|
|
|
|
|
|
# ================================================================== #
|
|
# E1 - calendar / .ics
|
|
# ================================================================== #
|
|
|
|
def test_e1_karen7_wallet_name():
|
|
"""The fake wallet identifies itself as "karen7" (used in .ics UIDs)."""
|
|
wallet = make_karen7_wallet()
|
|
assert str(wallet) == "karen7"
|
|
|
|
|
|
def test_e1_reminder_offsets_long_period():
|
|
"""Over a 30-day period, 3 reminders are spread out and all fall before
|
|
the deadline (offset >= 1 day)."""
|
|
offsets = compute_reminder_offsets(30, 3)
|
|
assert offsets == [30, 16, 1]
|
|
assert all(o >= 1 for o in offsets)
|
|
# Sorted descending: furthest-from-deadline reminder first.
|
|
assert offsets == sorted(offsets, reverse=True)
|
|
|
|
|
|
def test_e1_reminder_offsets_capped_one_per_day():
|
|
"""If the period is shorter than the requested count, at most one reminder
|
|
per day is produced (no duplicate same-day alarms)."""
|
|
offsets = compute_reminder_offsets(2, 3)
|
|
assert offsets == [2, 1]
|
|
assert len(offsets) == len(set(offsets))
|
|
|
|
|
|
def test_e1_reminder_offsets_no_room():
|
|
"""A zero-day (or shorter) period yields no reminders at all."""
|
|
assert compute_reminder_offsets(0, 3) == []
|
|
|
|
|
|
def test_e1_reminder_offsets_single():
|
|
"""A single reminder fires exactly one day before the deadline."""
|
|
assert compute_reminder_offsets(10, 1) == [1]
|
|
|
|
|
|
def test_e1_build_separate_events_for_karen7():
|
|
"""Build the SEPARATE reminder VEVENTs for karen7 the same way
|
|
open_or_save_calendar does (one VEVENT per offset, each on its own date)
|
|
and verify their shape.
|
|
|
|
We replicate the pure part of WillSettingsWidget.open_or_save_calendar here
|
|
so the test stays GUI-free (no Qt widget construction), while still
|
|
asserting the exact iCalendar structure the plugin emits: N distinct events,
|
|
unique UIDs, "(reminder N/total)" summaries, and the last event placed one
|
|
day before the locktime.
|
|
"""
|
|
from datetime import datetime, timedelta, timezone
|
|
|
|
# 30-day period -> offsets [30, 16, 1] (the last one = 1 day before deadline).
|
|
locktime = datetime(2026, 7, 23, 9, 0, 0, tzinfo=timezone.utc)
|
|
offsets = compute_reminder_offsets(30, 3)
|
|
total = len(offsets)
|
|
wallet = "karen7"
|
|
summary_base = f"BAL - Will execution of {wallet}"
|
|
|
|
lines = ["BEGIN:VCALENDAR", "VERSION:2.0"]
|
|
for idx, offset in enumerate(offsets, start=1):
|
|
event_dt = BalCalendar.format_time(locktime - timedelta(days=offset))
|
|
summary = BalCalendar.ical_escape(f"{summary_base} (reminder {idx}/{total})")
|
|
lines.extend([
|
|
"BEGIN:VEVENT",
|
|
f"UID:bal-{wallet}-{offset}d",
|
|
f"DTSTART:{event_dt}",
|
|
f"DTEND:{event_dt}",
|
|
f"SUMMARY:{summary}",
|
|
"END:VEVENT",
|
|
])
|
|
lines.append("END:VCALENDAR")
|
|
|
|
# One separate VEVENT per reminder offset (no VALARM blocks any more).
|
|
assert lines.count("BEGIN:VEVENT") == len(offsets)
|
|
assert lines.count("END:VEVENT") == len(offsets)
|
|
assert "BEGIN:VALARM" not in lines
|
|
# UIDs are unique per event so calendars do not merge them.
|
|
uids = [ln for ln in lines if ln.startswith("UID:")]
|
|
assert len(uids) == len(set(uids)) == len(offsets)
|
|
# Summaries are numbered to tell the events apart.
|
|
assert any("(reminder 1/3)" in ln for ln in lines)
|
|
assert any("(reminder 3/3)" in ln for ln in lines)
|
|
# The LAST event (offset 1) sits one day before the locktime.
|
|
last_dt = BalCalendar.format_time(locktime - timedelta(days=1))
|
|
assert f"DTSTART:{last_dt}" in lines
|
|
|
|
|
|
def test_e1_event_description_escaping():
|
|
"""Special iCalendar characters in karen7's event text are escaped so
|
|
the .ics file stays valid."""
|
|
raw = "Wallet karen7; heirs: alice, bob"
|
|
escaped = BalCalendar.ical_escape(raw)
|
|
assert "\\;" in escaped # semicolon escaped
|
|
assert "\\," in escaped # comma escaped
|
|
assert "karen7" in escaped
|
|
|
|
|
|
def test_e1_write_temp_ics_for_karen7():
|
|
"""A minimal VCALENDAR for karen7 can be written to a real temp .ics
|
|
file and read back byte-for-byte."""
|
|
content = (
|
|
"BEGIN:VCALENDAR\r\n"
|
|
"VERSION:2.0\r\n"
|
|
"BEGIN:VEVENT\r\n"
|
|
"UID:bal-karen7\r\n"
|
|
"SUMMARY:BAL will reminder\r\n"
|
|
"END:VEVENT\r\n"
|
|
"END:VCALENDAR\r\n"
|
|
)
|
|
path = BalCalendar.write_temp_ics(content)
|
|
try:
|
|
assert os.path.isfile(path)
|
|
with open(path, "rb") as f:
|
|
assert f.read() == content.encode("utf-8")
|
|
finally:
|
|
os.unlink(path)
|
|
|
|
|
|
# ================================================================== #
|
|
# E2 - inheritance / states
|
|
# ================================================================== #
|
|
|
|
def test_e2_karen7_heirs_loaded():
|
|
"""karen7's two default heirs are read from the wallet db."""
|
|
wallet = make_karen7_wallet()
|
|
heirs = Heirs(wallet)
|
|
assert "alice" in heirs
|
|
assert "bob" in heirs
|
|
assert len(heirs) == 2
|
|
|
|
|
|
def test_e2_karen7_add_remove_heir():
|
|
"""Adding and removing an heir updates both the model and the wallet db."""
|
|
wallet = make_karen7_wallet()
|
|
heirs = Heirs(wallet)
|
|
|
|
heirs["charlie"] = ["addr_charlie", "20000", "30d"]
|
|
assert "charlie" in heirs
|
|
assert "charlie" in wallet.db.get("heirs", {})
|
|
|
|
removed = heirs.pop("alice")
|
|
assert removed is not None
|
|
assert "alice" not in heirs
|
|
|
|
|
|
def test_e2_willitem_default_and_complete():
|
|
"""A fresh will-item for karen7 starts VALID-but-not-COMPLETE; marking
|
|
it COMPLETE flips only that flag."""
|
|
item = _make_willitem()
|
|
assert item.get_status("VALID") is True
|
|
assert item.get_status("COMPLETE") is False
|
|
|
|
assert item.set_status("COMPLETE", True) is True
|
|
assert item.get_status("COMPLETE") is True
|
|
# Re-setting the same value is a no-op (returns None).
|
|
assert item.set_status("COMPLETE", True) is None
|
|
|
|
|
|
def test_e2_invalidated_clears_valid():
|
|
"""Invalidating karen7's will clears its VALID flag."""
|
|
item = _make_willitem()
|
|
item.set_status("INVALIDATED", True)
|
|
assert item.get_status("INVALIDATED") is True
|
|
assert item.get_status("VALID") is False
|
|
|
|
|
|
def test_e2_pushed_clears_push_fail():
|
|
"""A successful push clears any previous PUSH_FAIL state."""
|
|
item = _make_willitem()
|
|
item.set_status("PUSH_FAIL", True)
|
|
assert item.get_status("PUSH_FAIL") is True
|
|
|
|
item.set_status("PUSHED", True)
|
|
assert item.get_status("PUSHED") is True
|
|
assert item.get_status("PUSH_FAIL") is False
|
|
|
|
|
|
def test_e2_only_valid_filters_invalidated():
|
|
"""Will.only_valid keeps only the still-valid items of karen7's will."""
|
|
good = _make_willitem()
|
|
bad = _make_willitem()
|
|
bad.set_status("INVALIDATED", True)
|
|
will = {"good": good, "bad": bad}
|
|
|
|
valid = list(Will.only_valid(will))
|
|
assert "good" in valid
|
|
assert "bad" not in valid
|
|
|
|
|
|
def test_e2_heir_change_triggers_rebuild():
|
|
"""If karen7 removes an heir after signing, the mismatch between the
|
|
frozen will and the current heirs must raise HeirNotFoundException so the
|
|
inheritance is rebuilt."""
|
|
lt = 1900000000
|
|
will_heirs = {
|
|
"alice": ["addr_alice", 5000, str(lt)],
|
|
"bob": ["addr_bob", 3000, str(lt)],
|
|
}
|
|
current_heirs = {"alice": ["addr_alice", 5000, str(lt)]} # bob removed
|
|
|
|
item = WillItem(
|
|
{
|
|
"tx": _VALID_TX_HEX,
|
|
"heirs": copy.deepcopy(will_heirs),
|
|
"willexecutor": None,
|
|
"status": "",
|
|
"description": "",
|
|
"time": 0,
|
|
"change": "",
|
|
"baltx_fees": 100,
|
|
}
|
|
)
|
|
item.STATUS = copy.deepcopy(WillItem.STATUS_DEFAULT)
|
|
item.tx.locktime = lt
|
|
will = {"willid_1": item}
|
|
|
|
raised = False
|
|
try:
|
|
Will.check_willexecutors_and_heirs(will, current_heirs, {}, False, 0, 100)
|
|
except HeirNotFoundException:
|
|
raised = True
|
|
assert raised, "removing an heir must raise HeirNotFoundException"
|
|
|
|
|
|
# ================================================================== #
|
|
# E3 - connectivity
|
|
# ================================================================== #
|
|
|
|
# Each simulated will-executor server takes this long to "answer".
|
|
_SLOW = 0.3
|
|
# Number of simulated servers for karen7.
|
|
_N = 6
|
|
|
|
|
|
def test_e3_ping_servers_parallel():
|
|
"""karen7's will-executor servers are pinged concurrently.
|
|
|
|
We stub ``get_info_task`` with a slow function: half the servers "fail".
|
|
The test asserts that (a) total time is far below the sequential sum
|
|
(proving concurrency), (b) the per-server ``on_each`` callback fires once
|
|
with the correct ok flag, and (c) results are written back into the
|
|
mapping.
|
|
"""
|
|
|
|
def slow_get_info(url, we, **kwargs):
|
|
"""Stub: sleep, then mark the server ok/KO based on its url."""
|
|
time.sleep(_SLOW)
|
|
we["status"] = "KO" if "dead" in url else 200
|
|
return we
|
|
|
|
original = Willexecutors.get_info_task
|
|
Willexecutors.get_info_task = staticmethod(slow_get_info)
|
|
try:
|
|
wes = {}
|
|
for i in range(_N):
|
|
kind = "dead" if i % 2 else "ok"
|
|
wes[f"https://{kind}-{i}.karen7.example"] = {}
|
|
|
|
seen = []
|
|
|
|
def on_each(url, we, ok):
|
|
"""Record each server's url and ok flag as it answers."""
|
|
seen.append((url, ok))
|
|
|
|
start = time.time()
|
|
Willexecutors.ping_servers_parallel(wes, on_each=on_each, max_workers=_N)
|
|
elapsed = time.time() - start
|
|
|
|
# Parallel must be far faster than running every server in sequence.
|
|
sequential = _N * _SLOW
|
|
assert elapsed < sequential * 0.6, (
|
|
f"not parallel: {elapsed:.2f}s vs sequential {sequential:.2f}s"
|
|
)
|
|
|
|
# The callback fired once per server with the right ok flag.
|
|
assert len(seen) == _N
|
|
for url, ok in seen:
|
|
assert ok == ("ok" in url), (url, ok)
|
|
|
|
# Ping results were written back into karen7's server mapping.
|
|
for url, we in wes.items():
|
|
assert we["status"] == (200 if "ok" in url else "KO"), (url, we)
|
|
finally:
|
|
Willexecutors.get_info_task = original
|
|
|
|
|
|
def test_e3_ping_empty_mapping():
|
|
"""Pinging an empty server mapping is a safe no-op (returns the mapping)."""
|
|
result = Willexecutors.ping_servers_parallel({})
|
|
assert result == {}
|
|
|
|
|
|
# ================================================================== #
|
|
# E4 - will-executor
|
|
# ================================================================== #
|
|
|
|
def test_e4_is_selected_default_false():
|
|
"""A brand-new will-executor for karen7 is not selected by default, and
|
|
is_selected initialises the flag to False."""
|
|
we = {"url": "https://we.karen7.example"}
|
|
assert Willexecutors.is_selected(we) is False
|
|
assert we["selected"] is False
|
|
|
|
|
|
def test_e4_is_selected_set_value():
|
|
"""is_selected can both set and read the selection flag."""
|
|
we = {"url": "https://we.karen7.example"}
|
|
assert Willexecutors.is_selected(we, True) is True
|
|
assert we["selected"] is True
|
|
assert Willexecutors.is_selected(we) is True
|
|
|
|
|
|
def test_e4_get_transactions_only_valid_complete_selected():
|
|
"""Only a VALID + COMPLETE + not-yet-PUSHED will whose will-executor is
|
|
*selected* contributes a transaction to be pushed.
|
|
|
|
This is the core filtering rule of get_willexecutor_transactions, verified
|
|
here against karen7's will.
|
|
"""
|
|
url = "https://we.karen7.example"
|
|
|
|
# 1) The good one: valid, complete, not pushed, selected -> included.
|
|
good = _make_willitem()
|
|
good.set_status("COMPLETE", True)
|
|
good.we = {"url": url, "selected": True}
|
|
|
|
# 2) Not complete -> excluded.
|
|
not_complete = _make_willitem()
|
|
not_complete.we = {"url": url, "selected": True}
|
|
|
|
# 3) Complete but already pushed -> excluded (unless force).
|
|
pushed = _make_willitem()
|
|
pushed.set_status("COMPLETE", True)
|
|
pushed.set_status("PUSHED", True)
|
|
pushed.we = {"url": url, "selected": True}
|
|
|
|
# 4) Complete, not pushed, but will-executor NOT selected -> excluded.
|
|
not_selected = _make_willitem()
|
|
not_selected.set_status("COMPLETE", True)
|
|
not_selected.we = {"url": url, "selected": False}
|
|
|
|
will = {
|
|
"good": good,
|
|
"not_complete": not_complete,
|
|
"pushed": pushed,
|
|
"not_selected": not_selected,
|
|
}
|
|
|
|
result = Willexecutors.get_willexecutor_transactions(will)
|
|
|
|
# Exactly one server entry, carrying only the "good" will-item's tx id.
|
|
assert url in result
|
|
assert result[url]["txsids"] == ["good"]
|
|
|
|
|
|
def test_e4_get_transactions_force_includes_pushed():
|
|
"""With ``force=True``, an already-PUSHED will is re-included so karen7
|
|
can re-broadcast it (the "Rebroadcast" button path)."""
|
|
url = "https://we.karen7.example"
|
|
|
|
pushed = _make_willitem()
|
|
pushed.set_status("COMPLETE", True)
|
|
pushed.set_status("PUSHED", True)
|
|
pushed.we = {"url": url, "selected": True}
|
|
|
|
will = {"pushed": pushed}
|
|
|
|
result = Willexecutors.get_willexecutor_transactions(will, force=True)
|
|
assert url in result
|
|
assert result[url]["txsids"] == ["pushed"]
|
|
|
|
|
|
def test_e4_compute_id():
|
|
"""compute_id builds a stable identifier from url + chain."""
|
|
we = {"url": "https://we.karen7.example", "chain": "bitcoin"}
|
|
assert Willexecutors.compute_id(we) == "https://we.karen7.example-bitcoin"
|
|
|
|
|
|
# ================================================================== #
|
|
# E5 - Build inheritance from the real wallet file
|
|
# ================================================================== #
|
|
|
|
def test_e5_build_with_real_wallet_heirs_and_utxos():
|
|
"""Read heirs and UTXOs from the real ``tests/karen7`` file, pass them
|
|
through ``Heirs.buildTransactions`` and report any unhandled error.
|
|
|
|
Reproduces the ``NoneType object has no attribute 'get'`` error the
|
|
user saw when building an inheritance with the same data from the GUI.
|
|
"""
|
|
wallet_path = os.path.join(os.path.dirname(__file__), "karen7")
|
|
with open(wallet_path) as f:
|
|
data = json.load(f)
|
|
|
|
assert "heirs" in data, "karen7 wallet has no heirs"
|
|
|
|
class FakeUTXO:
|
|
"""Minimal stand-in for PartialTxInput, providing only the fields
|
|
used by buildTransactions / prepare_transactions."""
|
|
def __init__(self, txid, out_idx, value_sats):
|
|
self._value = value_sats
|
|
self.prevout = MagicMock()
|
|
self.prevout.txid = txid
|
|
self.prevout.out_idx = out_idx
|
|
|
|
def value_sats(self):
|
|
return self._value
|
|
|
|
# Build UTXOs from unspent txo entries.
|
|
utxos = []
|
|
txo = data.get("txo", {})
|
|
for txid, outputs in txo.items():
|
|
if not isinstance(outputs, dict):
|
|
continue
|
|
for addr, out_map in outputs.items():
|
|
if not isinstance(out_map, dict):
|
|
continue
|
|
for idx, info in out_map.items():
|
|
if not isinstance(info, list) or len(info) < 2:
|
|
continue
|
|
value, spent = info[0], info[1]
|
|
if spent is False:
|
|
utxos.append(FakeUTXO(txid, int(idx), value))
|
|
|
|
assert len(utxos) > 0, "no unspent UTXOs in karen7 wallet"
|
|
|
|
# Build the Heirs model from the real wallet data.
|
|
heirs_data = data["heirs"]
|
|
h = Heirs.__new__(Heirs)
|
|
h.update(heirs_data)
|
|
|
|
assert len(h) == 4, f"expected 4 heirs, got {len(h)}"
|
|
assert list(h.keys()) == ["aaaa", "lucia", "mario", "mario2"]
|
|
|
|
# Mock the Electrum-heavy parts so the build can run in a test context.
|
|
wallet = MagicMock()
|
|
wallet.dust_threshold.return_value = 500
|
|
wallet.get_change_addresses_for_new_transaction.return_value = [
|
|
"bcrt1q0000000000000000000000000000000000000"
|
|
]
|
|
|
|
bal_plugin = MagicMock()
|
|
bal_plugin.NO_WILLEXECUTOR.get.return_value = True # bypass will-executors
|
|
bal_plugin.get_decimal_point.return_value = 8
|
|
|
|
# Mock PartialTxOutput.from_address_and_value to return something usable.
|
|
_fake_outputs = []
|
|
|
|
def _fake_from_address_and_value(address, value):
|
|
out = MagicMock()
|
|
out.value = value
|
|
out.address = address
|
|
out.scriptpubkey = b""
|
|
out.script_descriptor = ""
|
|
_fake_outputs.append(out)
|
|
return out
|
|
|
|
# Mock PartialTransaction.from_io to record inputs and return a fake tx.
|
|
_fake_txs = []
|
|
|
|
def _fake_from_io(inputs, outputs, locktime, version):
|
|
tx = MagicMock()
|
|
tx.txid.return_value = f"faketx_{len(_fake_txs):04x}"
|
|
tx.estimated_size.return_value = 100
|
|
tx.get_fee.return_value = 100
|
|
tx.input_value.return_value = sum(inp.value_sats() for inp in inputs)
|
|
tx.output_value.return_value = sum(out.value for out in outputs)
|
|
tx.get_output_idxs_from_address.return_value = [0]
|
|
# Attributes set by the caller after construction.
|
|
tx.description = ""
|
|
tx.heirsvalue = 0
|
|
tx.my_locktime = 0
|
|
tx.willexecutor = None
|
|
tx.heirs = {}
|
|
tx.available_utxos = []
|
|
_fake_txs.append(tx)
|
|
return tx
|
|
|
|
with patch(
|
|
"bal.core.heirs.bitcoin.is_address", return_value=True
|
|
), patch(
|
|
"bal.core.heirs.PartialTxOutput.from_address_and_value",
|
|
side_effect=_fake_from_address_and_value,
|
|
), patch(
|
|
"bal.core.heirs.PartialTransaction.from_io",
|
|
side_effect=_fake_from_io,
|
|
):
|
|
try:
|
|
result = h.buildTransactions(
|
|
bal_plugin, wallet, tx_fees=1, utxos=utxos
|
|
)
|
|
except Exception as exc:
|
|
tb = traceback.format_exc()
|
|
pytest.fail(
|
|
f"buildTransactions raised {type(exc).__name__}: {exc}\n\n{tb}"
|
|
)
|
|
|
|
assert result, (
|
|
"buildTransactions returned empty — no transactions were built "
|
|
"(check whether NO_WILLEXECUTOR needs to be True)"
|
|
)
|
|
|
|
|
|
# ------------------------------------------------------------------ #
|
|
# Main (allows running the file directly, like the other suites)
|
|
# ------------------------------------------------------------------ #
|
|
|
|
if __name__ == "__main__":
|
|
for name in sorted(dir()):
|
|
if name.startswith("test_"):
|
|
globals()[name]()
|
|
print(f" [OK] {name}")
|
|
print("[OK] All Group E (karen7) tests passed")
|