Add exclude providers unittest

1200wd · May 6, 2024 · 3e49628 · 3e49628
1 parent a2784a1
commit 3e49628
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/bitcoinlib/wallets.py b/bitcoinlib/wallets.py
@@ -2485,7 +2485,8 @@ def keys(self, account_id=None, name=None, key_id=None, change=None, depth=None,
                 keys2.append({k: v for (k, v) in key.items()
                               if k[:1] != '_' and k != 'wallet' and k not in private_fields})
             return keys2
-        qr.session.close()
+        # qr.session.close()
+        qr.session.commit()
         return keys
 
     def keys_networks(self, used=None, as_dict=False):

diff --git a/tests/test_services.py b/tests/test_services.py
@@ -792,6 +792,12 @@ def test_service_transaction_unconfirmed(self):
         self.assertIsNone(t.date)
         self.assertIsNone(t.block_height)
 
+    def test_service_exlude_providers(self):
+        srv = ServiceTest(network='testnet', cache_uri='')
+        providers = [srv.providers[pi]['provider'] for pi in srv.providers]
+        srv2 = ServiceTest(network='testnet', exclude_providers=providers[1:], cache_uri='')
+        self.assertEqual(len(srv2.providers), 1)
+
 
 class TestServiceCache(unittest.TestCase):