test: Remove RPCOverloadWrapper
With the legacy wallet now removed, these overloads are no longer needed.
This commit is contained in:
1 parent
4d32c19516
commit
b104d44227
2 files changed
+7
-14
No files matched your search
@@ -208,10 +208,10 @@ class TestNode():
|
|||||||
def __getattr__(self, name):
|
def __getattr__(self, name):
|
||||||
"""Dispatches any unrecognised messages to the RPC connection or a CLI instance."""
|
"""Dispatches any unrecognised messages to the RPC connection or a CLI instance."""
|
||||||
if self.use_cli:
|
if self.use_cli:
|
||||||
return getattr(RPCOverloadWrapper(self.cli), name)
|
return getattr(self.cli, name)
|
||||||
else:
|
else:
|
||||||
assert self.rpc_connected and self.rpc is not None, self._node_msg("Error: no RPC connection")
|
assert self.rpc_connected and self.rpc is not None, self._node_msg("Error: no RPC connection")
|
||||||
return getattr(RPCOverloadWrapper(self.rpc), name)
|
return getattr(self.rpc, name)
|
||||||
|
|
||||||
def start(self, extra_args=None, *, cwd=None, stdout=None, stderr=None, env=None, **kwargs):
|
def start(self, extra_args=None, *, cwd=None, stdout=None, stderr=None, env=None, **kwargs):
|
||||||
"""Start the node."""
|
"""Start the node."""
|
||||||
@@ -373,11 +373,11 @@ class TestNode():
|
|||||||
|
|
||||||
def get_wallet_rpc(self, wallet_name):
|
def get_wallet_rpc(self, wallet_name):
|
||||||
if self.use_cli:
|
if self.use_cli:
|
||||||
return RPCOverloadWrapper(self.cli("-rpcwallet={}".format(wallet_name)))
|
return self.cli("-rpcwallet={}".format(wallet_name))
|
||||||
else:
|
else:
|
||||||
assert self.rpc_connected and self.rpc, self._node_msg("RPC not connected")
|
assert self.rpc_connected and self.rpc, self._node_msg("RPC not connected")
|
||||||
wallet_path = "wallet/{}".format(urllib.parse.quote(wallet_name))
|
wallet_path = "wallet/{}".format(urllib.parse.quote(wallet_name))
|
||||||
return RPCOverloadWrapper(self.rpc / wallet_path)
|
return self.rpc / wallet_path
|
||||||
|
|
||||||
def version_is_at_least(self, ver):
|
def version_is_at_least(self, ver):
|
||||||
return self.version is None or self.version >= ver
|
return self.version is None or self.version >= ver
|
||||||
@@ -922,10 +922,3 @@ class TestNodeCLI():
|
|||||||
return json.loads(cli_stdout, parse_float=decimal.Decimal)
|
return json.loads(cli_stdout, parse_float=decimal.Decimal)
|
||||||
except (json.JSONDecodeError, decimal.InvalidOperation):
|
except (json.JSONDecodeError, decimal.InvalidOperation):
|
||||||
return cli_stdout.rstrip("\n")
|
return cli_stdout.rstrip("\n")
|
||||||
|
|
||||||
class RPCOverloadWrapper():
|
|
||||||
def __init__(self, rpc):
|
|
||||||
self.rpc = rpc
|
|
||||||
|
|
||||||
def __getattr__(self, name):
|
|
||||||
return getattr(self.rpc, name)
|
|
||||||
@@ -726,7 +726,7 @@ class WalletMigrationTest(BitcoinTestFramework):
|
|||||||
send_to_script(script=script_sh_pkh, amount=2)
|
send_to_script(script=script_sh_pkh, amount=2)
|
||||||
|
|
||||||
# Import script and check balance
|
# Import script and check balance
|
||||||
wallet.rpc.importaddress(address=script_pkh.hex(), label="raw_spk", rescan=True, p2sh=True)
|
wallet.importaddress(address=script_pkh.hex(), label="raw_spk", rescan=True, p2sh=True)
|
||||||
assert_equal(wallet.getbalances()['watchonly']['trusted'], 2)
|
assert_equal(wallet.getbalances()['watchonly']['trusted'], 2)
|
||||||
|
|
||||||
# Craft wsh(pkh(key)) and send coins to it
|
# Craft wsh(pkh(key)) and send coins to it
|
||||||
@@ -735,7 +735,7 @@ class WalletMigrationTest(BitcoinTestFramework):
|
|||||||
send_to_script(script=script_wsh_pkh, amount=3)
|
send_to_script(script=script_wsh_pkh, amount=3)
|
||||||
|
|
||||||
# Import script and check balance
|
# Import script and check balance
|
||||||
wallet.rpc.importaddress(address=script_wsh_pkh.hex(), label="raw_spk2", rescan=True, p2sh=False)
|
wallet.importaddress(address=script_wsh_pkh.hex(), label="raw_spk2", rescan=True, p2sh=False)
|
||||||
assert_equal(wallet.getbalances()['watchonly']['trusted'], 5)
|
assert_equal(wallet.getbalances()['watchonly']['trusted'], 5)
|
||||||
|
|
||||||
# Import sh(pkh()) script, by using importaddress(), with the p2sh flag enabled.
|
# Import sh(pkh()) script, by using importaddress(), with the p2sh flag enabled.
|
||||||
@@ -751,7 +751,7 @@ class WalletMigrationTest(BitcoinTestFramework):
|
|||||||
# Note: 'importaddress()' will add two scripts, a valid one sh(pkh()) and an invalid one 'sh(sh(pkh()))'.
|
# Note: 'importaddress()' will add two scripts, a valid one sh(pkh()) and an invalid one 'sh(sh(pkh()))'.
|
||||||
# Both of them will be stored with the same addressbook label. And only the latter one should
|
# Both of them will be stored with the same addressbook label. And only the latter one should
|
||||||
# be discarded during migration. The first one must be migrated.
|
# be discarded during migration. The first one must be migrated.
|
||||||
wallet.rpc.importaddress(address=script_sh_pkh.hex(), label=label_sh_pkh, rescan=False, p2sh=True)
|
wallet.importaddress(address=script_sh_pkh.hex(), label=label_sh_pkh, rescan=False, p2sh=True)
|
||||||
|
|
||||||
# Migrate wallet and re-check balance
|
# Migrate wallet and re-check balance
|
||||||
info_migration, wallet = self.migrate_and_get_rpc("raw_p2sh")
|
info_migration, wallet = self.migrate_and_get_rpc("raw_p2sh")
|
||||||
|
|||||||
Reference in new issue
Block a user