basic wallet test

Author: oren-z0

tests/test_timelock_recovery.py

@@ -0,0 +1,72 @@
+from io import StringIO
+import json
+import os, sys
+from electrum.bitcoin import address_to_script
+from electrum.plugins.timelock_recovery.timelock_recovery import TimelockRecoveryContext
+from electrum.simple_config import SimpleConfig
+from electrum.storage import WalletStorage
+from electrum.transaction import PartialTxOutput
+from electrum.wallet import Wallet
+from electrum.wallet_db import WalletDB
+
+from . import ElectrumTestCase
+
+
+class TestTimelockRecovery(ElectrumTestCase):
+    TESTNET = True
+    maxDiff = None
+
+    def setUp(self):
+        super(TestTimelockRecovery, self).setUp()
+        self.config = SimpleConfig({'electrum_path': self.electrum_path})
+
+        self.wallet_path = os.path.join(self.electrum_path, "timelock_recovery_wallet")
+
+        self._saved_stdout = sys.stdout
+        self._stdout_buffer = StringIO()
+        sys.stdout = self._stdout_buffer
+
+    def tearDown(self):
+        super(TestTimelockRecovery, self).tearDown()
+        # Restore the "real" stdout
+        sys.stdout = self._saved_stdout
+
+
+    async def test_get_alert_address(self):
+        with open(os.path.join(os.path.dirname(__file__), "test_timelock_recovery", "default_wallet"), "r") as f:
+            wallet_str = f.read()
+        storage = WalletStorage(self.wallet_path)
+        db = WalletDB(wallet_str, storage=storage, upgrade=True)
+        wallet = Wallet(db, config=self.config)
+
+        context = TimelockRecoveryContext(wallet)
+        alert_address = context.get_alert_address()
+        self.assertEqual(alert_address, 'tb1qt339ksrha0n5a6lwpql778erkm272hxgamdc0u')
+
+    async def test_get_alert_address(self):
+        with open(os.path.join(os.path.dirname(__file__), "test_timelock_recovery", "default_wallet"), "r") as f:
+            wallet_str = f.read()
+        storage = WalletStorage(self.wallet_path)
+        db = WalletDB(wallet_str, storage=storage, upgrade=True)
+        wallet = Wallet(db, config=self.config)
+
+        context = TimelockRecoveryContext(wallet)
+        alert_address = context.get_cancellation_address()
+        self.assertEqual(alert_address, 'tb1qt339ksrha0n5a6lwpql778erkm272hxgamdc0u')
+
+    async def test_make_unsigned_alert_tx(self):
+        with open(os.path.join(os.path.dirname(__file__), "test_timelock_recovery", "default_wallet"), "r") as f:
+            wallet_str = f.read()
+        storage = WalletStorage(self.wallet_path)
+        db = WalletDB(wallet_str, storage=storage, upgrade=True)
+        wallet = Wallet(db, config=self.config)
+
+        context = TimelockRecoveryContext(wallet)
+        context.outputs = [
+            PartialTxOutput(scriptpubkey=address_to_script('tb1qt339ksrha0n5a6lwpql778erkm272hxgamdc0u'), value='!'),
+        ]
+
+        alert_tx = context.make_unsigned_alert_tx(fee_est=5000)
+        alert_tx_json = json.dumps(alert_tx.to_json(), sort_keys=True, indent=2)
+        print(alert_tx_json)
+        self.assertEqual(json.dumps(alert_tx.to_json(), sort_keys=True, indent=None), "{}")