add tests

tests/smoke_test.py

@@ -0,0 +1,92 @@
+"""
+Smoke test for the BAL Electrum plugin.
+
+Goal: after every refactor step, prove that the plugin still imports cleanly
+under a real Electrum 4.7.2 + PyQt6 install, and that a handful of pure-logic
+behaviours produce *exactly* the same results as before (regression guard).
+
+Run with:
+    QT_QPA_PLATFORM=offscreen python3 tests/smoke_test.py <PLUGIN_IMPORT_NAME>
+
+where <PLUGIN_IMPORT_NAME> is the dotted module path the plugin is reachable
+at, e.g. "electrum.plugins.BAL" (original) or "electrum.plugins.bal" (new).
+"""
+import importlib
+import sys
+
+PKG = sys.argv[1] if len(sys.argv) > 1 else "electrum.plugins.BAL"
+
+
+def imp(mod):
+    return importlib.import_module(f"{PKG}.{mod}")
+
+
+def main():
+    # --- Qt must be initialised before importing any gui module ---
+    from PyQt6.QtWidgets import QApplication  # noqa
+    _app = QApplication.instance() or QApplication([])
+
+    results = {}
+
+    # 1) Core modules import (these must be GUI-free).
+    bal = imp_core("bal", "core.plugin_base")
+    util = imp_core("util", "core.util")
+    heirs = imp_core("heirs", "core.heirs")
+    will = imp_core("will", "core.will")
+    we = imp_core("willexecutors", "core.willexecutors")
+
+    # 2) GUI module imports.
+    qt = imp_gui()
+
+    # 3) Behaviour checks (pure logic, must be identical across versions).
+    BalTimestamp = bal.BalTimestamp
+    assert BalTimestamp("30d").duration_to_days() == 30, "BalTimestamp 30d"
+    assert BalTimestamp("1y").duration_to_days() == 365, "BalTimestamp 1y"
+    assert str(BalTimestamp("7d")) == "7d", "BalTimestamp str"
+
+    Util = util.Util
+    assert Util.is_perc("50%") is True
+    assert Util.is_perc("100") is False
+    assert Util.text_to_hex("BAL") == "42414c"
+    assert Util.hex_to_text("42414c") == "BAL"
+    assert Util.int_locktime(days=1) == 86400
+
+    # heirs constants must keep the same column layout (very delicate!)
+    assert heirs.HEIR_ADDRESS == 0
+    assert heirs.HEIR_AMOUNT == 1
+    assert heirs.HEIR_LOCKTIME == 2
+    assert heirs.HEIR_REAL_AMOUNT == 3
+    assert heirs.HEIR_DUST_AMOUNT == 4
+
+    # WillItem default status table must stay intact.
+    assert will.WillItem.STATUS_DEFAULT["VALID"][1] is True
+
+    # 4) Plugin class wiring.
+    assert qt.Plugin.__bases__[0] is bal.BalPlugin
+    for h in ("create_status_bar", "init_menubar", "load_wallet", "close_wallet"):
+        assert hasattr(qt.Plugin, h), f"missing hook {h}"
+
+    print(f"[OK] smoke test passed for package '{PKG}'")
+
+
+def imp_core(old_name, new_name):
+    """Import a core module, trying the new layout first then the old flat one."""
+    for candidate in (new_name, old_name):
+        try:
+            return importlib.import_module(f"{PKG}.{candidate}")
+        except ModuleNotFoundError:
+            continue
+    raise ModuleNotFoundError(f"cannot import {old_name}/{new_name} from {PKG}")
+
+
+def imp_gui():
+    for candidate in ("gui.qt.plugin", "qt"):
+        try:
+            return importlib.import_module(f"{PKG}.{candidate}")
+        except ModuleNotFoundError:
+            continue
+    raise ModuleNotFoundError(f"cannot import gui module from {PKG}")
+
+
+if __name__ == "__main__":
+    main()