add test

ordinals · Jan 16, 2023 · 08395d2 · 08395d2
1 parent 45a55fc
commit 08395d2
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 0 deletions.
diff --git a/test-bitcoincore-rpc/src/lib.rs b/test-bitcoincore-rpc/src/lib.rs
@@ -240,6 +240,10 @@ impl Handle {
       Network::Regtest => Network::Regtest.to_string(),
     }
   }
+
+  pub fn loaded_wallets(&self) -> BTreeSet<String> {
+    self.state().loaded_wallets.clone()
+  }
 }
 
 impl Drop for Handle {

diff --git a/test-bitcoincore-rpc/src/server.rs b/test-bitcoincore-rpc/src/server.rs
@@ -565,6 +565,7 @@ impl Api for Server {
 
   fn load_wallet(&self, wallet: String) -> Result<LoadWalletResult, jsonrpc_core::Error> {
     if self.state().wallets.contains(&wallet) {
+      self.state().loaded_wallets.insert(wallet.clone());
       Ok(LoadWalletResult {
         name: wallet,
         warning: None,

diff --git a/test-bitcoincore-rpc/src/state.rs b/test-bitcoincore-rpc/src/state.rs
@@ -15,6 +15,7 @@ pub(crate) struct State {
   pub(crate) version: usize,
   pub(crate) wallet_name: String,
   pub(crate) wallets: BTreeSet<String>,
+  pub(crate) loaded_wallets: BTreeSet<String>,
 }
 
 impl State {
@@ -47,6 +48,7 @@ impl State {
       version,
       wallet_name: wallet_name.to_string(),
       wallets: BTreeSet::new(),
+      loaded_wallets: BTreeSet::new(),
     }
   }
 

diff --git a/tests/wallet/transactions.rs b/tests/wallet/transactions.rs
@@ -5,10 +5,14 @@ fn transactions() {
   let rpc_server = test_bitcoincore_rpc::spawn();
   create_wallet(&rpc_server);
 
+  assert_eq!(rpc_server.loaded_wallets().len(), 0);
+
   CommandBuilder::new("wallet transactions")
     .rpc_server(&rpc_server)
     .run();
 
+  assert_eq!(rpc_server.loaded_wallets().len(), 1);
+
   rpc_server.mine_blocks(1);
 
   CommandBuilder::new("wallet transactions")