Changeset View
Changeset View
Standalone View
Standalone View
test/functional/multiwallet.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # Copyright (c) 2017 The Bitcoin Core developers | # Copyright (c) 2017 The Bitcoin Core developers | ||||
| # Distributed under the MIT software license, see the accompanying | # Distributed under the MIT software license, see the accompanying | ||||
| # file COPYING or http://www.opensource.org/licenses/mit-license.php. | # file COPYING or http://www.opensource.org/licenses/mit-license.php. | ||||
| """Test multiwallet. | """Test multiwallet. | ||||
| Verify that a bitcoind node can load multiple wallet files | Verify that a bitcoind node can load multiple wallet files | ||||
| """ | """ | ||||
| import os | import os | ||||
| from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
| from test_framework.util import * | from test_framework.util import assert_equal, assert_raises_jsonrpc | ||||
| class MultiWalletTest(BitcoinTestFramework): | class MultiWalletTest(BitcoinTestFramework): | ||||
| def __init__(self): | def __init__(self): | ||||
| super().__init__() | super().__init__() | ||||
| self.setup_clean_chain = True | self.setup_clean_chain = True | ||||
| self.num_nodes = 1 | self.num_nodes = 1 | ||||
| self.extra_args = [['-wallet=w1', '-wallet=w2', '-wallet=w3']] | self.extra_args = [['-wallet=w1', '-wallet=w2', '-wallet=w3']] | ||||
| def run_test(self): | def run_test(self): | ||||
| self.stop_node(0) | self.stop_node(0) | ||||
| # should not initialize if there are duplicate wallets | # should not initialize if there are duplicate wallets | ||||
| self.assert_start_raises_init_error(0, self.options.tmpdir, [ | self.assert_start_raises_init_error(0, self.options.tmpdir, [ | ||||
| '-wallet=w1', '-wallet=w1'], 'Error loading wallet w1. Duplicate -wallet filename specified.') | '-wallet=w1', '-wallet=w1'], 'Error loading wallet w1. Duplicate -wallet filename specified.') | ||||
| # should not initialize if wallet file is a directory | # should not initialize if wallet file is a directory | ||||
| os.mkdir(os.path.join(self.options.tmpdir, 'node0', 'regtest', 'w11')) | os.mkdir(os.path.join(self.options.tmpdir, 'node0', 'regtest', 'w11')) | ||||
| self.assert_start_raises_init_error(0, self.options.tmpdir, [ | self.assert_start_raises_init_error(0, self.options.tmpdir, [ | ||||
| '-wallet=w11'], 'Error loading wallet w11. -wallet filename must be a regular file.') | '-wallet=w11'], 'Error loading wallet w11. -wallet filename must be a regular file.') | ||||
| # should not initialize if wallet file is a symlink | # should not initialize if wallet file is a symlink | ||||
| os.symlink(os.path.join(self.options.tmpdir, 'node0', 'regtest', 'w1'), | os.symlink(os.path.join(self.options.tmpdir, 'node0', 'regtest', 'w1'), | ||||
| os.path.join(self.options.tmpdir, 'node0', 'regtest', 'w12')) | os.path.join(self.options.tmpdir, 'node0', 'regtest', 'w12')) | ||||
| self.assert_start_raises_init_error(0, self.options.tmpdir, [ | self.assert_start_raises_init_error(0, self.options.tmpdir, [ | ||||
| '-wallet=w12'], 'Error loading wallet w12. -wallet filename must be a regular file.') | '-wallet=w12'], 'Error loading wallet w12. -wallet filename must be a regular file.') | ||||
| self.nodes[0] = self.start_node( | self.nodes[0] = self.start_node( | ||||
| 0, self.options.tmpdir, self.extra_args[0]) | 0, self.options.tmpdir, self.extra_args[0]) | ||||
| w1 = self.nodes[0] / "wallet/w1" | w1 = self.nodes[0] / "wallet/w1" | ||||
| w1.generate(1) | w1.generate(1) | ||||
| # accessing invalid wallet fails | # accessing invalid wallet fails | ||||
| assert_raises_jsonrpc(-18, "Requested wallet does not exist or is not loaded", | assert_raises_jsonrpc(-18, "Requested wallet does not exist or is not loaded", | ||||
| (self.nodes[0] / "wallet/bad").getwalletinfo) | (self.nodes[0] / "wallet/bad").getwalletinfo) | ||||
| # accessing wallet RPC without using wallet endpoint fails | # accessing wallet RPC without using wallet endpoint fails | ||||
| assert_raises_jsonrpc(-19, "Method not found", | assert_raises_jsonrpc(-19, "Wallet file not specified", | ||||
| self.nodes[0].getwalletinfo) | self.nodes[0].getwalletinfo) | ||||
| # check w1 wallet balance | # check w1 wallet balance | ||||
| walletinfo = w1.getwalletinfo() | w1_info = w1.getwalletinfo() | ||||
| assert_equal(walletinfo['immature_balance'], 50) | assert_equal(w1_info['immature_balance'], 50) | ||||
| w1_name = w1_info['walletname'] | |||||
| assert_equal(w1_name, "w1") | |||||
| # check w1 wallet balance | # check w1 wallet balance | ||||
| w2 = self.nodes[0] / "wallet/w2" | w2 = self.nodes[0] / "wallet/w2" | ||||
| walletinfo = w2.getwalletinfo() | w2_info = w2.getwalletinfo() | ||||
| assert_equal(walletinfo['immature_balance'], 0) | assert_equal(w2_info['immature_balance'], 0) | ||||
| w2_name = w2_info['walletname'] | |||||
| assert_equal(w2_name, "w2") | |||||
| w3 = self.nodes[0] / "wallet/w3" | w3 = self.nodes[0] / "wallet/w3" | ||||
| w3_name = w3.getwalletinfo()['walletname'] | |||||
| assert_equal(w3_name, "w3") | |||||
| assert_equal({"w1", "w2", "w3"}, {w1_name, w2_name, w3_name}) | |||||
| w1.generate(101) | w1.generate(101) | ||||
| assert_equal(w1.getbalance(), 100) | assert_equal(w1.getbalance(), 100) | ||||
| assert_equal(w2.getbalance(), 0) | assert_equal(w2.getbalance(), 0) | ||||
| assert_equal(w3.getbalance(), 0) | assert_equal(w3.getbalance(), 0) | ||||
| w1.sendtoaddress(w2.getnewaddress(), 1) | w1.sendtoaddress(w2.getnewaddress(), 1) | ||||
| w1.sendtoaddress(w3.getnewaddress(), 2) | w1.sendtoaddress(w3.getnewaddress(), 2) | ||||
| w1.generate(1) | w1.generate(1) | ||||
| assert_equal(w2.getbalance(), 1) | assert_equal(w2.getbalance(), 1) | ||||
| assert_equal(w3.getbalance(), 2) | assert_equal(w3.getbalance(), 2) | ||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| MultiWalletTest().main() | MultiWalletTest().main() | ||||