AaronFiore/electrum
1
0
Fork 0
Code Activity

wallet: set_frozen_state_of_coins to handle freeze=None

Browse Source 
Internally whether a coin is frozen is tri-state:
- forced-True, set by the user
- forced-False, set by the user
- unset/default: is_frozen_coin() can decide whether the coin should be frozen

This patch lets set_frozen_state_of_coins() undo a previous explicit setting of True/False,
by calling it with a value of None.
Note: there is still no way in the GUI to undo an explicit setting of True/False.
This commit is contained in:
SomberNight
2025-03-13 18:30:53 +00:00
parent 9351183876
commit 8ccd31fe49
2 changed files with 43 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
tests/test_wallet_vertical.py
View File

@@ -2895,16 +2895,39 @@ class TestWalletSending(ElectrumTestCase):
self.assertEqual(
{'52e669a20a26c8b3df5b41e5e6309b18bcde8e1ad7ea17a18f63b6dc6c8becc0:1'},
{txi.prevout.to_str() for txi in wallet.get_spendable_coins(["tb1q6n99dl96mx8mfh90m3tn5awk5mllkzdh25dw7z"])})
utxo1 = "c36a6e1cd54df108e69574f70bc9b88dc13beddc70cfad9feb7f8f6593255d4a:1"
utxo2 = "52e669a20a26c8b3df5b41e5e6309b18bcde8e1ad7ea17a18f63b6dc6c8becc0:1"
# test freezing an address
wallet.set_frozen_state_of_addresses(["tb1q6n99dl96mx8mfh90m3tn5awk5mllkzdh25dw7z"], freeze=True)
self.assertEqual(
{'c36a6e1cd54df108e69574f70bc9b88dc13beddc70cfad9feb7f8f6593255d4a:1'},
{txi.prevout.to_str() for txi in wallet.get_spendable_coins()})
wallet.set_frozen_state_of_addresses(["tb1q6n99dl96mx8mfh90m3tn5awk5mllkzdh25dw7z"], freeze=False)
self.assertEqual(
{'c36a6e1cd54df108e69574f70bc9b88dc13beddc70cfad9feb7f8f6593255d4a:1',
'52e669a20a26c8b3df5b41e5e6309b18bcde8e1ad7ea17a18f63b6dc6c8becc0:1'},
{txi.prevout.to_str() for txi in wallet.get_spendable_coins()})
with self.subTest(msg="freeze_address"):
wallet.set_frozen_state_of_addresses(["tb1q6n99dl96mx8mfh90m3tn5awk5mllkzdh25dw7z"], freeze=True)
self.assertEqual(
{utxo1},
{txi.prevout.to_str() for txi in wallet.get_spendable_coins()})
wallet.set_frozen_state_of_addresses(["tb1q6n99dl96mx8mfh90m3tn5awk5mllkzdh25dw7z"], freeze=False)
self.assertEqual(
{utxo1, utxo2},
{txi.prevout.to_str() for txi in wallet.get_spendable_coins()})
# test freezing a utxo
with self.subTest(msg="freeze_coin"):
self.assertTrue(utxo1 not in wallet._frozen_coins)
wallet.set_frozen_state_of_coins([utxo1], freeze=True)
self.assertEqual(wallet._frozen_coins.get(utxo1), True)
self.assertEqual(
{utxo2},
{txi.prevout.to_str() for txi in wallet.get_spendable_coins()})
wallet.set_frozen_state_of_coins([utxo1], freeze=False)
self.assertEqual(wallet._frozen_coins.get(utxo1), False)
self.assertEqual(
{utxo1, utxo2},
{txi.prevout.to_str() for txi in wallet.get_spendable_coins()})
wallet.set_frozen_state_of_coins([utxo1], freeze=None)
self.assertTrue(utxo1 not in wallet._frozen_coins)
@mock.patch.object(wallet.Abstract_Wallet, 'save_db')
async def test_export_psbt_with_xpubs__multisig(self, mock_save_db):