tests(python): add in-memory connection

bitcoindevkit · Aug 7, 2024 · 52ee8a2 · 52ee8a2
1 parent cdbf424
commit 52ee8a2
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 9 deletions.
diff --git a/bdk-python/tests/test_live_tx_builder.py b/bdk-python/tests/test_live_tx_builder.py
@@ -25,10 +25,12 @@ def test_tx_builder(self):
             "wpkh(tprv8ZgxMBicQKsPf2qfrEygW6fdYseJDDrVnDv26PH5BHdvSuG6ecCbHqLVof9yZcMoM31z9ur3tTYbSnr1WBqbGX97CbXcmp5H6qeMpyvx35B/84h/1h/0h/0/*)",
             bdk.Network.SIGNET
         )
+        connection: bdk.Connection = bdk.Connection.new_in_memory()
         wallet: bdk.Wallet = bdk.Wallet(
             descriptor,
             change_descriptor,
-            bdk.Network.SIGNET
+            bdk.Network.SIGNET,
+            connection
         )
         esplora_client: bdk.EsploraClient = bdk.EsploraClient(url = SIGNET_ESPLORA_URL)
         full_scan_request: bdk.FullScanRequest = wallet.start_full_scan()
@@ -55,10 +57,12 @@ def test_tx_builder(self):
         self.assertTrue(psbt.serialize().startswith("cHNi"), "The PSBT should start with cHNi")
 
     def complex_tx_builder(self):
+        connection: bdk.Connection = bdk.Connection.new_in_memory()
         wallet: bdk.Wallet = bdk.Wallet(
             descriptor,
             change_descriptor,
-            bdk.Network.SIGNET
+            bdk.Network.SIGNET,
+            connection
         )
         esplora_client: bdk.EsploraClient = bdk.EsploraClient(url = SIGNET_ESPLORA_URL)
         full_scan_request: bdk.FullScanRequest = wallet.start_full_scan()
@@ -95,4 +99,4 @@ def complex_tx_builder(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()
diff --git a/bdk-python/tests/test_live_wallet.py b/bdk-python/tests/test_live_wallet.py
@@ -21,10 +21,12 @@ def tearDown(self) -> None:
             os.remove("./bdk_persistence.sqlite")
 
     def test_synced_balance(self):
+        connection: bdk.Connection = bdk.Connection.new_in_memory()
         wallet: bdk.Wallet = bdk.Wallet(
             descriptor,
             change_descriptor,
-            bdk.Network.SIGNET
+            bdk.Network.SIGNET,
+            connection
         )
         esplora_client: bdk.EsploraClient = bdk.EsploraClient(url = SIGNET_ESPLORA_URL)
         full_scan_request: bdk.FullScanRequest = wallet.start_full_scan()
@@ -51,10 +53,12 @@ def test_synced_balance(self):
 
 
     def test_broadcast_transaction(self):
+        connection: bdk.Connection = bdk.Connection.new_in_memory()
         wallet: bdk.Wallet = bdk.Wallet(
             descriptor,
             change_descriptor,
-            bdk.Network.SIGNET
+            bdk.Network.SIGNET,
+            connection
         )
         esplora_client: bdk.EsploraClient = bdk.EsploraClient(url = SIGNET_ESPLORA_URL)
         full_scan_request: bdk.FullScanRequest = wallet.start_full_scan()
@@ -92,4 +96,4 @@ def test_broadcast_transaction(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()
diff --git a/bdk-python/tests/test_offline_wallet.py b/bdk-python/tests/test_offline_wallet.py
@@ -18,10 +18,12 @@ def tearDown(self) -> None:
             os.remove("./bdk_persistence.sqlite")
 
     def test_new_address(self):
-        wallet: Wallet = bdk.Wallet(
+        connection: bdk.Connection = bdk.Connection.new_in_memory()
+        wallet: bdk.Wallet = bdk.Wallet(
             descriptor,
             change_descriptor,
-            bdk.Network.TESTNET
+            bdk.Network.TESTNET,
+            connection
         )
         address_info: bdk.AddressInfo = wallet.reveal_next_address(bdk.KeychainKind.EXTERNAL)
 
@@ -33,10 +35,12 @@ def test_new_address(self):
         self.assertEqual("tb1qrnfslnrve9uncz9pzpvf83k3ukz22ljgees989", address_info.address.__str__())
 
     def test_balance(self):
+        connection: bdk.Connection = bdk.Connection.new_in_memory()
         wallet: bdk.Wallet = bdk.Wallet(
             descriptor,
             change_descriptor,
-            bdk.Network.TESTNET
+            bdk.Network.TESTNET,
+            connection
         )
 
         self.assertEqual(wallet.balance().total.to_sat(), 0)