forked from bitcoinafterlife/bal-electrum-plugin
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
"""
|
|
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()
|