2 # Copyright (c) 2017 The Bitcoin Core developers
3 # Distributed under the MIT software license, see the accompanying
4 # file COPYING or http://www.opensource.org/licenses/mit-license.php.
7 Verify that a bitcoind node can load multiple wallet files
12 from test_framework
.test_framework
import BitcoinTestFramework
13 from test_framework
.util
import assert_equal
, assert_raises_rpc_error
15 class MultiWalletTest(BitcoinTestFramework
):
16 def set_test_params(self
):
17 self
.setup_clean_chain
= True
19 self
.extra_args
= [['-wallet=w1', '-wallet=w2', '-wallet=w3']]
22 assert_equal(set(self
.nodes
[0].listwallets()), {"w1", "w2", "w3"})
26 # should not initialize if there are duplicate wallets
27 self
.assert_start_raises_init_error(0, ['-wallet=w1', '-wallet=w1'], 'Error loading wallet w1. Duplicate -wallet filename specified.')
29 # should not initialize if wallet file is a directory
30 wallet_dir
= os
.path
.join(self
.options
.tmpdir
, 'node0', 'regtest', 'wallets')
31 os
.mkdir(os
.path
.join(wallet_dir
, 'w11'))
32 self
.assert_start_raises_init_error(0, ['-wallet=w11'], 'Error loading wallet w11. -wallet filename must be a regular file.')
34 # should not initialize if one wallet is a copy of another
35 shutil
.copyfile(os
.path
.join(wallet_dir
, 'w2'), os
.path
.join(wallet_dir
, 'w22'))
36 self
.assert_start_raises_init_error(0, ['-wallet=w2', '-wallet=w22'], 'duplicates fileid')
38 # should not initialize if wallet file is a symlink
39 os
.symlink(os
.path
.join(wallet_dir
, 'w1'), os
.path
.join(wallet_dir
, 'w12'))
40 self
.assert_start_raises_init_error(0, ['-wallet=w12'], 'Error loading wallet w12. -wallet filename must be a regular file.')
42 # should not initialize if the specified walletdir does not exist
43 self
.assert_start_raises_init_error(0, ['-walletdir=bad'], 'Error: Specified wallet directory "bad" does not exist.')
45 # if wallets/ doesn't exist, datadir should be the default wallet dir
46 wallet_dir2
= os
.path
.join(self
.options
.tmpdir
, 'node0', 'regtest', 'walletdir')
47 os
.rename(wallet_dir
, wallet_dir2
)
48 self
.start_node(0, ['-wallet=w4', '-wallet=w5'])
49 assert_equal(set(self
.nodes
[0].listwallets()), {"w4", "w5"})
50 w5
= self
.nodes
[0].get_wallet_rpc("w5")
54 # now if wallets/ exists again, but the rootdir is specified as the walletdir, w4 and w5 should still be loaded
55 os
.rename(wallet_dir2
, wallet_dir
)
56 self
.start_node(0, ['-wallet=w4', '-wallet=w5', '-walletdir=' + os
.path
.join(self
.options
.tmpdir
, 'node0', 'regtest')])
57 assert_equal(set(self
.nodes
[0].listwallets()), {"w4", "w5"})
58 w5
= self
.nodes
[0].get_wallet_rpc("w5")
59 w5_info
= w5
.getwalletinfo()
60 assert_equal(w5_info
['immature_balance'], 50)
64 self
.start_node(0, self
.extra_args
[0])
66 w1
= self
.nodes
[0].get_wallet_rpc("w1")
67 w2
= self
.nodes
[0].get_wallet_rpc("w2")
68 w3
= self
.nodes
[0].get_wallet_rpc("w3")
69 wallet_bad
= self
.nodes
[0].get_wallet_rpc("bad")
73 # accessing invalid wallet fails
74 assert_raises_rpc_error(-18, "Requested wallet does not exist or is not loaded", wallet_bad
.getwalletinfo
)
76 # accessing wallet RPC without using wallet endpoint fails
77 assert_raises_rpc_error(-19, "Wallet file not specified", self
.nodes
[0].getwalletinfo
)
79 # check w1 wallet balance
80 w1_info
= w1
.getwalletinfo()
81 assert_equal(w1_info
['immature_balance'], 50)
82 w1_name
= w1_info
['walletname']
83 assert_equal(w1_name
, "w1")
85 # check w2 wallet balance
86 w2_info
= w2
.getwalletinfo()
87 assert_equal(w2_info
['immature_balance'], 0)
88 w2_name
= w2_info
['walletname']
89 assert_equal(w2_name
, "w2")
91 w3_name
= w3
.getwalletinfo()['walletname']
92 assert_equal(w3_name
, "w3")
94 assert_equal({"w1", "w2", "w3"}, {w1_name
, w2_name
, w3_name
})
97 assert_equal(w1
.getbalance(), 100)
98 assert_equal(w2
.getbalance(), 0)
99 assert_equal(w3
.getbalance(), 0)
101 w1
.sendtoaddress(w2
.getnewaddress(), 1)
102 w1
.sendtoaddress(w3
.getnewaddress(), 2)
104 assert_equal(w2
.getbalance(), 1)
105 assert_equal(w3
.getbalance(), 2)
107 batch
= w1
.batch([w1
.getblockchaininfo
.get_request(), w1
.getwalletinfo
.get_request()])
108 assert_equal(batch
[0]["result"]["chain"], "regtest")
109 assert_equal(batch
[1]["result"]["walletname"], "w1")
111 if __name__
== '__main__':
112 MultiWalletTest().main()