2017-07-18 17:15:46 +02:00
#!/usr/bin/env python3
# Copyright (c) 2017 The Bitcoin Core developers
# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
2017-07-21 15:34:11 +02:00
""" Test multiwallet.
2020-06-11 10:39:04 +02:00
Verify that a dashd node can load multiple wallet files
2017-07-21 15:34:11 +02:00
"""
2022-04-14 07:07:16 +02:00
from threading import Thread
2020-04-17 13:55:36 +02:00
from decimal import Decimal
2017-07-28 12:42:28 +02:00
import os
2017-10-19 18:12:59 +02:00
import shutil
2018-10-24 16:41:18 +02:00
import time
2017-07-28 12:42:28 +02:00
2022-04-14 07:07:16 +02:00
from test_framework . authproxy import JSONRPCException
2017-07-18 17:15:46 +02:00
from test_framework . test_framework import BitcoinTestFramework
2021-04-08 22:29:01 +02:00
from test_framework . test_node import ErrorMatch
2018-03-22 11:04:37 +01:00
from test_framework . util import (
assert_equal ,
assert_raises_rpc_error ,
2022-04-14 07:07:16 +02:00
get_rpc_proxy ,
2018-03-22 11:04:37 +01:00
)
2021-12-12 14:27:02 +01:00
FEATURE_LATEST = 120200
2022-04-14 07:07:16 +02:00
got_loading_error = False
def test_load_unload ( node , name ) :
global got_loading_error
for i in range ( 10 ) :
if got_loading_error :
return
try :
node . loadwallet ( name )
node . unloadwallet ( name )
except JSONRPCException as e :
if e . error [ ' code ' ] == - 4 and ' Wallet already being loading ' in e . error [ ' message ' ] :
got_loading_error = True
return
2017-07-18 17:15:46 +02:00
class MultiWalletTest ( BitcoinTestFramework ) :
2017-09-01 18:47:13 +02:00
def set_test_params ( self ) :
2017-07-18 17:15:46 +02:00
self . setup_clean_chain = True
2018-01-16 10:54:13 +01:00
self . num_nodes = 2
2020-03-15 15:14:34 +01:00
self . rpc_timeout = 120
2022-11-30 20:23:31 +01:00
self . extra_args = [ [ " -wallet= " ] , [ " -wallet= " ] ]
2017-07-18 17:15:46 +02:00
2018-09-13 12:33:15 +02:00
def skip_test_if_missing_module ( self ) :
self . skip_if_no_wallet ( )
2021-12-12 14:27:02 +01:00
def add_options ( self , parser ) :
parser . add_argument (
' --data_wallets_dir ' ,
default = os . path . join ( os . path . dirname ( os . path . realpath ( __file__ ) ) , ' data/wallets/ ' ) ,
help = ' Test data with wallet directories (default: %(default)s ) ' ,
)
2017-07-18 17:15:46 +02:00
def run_test ( self ) :
2018-01-12 23:24:36 +01:00
node = self . nodes [ 0 ]
2021-01-22 15:58:07 +01:00
data_dir = lambda * p : os . path . join ( node . datadir , self . chain , * p )
2018-01-12 23:24:36 +01:00
wallet_dir = lambda * p : data_dir ( ' wallets ' , * p )
wallet = lambda name : node . get_wallet_rpc ( name )
2018-08-07 13:27:21 +02:00
def wallet_file ( name ) :
if os . path . isdir ( wallet_dir ( name ) ) :
2022-11-08 18:41:44 +01:00
return wallet_dir ( name , " wallet.dat " )
2018-08-07 13:27:21 +02:00
return wallet_dir ( name )
2022-11-08 18:41:44 +01:00
assert_equal ( self . nodes [ 0 ] . listwalletdir ( ) , { ' wallets ' : [ { ' name ' : ' ' } ] } )
2021-08-12 08:02:09 +02:00
2018-03-07 17:05:08 +01:00
# check wallet.dat is created
2018-01-16 10:54:13 +01:00
self . stop_nodes ( )
2022-11-08 18:41:44 +01:00
assert_equal ( os . path . isfile ( wallet_dir ( ' wallet.dat ' ) ) , True )
2018-03-07 17:05:08 +01:00
# create symlink to verify wallet directory path can be referenced
# through symlink
2018-09-24 22:10:13 +02:00
if os . name != ' nt ' :
os . mkdir ( wallet_dir ( ' w7 ' ) )
os . symlink ( ' w7 ' , wallet_dir ( ' w7_symlink ' ) )
2018-03-07 17:05:08 +01:00
# rename wallet.dat to make sure plain wallet file paths (as opposed to
# directory paths) can be loaded
2022-11-08 18:41:44 +01:00
os . rename ( wallet_dir ( " wallet.dat " ) , wallet_dir ( " w8 " ) )
2018-03-07 17:05:08 +01:00
2018-08-07 13:27:21 +02:00
# create another dummy wallet for use in testing backups later
2022-11-30 20:23:31 +01:00
self . start_node ( 0 , [ " -wallet= " ] )
2018-08-07 13:27:21 +02:00
self . stop_nodes ( )
empty_wallet = os . path . join ( self . options . tmpdir , ' empty.dat ' )
2022-11-08 18:41:44 +01:00
os . rename ( wallet_dir ( " wallet.dat " ) , empty_wallet )
2018-08-07 13:27:21 +02:00
2018-03-07 17:05:08 +01:00
# restart node with a mix of wallet names:
# w1, w2, w3 - to verify new wallets created when non-existing paths specified
# w - to verify wallet name matching works when one wallet path is prefix of another
# sub/w5 - to verify relative wallet path is created correctly
# extern/w6 - to verify absolute wallet path is created correctly
# w7_symlink - to verify symlinked wallet path is initialized correctly
# w8 - to verify existing wallet file is loaded correctly
# '' - to verify default wallet file is created correctly
2022-11-08 18:41:44 +01:00
wallet_names = [ ' w1 ' , ' w2 ' , ' w3 ' , ' w ' , ' sub/w5 ' , os . path . join ( self . options . tmpdir , ' extern/w6 ' ) , ' w7_symlink ' , ' w8 ' , ' ' ]
2018-09-24 22:10:13 +02:00
if os . name == ' nt ' :
wallet_names . remove ( ' w7_symlink ' )
2022-11-08 18:41:44 +01:00
extra_args = [ ' -wallet= {} ' . format ( n ) for n in wallet_names ]
2022-11-03 02:50:40 +01:00
self . start_node ( 0 , extra_args )
2022-11-08 18:41:44 +01:00
assert_equal ( sorted ( map ( lambda w : w [ ' name ' ] , self . nodes [ 0 ] . listwalletdir ( ) [ ' wallets ' ] ) ) , [ ' ' , os . path . join ( ' sub ' , ' w5 ' ) , ' w ' , ' w1 ' , ' w2 ' , ' w3 ' , ' w7 ' , ' w7_symlink ' , ' w8 ' ] )
2021-08-12 08:02:09 +02:00
2018-03-07 17:05:08 +01:00
assert_equal ( set ( node . listwallets ( ) ) , set ( wallet_names ) )
# check that all requested wallets were created
self . stop_node ( 0 )
for wallet_name in wallet_names :
2018-08-07 13:27:21 +02:00
assert_equal ( os . path . isfile ( wallet_file ( wallet_name ) ) , True )
2018-03-07 17:05:08 +01:00
2022-11-03 02:50:40 +01:00
# should not initialize if wallet path can't be created
exp_stderr = " boost::filesystem::create_directory: "
2022-11-08 18:41:44 +01:00
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -wallet=wallet.dat/bad ' ] , exp_stderr , match = ErrorMatch . PARTIAL_REGEX )
2022-11-03 02:50:40 +01:00
2018-03-22 10:18:33 +01:00
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -walletdir=wallets ' ] , ' Error: Specified -walletdir " wallets " does not exist ' )
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -walletdir=wallets ' ] , ' Error: Specified -walletdir " wallets " is a relative path ' , cwd = data_dir ( ) )
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -walletdir=debug.log ' ] , ' Error: Specified -walletdir " debug.log " is not a directory ' , cwd = data_dir ( ) )
2018-01-19 17:44:27 +01:00
2017-07-28 12:42:28 +02:00
# should not initialize if there are duplicate wallets
2018-03-22 10:18:33 +01:00
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -wallet=w1 ' , ' -wallet=w1 ' ] , ' Error: Error loading wallet w1. Duplicate -wallet filename specified. ' )
2017-07-28 12:42:28 +02:00
2017-10-19 18:12:59 +02:00
# should not initialize if one wallet is a copy of another
2018-03-07 17:05:08 +01:00
shutil . copyfile ( wallet_dir ( ' w8 ' ) , wallet_dir ( ' w8_copy ' ) )
2021-09-10 02:45:16 +02:00
exp_stderr = r " Can ' t open database w8_copy \ (duplicates fileid \ w+ from w8 \ ) "
2021-04-08 22:29:01 +02:00
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -wallet=w8 ' , ' -wallet=w8_copy ' ] , exp_stderr , match = ErrorMatch . PARTIAL_REGEX )
2017-10-19 18:12:59 +02:00
2017-07-28 12:42:28 +02:00
# should not initialize if wallet file is a symlink
2018-09-24 22:10:13 +02:00
if os . name != ' nt ' :
os . symlink ( ' w8 ' , wallet_dir ( ' w8_symlink ' ) )
2021-09-10 02:45:16 +02:00
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -wallet=w8_symlink ' ] , r ' Error: Invalid -wallet path \' w8_symlink \' \ . .* ' , match = ErrorMatch . FULL_REGEX )
2017-07-28 12:42:28 +02:00
2017-11-18 14:32:50 +01:00
# should not initialize if the specified walletdir does not exist
2018-03-22 10:18:33 +01:00
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -walletdir=bad ' ] , ' Error: Specified -walletdir " bad " does not exist ' )
2017-12-20 23:37:52 +01:00
# should not initialize if the specified walletdir is not a directory
2018-01-12 23:24:36 +01:00
not_a_dir = wallet_dir ( ' notadir ' )
2018-06-16 15:21:01 +02:00
open ( not_a_dir , ' a ' , encoding = " utf8 " ) . close ( )
2021-04-08 22:29:01 +02:00
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -walletdir= ' + not_a_dir ] , ' Error: Specified -walletdir " ' + not_a_dir + ' " is not a directory ' )
2017-11-18 14:32:50 +01:00
2018-04-23 20:23:11 +02:00
self . log . info ( " Do not allow -upgradewallet with multiwallet " )
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -upgradewallet ' , ' -wallet=w1 ' , ' -wallet=w2 ' ] , " Error: -upgradewallet is only allowed with a single wallet file " )
self . nodes [ 0 ] . assert_start_raises_init_error ( [ ' -upgradewallet=1 ' , ' -wallet=w1 ' , ' -wallet=w2 ' ] , " Error: -upgradewallet is only allowed with a single wallet file " )
2017-11-18 14:32:50 +01:00
# if wallets/ doesn't exist, datadir should be the default wallet dir
2018-01-12 23:24:36 +01:00
wallet_dir2 = data_dir ( ' walletdir ' )
os . rename ( wallet_dir ( ) , wallet_dir2 )
2022-11-08 18:41:44 +01:00
self . start_node ( 0 , [ ' -wallet=w4 ' , ' -wallet=w5 ' ] )
2018-01-12 23:24:36 +01:00
assert_equal ( set ( node . listwallets ( ) ) , { " w4 " , " w5 " } )
w5 = wallet ( " w5 " )
2018-10-24 00:11:26 +02:00
node . generatetoaddress ( nblocks = 1 , address = w5 . getnewaddress ( ) )
2017-11-18 14:32:50 +01:00
# now if wallets/ exists again, but the rootdir is specified as the walletdir, w4 and w5 should still be loaded
2018-01-12 23:24:36 +01:00
os . rename ( wallet_dir2 , wallet_dir ( ) )
2022-11-08 18:41:44 +01:00
self . restart_node ( 0 , [ ' -wallet=w4 ' , ' -wallet=w5 ' , ' -walletdir= ' + data_dir ( ) ] )
2018-01-12 23:24:36 +01:00
assert_equal ( set ( node . listwallets ( ) ) , { " w4 " , " w5 " } )
w5 = wallet ( " w5 " )
2017-11-18 14:32:50 +01:00
w5_info = w5 . getwalletinfo ( )
2020-02-06 16:45:57 +01:00
assert_equal ( w5_info [ ' immature_balance ' ] , 500 )
2017-11-18 14:32:50 +01:00
2018-01-16 10:54:13 +01:00
competing_wallet_dir = os . path . join ( self . options . tmpdir , ' competing_walletdir ' )
os . mkdir ( competing_wallet_dir )
2022-11-30 20:23:31 +01:00
self . restart_node ( 0 , [ ' -walletdir= ' + competing_wallet_dir , ' -wallet= ' ] )
2021-09-10 02:45:16 +02:00
exp_stderr = r " Error: Error initializing wallet database environment \" \ S+competing_walletdir \" ! "
2021-04-08 22:29:01 +02:00
self . nodes [ 1 ] . assert_start_raises_init_error ( [ ' -walletdir= ' + competing_wallet_dir ] , exp_stderr , match = ErrorMatch . PARTIAL_REGEX )
2017-11-18 14:32:50 +01:00
2022-11-03 02:50:40 +01:00
self . restart_node ( 0 , extra_args )
2017-07-28 12:42:28 +02:00
2022-11-08 18:41:44 +01:00
assert_equal ( sorted ( map ( lambda w : w [ ' name ' ] , self . nodes [ 0 ] . listwalletdir ( ) [ ' wallets ' ] ) ) , [ ' ' , os . path . join ( ' sub ' , ' w5 ' ) , ' w ' , ' w1 ' , ' w2 ' , ' w3 ' , ' w7 ' , ' w7_symlink ' , ' w8 ' , ' w8_copy ' ] )
2021-08-12 08:02:09 +02:00
2018-03-07 17:05:08 +01:00
wallets = [ wallet ( w ) for w in wallet_names ]
2018-01-12 23:24:36 +01:00
wallet_bad = wallet ( " bad " )
2017-08-15 23:34:07 +02:00
2018-03-07 17:05:08 +01:00
# check wallet names and balances
2018-10-24 00:11:26 +02:00
node . generatetoaddress ( nblocks = 1 , address = wallets [ 0 ] . getnewaddress ( ) )
2018-03-07 17:05:08 +01:00
for wallet_name , wallet in zip ( wallet_names , wallets ) :
info = wallet . getwalletinfo ( )
2020-04-15 01:27:49 +02:00
assert_equal ( info [ ' immature_balance ' ] , 500 if wallet is wallets [ 0 ] else 0 )
2018-03-07 17:05:08 +01:00
assert_equal ( info [ ' walletname ' ] , wallet_name )
2017-07-18 17:15:46 +02:00
2017-07-27 17:51:14 +02:00
# accessing invalid wallet fails
2019-09-25 11:34:51 +02:00
assert_raises_rpc_error ( - 18 , " Requested wallet does not exist or is not loaded " , wallet_bad . getwalletinfo )
2017-07-27 17:51:14 +02:00
2017-07-21 15:34:11 +02:00
# accessing wallet RPC without using wallet endpoint fails
2018-01-12 23:24:36 +01:00
assert_raises_rpc_error ( - 19 , " Wallet file not specified " , node . getwalletinfo )
2017-07-18 17:15:46 +02:00
2018-03-07 17:05:08 +01:00
w1 , w2 , w3 , w4 , * _ = wallets
2018-10-24 00:11:26 +02:00
node . generatetoaddress ( nblocks = 101 , address = w1 . getnewaddress ( ) )
2019-08-03 17:56:18 +02:00
assert_equal ( w1 . getbalance ( ) , 1000 )
2017-07-18 17:15:46 +02:00
assert_equal ( w2 . getbalance ( ) , 0 )
assert_equal ( w3 . getbalance ( ) , 0 )
2017-11-22 17:31:32 +01:00
assert_equal ( w4 . getbalance ( ) , 0 )
2017-07-18 17:15:46 +02:00
w1 . sendtoaddress ( w2 . getnewaddress ( ) , 1 )
w1 . sendtoaddress ( w3 . getnewaddress ( ) , 2 )
2017-11-22 17:31:32 +01:00
w1 . sendtoaddress ( w4 . getnewaddress ( ) , 3 )
2018-10-24 00:11:26 +02:00
node . generatetoaddress ( nblocks = 1 , address = w1 . getnewaddress ( ) )
2017-07-18 17:15:46 +02:00
assert_equal ( w2 . getbalance ( ) , 1 )
assert_equal ( w3 . getbalance ( ) , 2 )
2017-11-22 17:31:32 +01:00
assert_equal ( w4 . getbalance ( ) , 3 )
2017-07-18 17:15:46 +02:00
2017-10-12 14:53:33 +02:00
batch = w1 . batch ( [ w1 . getblockchaininfo . get_request ( ) , w1 . getwalletinfo . get_request ( ) ] )
2021-01-22 15:58:07 +01:00
assert_equal ( batch [ 0 ] [ " result " ] [ " chain " ] , self . chain )
2017-10-12 14:53:33 +02:00
assert_equal ( batch [ 1 ] [ " result " ] [ " walletname " ] , " w1 " )
2021-04-09 10:26:57 +02:00
self . log . info ( ' Check for per-wallet settxfee call ' )
assert_equal ( w1 . getwalletinfo ( ) [ ' paytxfee ' ] , 0 )
assert_equal ( w2 . getwalletinfo ( ) [ ' paytxfee ' ] , 0 )
2020-04-17 13:55:36 +02:00
w2 . settxfee ( 0.001 )
2021-04-09 10:26:57 +02:00
assert_equal ( w1 . getwalletinfo ( ) [ ' paytxfee ' ] , 0 )
2020-04-17 13:55:36 +02:00
assert_equal ( w2 . getwalletinfo ( ) [ ' paytxfee ' ] , Decimal ( ' 0.00100000 ' ) )
2018-04-19 23:42:40 +02:00
self . log . info ( " Test dynamic wallet loading " )
self . restart_node ( 0 , [ ' -nowallet ' ] )
assert_equal ( node . listwallets ( ) , [ ] )
2020-10-08 15:06:57 +02:00
assert_raises_rpc_error ( - 18 , " No wallet is loaded. Load a wallet using loadwallet or create a new one with createwallet. (Note: A default wallet is no longer automatically created) " , node . getwalletinfo )
2018-04-19 23:42:40 +02:00
self . log . info ( " Load first wallet " )
loadwallet_name = node . loadwallet ( wallet_names [ 0 ] )
assert_equal ( loadwallet_name [ ' name ' ] , wallet_names [ 0 ] )
assert_equal ( node . listwallets ( ) , wallet_names [ 0 : 1 ] )
node . getwalletinfo ( )
w1 = node . get_wallet_rpc ( wallet_names [ 0 ] )
w1 . getwalletinfo ( )
self . log . info ( " Load second wallet " )
loadwallet_name = node . loadwallet ( wallet_names [ 1 ] )
assert_equal ( loadwallet_name [ ' name ' ] , wallet_names [ 1 ] )
assert_equal ( node . listwallets ( ) , wallet_names [ 0 : 2 ] )
assert_raises_rpc_error ( - 19 , " Wallet file not specified " , node . getwalletinfo )
w2 = node . get_wallet_rpc ( wallet_names [ 1 ] )
w2 . getwalletinfo ( )
2022-04-14 07:07:16 +02:00
self . log . info ( " Concurrent wallet loading " )
threads = [ ]
for _ in range ( 3 ) :
n = node . cli if self . options . usecli else get_rpc_proxy ( node . url , 1 , timeout = 600 , coveragedir = node . coverage_dir )
t = Thread ( target = test_load_unload , args = ( n , wallet_names [ 2 ] , ) )
t . start ( )
threads . append ( t )
for t in threads :
t . join ( )
global got_loading_error
assert_equal ( got_loading_error , True )
2018-04-19 23:42:40 +02:00
self . log . info ( " Load remaining wallets " )
for wallet_name in wallet_names [ 2 : ] :
loadwallet_name = self . nodes [ 0 ] . loadwallet ( wallet_name )
assert_equal ( loadwallet_name [ ' name ' ] , wallet_name )
assert_equal ( set ( self . nodes [ 0 ] . listwallets ( ) ) , set ( wallet_names ) )
# Fail to load if wallet doesn't exist
2022-10-17 04:33:10 +02:00
path = os . path . join ( self . options . tmpdir , " node0 " , " regtest " , " wallets " , " wallets " )
assert_raises_rpc_error ( - 18 , " Wallet file verification failed. Failed to load database path ' {} ' . Path does not exist. " . format ( path ) , self . nodes [ 0 ] . loadwallet , ' wallets ' )
2018-04-19 23:42:40 +02:00
# Fail to load duplicate wallets
2022-11-08 18:41:44 +01:00
path = os . path . join ( self . options . tmpdir , " node0 " , " regtest " , " wallets " , " w1 " , " wallet.dat " )
2022-10-17 04:33:10 +02:00
assert_raises_rpc_error ( - 4 , " Wallet file verification failed. Refusing to load database. Data file ' {} ' is already loaded. " . format ( path ) , self . nodes [ 0 ] . loadwallet , wallet_names [ 0 ] )
2018-04-19 23:42:40 +02:00
2018-11-20 15:15:51 +01:00
# Fail to load duplicate wallets by different ways (directory and filepath)
2022-11-08 18:41:44 +01:00
path = os . path . join ( self . options . tmpdir , " node0 " , " regtest " , " wallets " , " wallet.dat " )
assert_raises_rpc_error ( - 4 , " Wallet file verification failed. Refusing to load database. Data file ' {} ' is already loaded. " . format ( path ) , self . nodes [ 0 ] . loadwallet , ' wallet.dat ' )
2018-11-20 15:15:51 +01:00
2018-04-19 23:42:40 +02:00
# Fail to load if one wallet is a copy of another
2020-07-16 19:19:37 +02:00
assert_raises_rpc_error ( - 4 , " BerkeleyDatabase: Can ' t open database w8_copy (duplicates fileid " , self . nodes [ 0 ] . loadwallet , ' w8_copy ' )
2018-04-19 23:42:40 +02:00
2018-10-24 17:22:25 +02:00
# Fail to load if one wallet is a copy of another, test this twice to make sure that we don't re-introduce #14304
2020-07-16 19:19:37 +02:00
assert_raises_rpc_error ( - 4 , " BerkeleyDatabase: Can ' t open database w8_copy (duplicates fileid " , self . nodes [ 0 ] . loadwallet , ' w8_copy ' )
2018-10-24 17:22:25 +02:00
2018-04-19 23:42:40 +02:00
# Fail to load if wallet file is a symlink
2018-09-24 22:10:13 +02:00
if os . name != ' nt ' :
2022-04-07 06:43:16 +02:00
assert_raises_rpc_error ( - 4 , " Wallet file verification failed. Invalid -wallet path ' w8_symlink ' " , self . nodes [ 0 ] . loadwallet , ' w8_symlink ' )
2018-04-19 23:42:40 +02:00
2018-06-01 10:46:30 +02:00
self . log . info ( " Test dynamic wallet creation. " )
# Fail to create a wallet if it already exists.
2022-10-17 04:33:10 +02:00
path = os . path . join ( self . options . tmpdir , " node0 " , " regtest " , " wallets " , " w2 " )
assert_raises_rpc_error ( - 4 , " Failed to create database path ' {} ' . Database already exists. " . format ( path ) , self . nodes [ 0 ] . createwallet , ' w2 ' )
2018-06-01 10:46:30 +02:00
# Successfully create a wallet with a new name
loadwallet_name = self . nodes [ 0 ] . createwallet ( ' w9 ' )
assert_equal ( loadwallet_name [ ' name ' ] , ' w9 ' )
w9 = node . get_wallet_rpc ( ' w9 ' )
assert_equal ( w9 . getwalletinfo ( ) [ ' walletname ' ] , ' w9 ' )
assert ' w9 ' in self . nodes [ 0 ] . listwallets ( )
# Successfully create a wallet using a full path
new_wallet_dir = os . path . join ( self . options . tmpdir , ' new_walletdir ' )
new_wallet_name = os . path . join ( new_wallet_dir , ' w10 ' )
loadwallet_name = self . nodes [ 0 ] . createwallet ( new_wallet_name )
assert_equal ( loadwallet_name [ ' name ' ] , new_wallet_name )
w10 = node . get_wallet_rpc ( new_wallet_name )
assert_equal ( w10 . getwalletinfo ( ) [ ' walletname ' ] , new_wallet_name )
assert new_wallet_name in self . nodes [ 0 ] . listwallets ( )
2020-07-08 14:55:33 +02:00
# Fail to load if a directory is specified that doesn't contain a wallet
os . mkdir ( wallet_dir ( ' empty_wallet_dir ' ) )
2022-10-17 04:33:10 +02:00
path = os . path . join ( self . options . tmpdir , " node0 " , " regtest " , " wallets " , " empty_wallet_dir " )
assert_raises_rpc_error ( - 18 , " Wallet file verification failed. Failed to load database path ' {} ' . Data is not in recognized format. " . format ( path ) , self . nodes [ 0 ] . loadwallet , ' empty_wallet_dir ' )
2020-07-08 14:55:33 +02:00
2018-06-21 16:24:01 +02:00
self . log . info ( " Test dynamic wallet unloading " )
# Test `unloadwallet` errors
assert_raises_rpc_error ( - 1 , " JSON value is not a string as expected " , self . nodes [ 0 ] . unloadwallet )
assert_raises_rpc_error ( - 18 , " Requested wallet does not exist or is not loaded " , self . nodes [ 0 ] . unloadwallet , " dummy " )
assert_raises_rpc_error ( - 18 , " Requested wallet does not exist or is not loaded " , node . get_wallet_rpc ( " dummy " ) . unloadwallet )
assert_raises_rpc_error ( - 8 , " Cannot unload the requested wallet " , w1 . unloadwallet , " w2 " ) ,
# Successfully unload the specified wallet name
self . nodes [ 0 ] . unloadwallet ( " w1 " )
assert ' w1 ' not in self . nodes [ 0 ] . listwallets ( )
# Successfully unload the wallet referenced by the request endpoint
2018-10-24 16:41:18 +02:00
# Also ensure unload works during walletpassphrase timeout
w2 . encryptwallet ( ' test ' )
w2 . walletpassphrase ( ' test ' , 1 )
2018-06-21 16:24:01 +02:00
w2 . unloadwallet ( )
2018-10-24 16:41:18 +02:00
time . sleep ( 1.1 )
2018-06-21 16:24:01 +02:00
assert ' w2 ' not in self . nodes [ 0 ] . listwallets ( )
# Successfully unload all wallets
for wallet_name in self . nodes [ 0 ] . listwallets ( ) :
self . nodes [ 0 ] . unloadwallet ( wallet_name )
assert_equal ( self . nodes [ 0 ] . listwallets ( ) , [ ] )
2020-10-08 15:06:57 +02:00
assert_raises_rpc_error ( - 18 , " No wallet is loaded. Load a wallet using loadwallet or create a new one with createwallet. (Note: A default wallet is no longer automatically created) " , self . nodes [ 0 ] . getwalletinfo )
2018-06-21 16:24:01 +02:00
# Successfully load a previously unloaded wallet
self . nodes [ 0 ] . loadwallet ( ' w1 ' )
assert_equal ( self . nodes [ 0 ] . listwallets ( ) , [ ' w1 ' ] )
assert_equal ( w1 . getwalletinfo ( ) [ ' walletname ' ] , ' w1 ' )
2022-11-08 18:41:44 +01:00
assert_equal ( sorted ( map ( lambda w : w [ ' name ' ] , self . nodes [ 0 ] . listwalletdir ( ) [ ' wallets ' ] ) ) , [ ' ' , os . path . join ( ' sub ' , ' w5 ' ) , ' w ' , ' w1 ' , ' w2 ' , ' w3 ' , ' w7 ' , ' w7_symlink ' , ' w8 ' , ' w8_copy ' , ' w9 ' ] )
2021-08-12 08:02:09 +02:00
2018-08-07 13:27:21 +02:00
# Test backing up and restoring wallets
self . log . info ( " Test wallet backup " )
self . restart_node ( 0 , [ ' -nowallet ' ] )
for wallet_name in wallet_names :
self . nodes [ 0 ] . loadwallet ( wallet_name )
for wallet_name in wallet_names :
rpc = self . nodes [ 0 ] . get_wallet_rpc ( wallet_name )
addr = rpc . getnewaddress ( )
backup = os . path . join ( self . options . tmpdir , ' backup.dat ' )
rpc . backupwallet ( backup )
self . nodes [ 0 ] . unloadwallet ( wallet_name )
shutil . copyfile ( empty_wallet , wallet_file ( wallet_name ) )
self . nodes [ 0 ] . loadwallet ( wallet_name )
assert_equal ( rpc . getaddressinfo ( addr ) [ ' ismine ' ] , False )
self . nodes [ 0 ] . unloadwallet ( wallet_name )
shutil . copyfile ( backup , wallet_file ( wallet_name ) )
self . nodes [ 0 ] . loadwallet ( wallet_name )
assert_equal ( rpc . getaddressinfo ( addr ) [ ' ismine ' ] , True )
2019-02-05 22:09:24 +01:00
# Test .walletlock file is closed
self . start_node ( 1 )
wallet = os . path . join ( self . options . tmpdir , ' my_wallet ' )
self . nodes [ 0 ] . createwallet ( wallet )
assert_raises_rpc_error ( - 4 , " Error initializing wallet database environment " , self . nodes [ 1 ] . loadwallet , wallet )
self . nodes [ 0 ] . unloadwallet ( wallet )
self . nodes [ 1 ] . loadwallet ( wallet )
2021-12-12 14:27:02 +01:00
# Fail to load if wallet is downgraded
shutil . copytree ( os . path . join ( self . options . data_wallets_dir , ' high_minversion ' ) , wallet_dir ( ' high_minversion ' ) )
self . restart_node ( 0 , extra_args = [ ' -upgradewallet= {} ' . format ( FEATURE_LATEST ) ] )
assert { ' name ' : ' high_minversion ' } in self . nodes [ 0 ] . listwalletdir ( ) [ ' wallets ' ]
self . log . info ( " Fail -upgradewallet that results in downgrade " )
assert_raises_rpc_error (
- 4 ,
2022-03-24 21:07:40 +01:00
' Wallet loading failed. Error loading {} : Wallet requires newer version of {} ' . format (
wallet_dir ( ' high_minversion ' , ' wallet.dat ' ) , " Dash Core " ) ,
2021-12-12 14:27:02 +01:00
lambda : self . nodes [ 0 ] . loadwallet ( filename = ' high_minversion ' ) ,
)
2018-08-07 13:27:21 +02:00
2017-07-18 17:15:46 +02:00
if __name__ == ' __main__ ' :
MultiWalletTest ( ) . main ( )