From 72985f14ad6fca36384516c1dc3812a733d5d4a0 Mon Sep 17 00:00:00 2001 From: thunderbiscuit Date: Tue, 21 May 2024 10:30:48 -0400 Subject: [PATCH] tests: update python tests to use amount type --- bdk-python/tests/test_live_tx_builder.py | 12 ++++-------- bdk-python/tests/test_live_wallet.py | 7 +++---- 2 files changed, 7 insertions(+), 12 deletions(-) diff --git a/bdk-python/tests/test_live_tx_builder.py b/bdk-python/tests/test_live_tx_builder.py index 31aad2f..431b461 100644 --- a/bdk-python/tests/test_live_tx_builder.py +++ b/bdk-python/tests/test_live_tx_builder.py @@ -13,7 +13,7 @@ class LiveTxBuilderTest(unittest.TestCase): def test_tx_builder(self): descriptor: bdk.Descriptor = bdk.Descriptor( - "wpkh([c258d2e4/84h/1h/0h]tpubDDYkZojQFQjht8Tm4jsS3iuEmKjTiEGjG6KnuFNKKJb5A6ZUCUZKdvLdSDWofKi4ToRCwb9poe1XdqfUnP4jaJjCB2Zwv11ZLgSbnZSNecE/0/*)", + "wpkh(tprv8ZgxMBicQKsPf2qfrEygW6fdYseJDDrVnDv26PH5BHdvSuG6ecCbHqLVof9yZcMoM31z9ur3tTYbSnr1WBqbGX97CbXcmp5H6qeMpyvx35B/84h/1h/0h/0/*)", bdk.Network.SIGNET ) wallet: bdk.Wallet = bdk.Wallet( @@ -43,22 +43,18 @@ class LiveTxBuilderTest(unittest.TestCase): network=bdk.Network.SIGNET ) - psbt = bdk.TxBuilder().add_recipient(script=recipient.script_pubkey(), amount=4200).fee_rate(fee_rate=bdk.FeeRate.from_sat_per_vb(2)).finish(wallet) + psbt = bdk.TxBuilder().add_recipient(script=recipient.script_pubkey(), amount=bdk.Amount.from_sat(4200)).fee_rate(fee_rate=bdk.FeeRate.from_sat_per_vb(2)).finish(wallet) self.assertTrue(psbt.serialize().startswith("cHNi"), "The PSBT should start with cHNi") def complex_tx_builder(self): descriptor: bdk.Descriptor = bdk.Descriptor( - "wpkh([c258d2e4/84h/1h/0h]tpubDDYkZojQFQjht8Tm4jsS3iuEmKjTiEGjG6KnuFNKKJb5A6ZUCUZKdvLdSDWofKi4ToRCwb9poe1XdqfUnP4jaJjCB2Zwv11ZLgSbnZSNecE/0/*)", - bdk.Network.SIGNET - ) - change_descriptor: bdk.Descriptor = bdk.Descriptor( - "wpkh([c258d2e4/84h/1h/0h]tpubDDYkZojQFQjht8Tm4jsS3iuEmKjTiEGjG6KnuFNKKJb5A6ZUCUZKdvLdSDWofKi4ToRCwb9poe1XdqfUnP4jaJjCB2Zwv11ZLgSbnZSNecE/1/*)", + "wpkh(tprv8ZgxMBicQKsPf2qfrEygW6fdYseJDDrVnDv26PH5BHdvSuG6ecCbHqLVof9yZcMoM31z9ur3tTYbSnr1WBqbGX97CbXcmp5H6qeMpyvx35B/84h/1h/0h/0/*)", bdk.Network.SIGNET ) wallet: bdk.Wallet = bdk.Wallet( descriptor, - change_descriptor, + None, "./bdk_persistence.db", bdk.Network.SIGNET ) diff --git a/bdk-python/tests/test_live_wallet.py b/bdk-python/tests/test_live_wallet.py index 8d27ad0..5a1a1c8 100644 --- a/bdk-python/tests/test_live_wallet.py +++ b/bdk-python/tests/test_live_wallet.py @@ -43,8 +43,8 @@ class LiveWalletTest(unittest.TestCase): for tx in transactions: sent_and_received = wallet.sent_and_received(tx.transaction) print(f"Transaction: {tx.transaction.txid()}") - print(f"Sent {sent_and_received.sent}") - print(f"Received {sent_and_received.received}") + print(f"Sent {sent_and_received.sent.to_sat()}") + print(f"Received {sent_and_received.received.to_sat()}") def test_broadcast_transaction(self): @@ -79,8 +79,7 @@ class LiveWalletTest(unittest.TestCase): network=bdk.Network.SIGNET ) - psbt: bdk.Psbt = bdk.TxBuilder().add_recipient(script=recipient.script_pubkey(), amount=4200).fee_rate(fee_rate=bdk.FeeRate.from_sat_per_vb(2)).finish(wallet) - # print(psbt.serialize()) + psbt: bdk.Psbt = bdk.TxBuilder().add_recipient(script=recipient.script_pubkey(), amount=bdk.Amount.from_sat(4200)).fee_rate(fee_rate=bdk.FeeRate.from_sat_per_vb(2)).finish(wallet) self.assertTrue(psbt.serialize().startswith("cHNi"), "The PSBT should start with cHNi") walletDidSign = wallet.sign(psbt)