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
11 from test_framework
.test_framework
import BitcoinTestFramework
12 from test_framework
.util
import assert_equal
, assert_raises_jsonrpc
14 class MultiWalletTest(BitcoinTestFramework
):
18 self
.setup_clean_chain
= True
20 self
.extra_args
= [['-wallet=w1', '-wallet=w2', '-wallet=w3']]
25 # should not initialize if there are duplicate wallets
26 self
.assert_start_raises_init_error(0, self
.options
.tmpdir
, ['-wallet=w1', '-wallet=w1'], 'Error loading wallet w1. Duplicate -wallet filename specified.')
28 # should not initialize if wallet file is a directory
29 os
.mkdir(os
.path
.join(self
.options
.tmpdir
, 'node0', 'regtest', 'w11'))
30 self
.assert_start_raises_init_error(0, self
.options
.tmpdir
, ['-wallet=w11'], 'Error loading wallet w11. -wallet filename must be a regular file.')
32 # should not initialize if wallet file is a symlink
33 os
.symlink(os
.path
.join(self
.options
.tmpdir
, 'node0', 'regtest', 'w1'), os
.path
.join(self
.options
.tmpdir
, 'node0', 'regtest', 'w12'))
34 self
.assert_start_raises_init_error(0, self
.options
.tmpdir
, ['-wallet=w12'], 'Error loading wallet w12. -wallet filename must be a regular file.')
36 self
.nodes
[0] = self
.start_node(0, self
.options
.tmpdir
, self
.extra_args
[0])
38 w1
= self
.nodes
[0].get_wallet_rpc("w1")
39 w2
= self
.nodes
[0].get_wallet_rpc("w2")
40 w3
= self
.nodes
[0].get_wallet_rpc("w3")
41 wallet_bad
= self
.nodes
[0].get_wallet_rpc("bad")
45 # accessing invalid wallet fails
46 assert_raises_jsonrpc(-18, "Requested wallet does not exist or is not loaded", wallet_bad
.getwalletinfo
)
48 # accessing wallet RPC without using wallet endpoint fails
49 assert_raises_jsonrpc(-19, "Wallet file not specified", self
.nodes
[0].getwalletinfo
)
51 # check w1 wallet balance
52 w1_info
= w1
.getwalletinfo()
53 assert_equal(w1_info
['immature_balance'], 50)
54 w1_name
= w1_info
['walletname']
55 assert_equal(w1_name
, "w1")
57 # check w2 wallet balance
58 w2_info
= w2
.getwalletinfo()
59 assert_equal(w2_info
['immature_balance'], 0)
60 w2_name
= w2_info
['walletname']
61 assert_equal(w2_name
, "w2")
63 w3_name
= w3
.getwalletinfo()['walletname']
64 assert_equal(w3_name
, "w3")
66 assert_equal({"w1", "w2", "w3"}, {w1_name
, w2_name
, w3_name
})
69 assert_equal(w1
.getbalance(), 100)
70 assert_equal(w2
.getbalance(), 0)
71 assert_equal(w3
.getbalance(), 0)
73 w1
.sendtoaddress(w2
.getnewaddress(), 1)
74 w1
.sendtoaddress(w3
.getnewaddress(), 2)
76 assert_equal(w2
.getbalance(), 1)
77 assert_equal(w3
.getbalance(), 2)
79 if __name__
== '__main__':
80 MultiWalletTest().main()