""" Tests for wallet-dependent methods in ``bal.core.will``. Uses mocking to simulate Electrum wallet, network, and db. Run: source electrum/env/bin/activate QT_QPA_PLATFORM=offscreen python3 tests/test_core_will_extra.py """ import os import sys from unittest.mock import MagicMock, patch sys.path.insert(0, os.path.join(os.path.dirname(__file__), os.pardir)) from electrum.transaction import Transaction from bal.core.will import Will, WillItem _VALID_TX_HEX = ( "01000000012a5c9a94fcde98f5581cd00162c60a13936ceb75389ea65b" "f38633b424eb4031000000006c493046022100a82bbc57a0136751e543" "3f41cf000b3f1a99c6744775e76ec764fb78c54ee100022100f9e80b7d" "e89de861dc6fb0c1429d5da72c2b6b2ee2406bc9bfb1beedd729d98501" "2102e61d176da16edd1d258a200ad9759ef63adf8e14cd97f53227bae3" "5cdb84d2f6ffffffff0140420f00000000001976a914230ac37834073a" "42146f11ef8414ae929feaafc388ac00000000" ) # Patch Transaction.add_info_from_wallet so it's a no-op during all tests _patcher = patch.object(Transaction, "add_info_from_wallet") _patcher.start() # ------------------------------------------------------------------ # # Will.check_tx_height # ------------------------------------------------------------------ # def test_check_tx_height(): wallet = MagicMock() wallet.get_tx_info.return_value.tx_mined_status.height.return_value = 100 tx = MagicMock() assert Will.check_tx_height(tx, wallet) == 100 def test_check_tx_height_zero(): wallet = MagicMock() wallet.get_tx_info.return_value.tx_mined_status.height.return_value = 0 tx = MagicMock() assert Will.check_tx_height(tx, wallet) == 0 # ------------------------------------------------------------------ # # Will.add_info_from_will # ------------------------------------------------------------------ # def test_add_info_from_will(): wallet = MagicMock() willitem = MagicMock() will = {"wid": willitem} Will.add_info_from_will(will, "wid", wallet) willitem.tx.add_info_from_wallet.assert_called_once_with(wallet) def test_add_info_from_will_no_wallet(): willitem = MagicMock() will = {"wid": willitem} Will.add_info_from_will(will, "wid", None) def test_add_info_from_will_tx_is_str(): wallet = MagicMock() willitem = MagicMock() willitem.tx = _VALID_TX_HEX will = {"wid": willitem} Will.add_info_from_will(will, "wid", wallet) assert hasattr(willitem.tx, "add_info_from_wallet") # ------------------------------------------------------------------ # # Will.check_invalidated # ------------------------------------------------------------------ # def test_check_invalidated_confirmed(): wallet = MagicMock() wallet.get_tx_info.return_value.tx_mined_status.height.return_value = 100 item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100}) will = {"wid": item} Will.check_invalidated(will, [], wallet) assert item.get_status("CONFIRMED") is True def test_check_invalidated_mempool(): # PENDING was renamed to MEMPOOL (A2): a tx seen with height 0 (in the # mempool, not yet mined) is flagged as MEMPOOL. wallet = MagicMock() wallet.get_tx_info.return_value.tx_mined_status.height.return_value = 0 item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100}) will = {"wid": item} Will.check_invalidated(will, [], wallet) assert item.get_status("MEMPOOL") is True def test_legacy_pending_migrates_to_mempool(): # Backward-compatibility (A2, "Modo B"): a will saved by an older plugin # version stores the flag under the legacy "PENDING" key. Loading it must # carry that flag over to the new "MEMPOOL" status, so nothing is lost. item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100, "PENDING": True}) assert item.get_status("MEMPOOL") is True def test_new_mempool_wins_over_legacy_pending(): # If both the new "MEMPOOL" key and the legacy "PENDING" key are present, # the new key wins: an explicit MEMPOOL=False is NOT overridden by a stale # legacy PENDING=True. item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100, "MEMPOOL": False, "PENDING": True}) assert item.get_status("MEMPOOL") is False def test_updated_status_keeps_valid(): # A2 rule: setting UPDATED must NOT clear the VALID flag (the tx is replaced # by a new one that keeps the same locktime and same heirs). item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100, "VALID": True}) item.set_status("UPDATED", True) assert item.get_status("UPDATED") is True assert item.get_status("VALID") is True def test_anticipated_status_keeps_valid(): # A2 rule: setting ANTICIPATED must NOT clear the VALID flag (anticipating # only moves the locktime 1 day earlier; the tx stays valid). item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100, "VALID": True}) item.set_status("ANTICIPATED", True) assert item.get_status("ANTICIPATED") is True assert item.get_status("VALID") is True def test_mempool_status_clears_valid(): # A2 rule (unchanged from old PENDING behaviour): setting MEMPOOL clears the # VALID flag. item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100, "VALID": True}) item.set_status("MEMPOOL", True) assert item.get_status("MEMPOOL") is True assert item.get_status("VALID") is False def test_check_invalidated_invalidated(): wallet = MagicMock() wallet.get_tx_info.return_value.tx_mined_status.height.return_value = -1 item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100}) will = {"wid": item} Will.check_invalidated(will, [], wallet) assert item.get_status("INVALIDATED") is True # ------------------------------------------------------------------ # # Will.check_will (exercises check_invalidated + search_rai) # ------------------------------------------------------------------ # def test_check_will(): wallet = MagicMock() wallet.get_tx_info.return_value.tx_mined_status.height.return_value = 0 item = WillItem({"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100}) will = {"wid": item} # check_will signature is now (will, all_utxos, wallet, timestamp_to_check): # block_to_check was removed in A1 (locktimes are always timestamps). Will.check_will(will, [], wallet, 9999999999) # should be MEMPOOL (height=0); PENDING was renamed to MEMPOOL in A2. assert item.get_status("MEMPOOL") is True # ------------------------------------------------------------------ # # WillItem.__init__ with wallet # ------------------------------------------------------------------ # def test_willitem_init_with_wallet(): wallet = MagicMock() w = {"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100} item = WillItem(w, wallet=wallet) assert item is not None def test_willitem_init_without_wallet(): w = {"tx": _VALID_TX_HEX, "heirs": {"a": ["addr", 100, "30d"]}, "willexecutor": None, "status": "", "description": "", "time": 0, "change": "", "baltx_fees": 100} item = WillItem(w) assert item is not None # ------------------------------------------------------------------ # # Main # ------------------------------------------------------------------ # if __name__ == "__main__": for name in sorted(dir()): if name.startswith("test_"): globals()[name]() print(f" [OK] {name}") print("[OK] All will-extra tests passed")