text: #!/usr/bin/env python3
  # Copyright (c) 2018 The Bitcoin Core developers
  # Copyright (c) 2021 The Vivuscoin Core developers
  # Distributed under the MIT software license, see the accompanying
  # file COPYING or http://www.opensource.org/licenses/mit-license.php.
  """Useful util functions for testing the wallet"""
  from collections import namedtuple
  
  from test_framework.address import (
      key_to_p2pkh,
      key_to_p2sh_p2wpkh,
      key_to_p2wpkh,
      script_to_p2sh,
      script_to_p2sh_p2wsh,
      script_to_p2wsh,
  )
  from test_framework.script import (
      CScript,
      OP_0,
      OP_2,
      OP_3,
      OP_CHECKMULTISIG,
      OP_CHECKSIG,
      OP_DUP,
      OP_EQUAL,
      OP_EQUALVERIFY,
      OP_HASH160,
      hash160,
      sha256,
  )
  from test_framework.util import hex_str_to_bytes
  
  Key = namedtuple('Key', ['privkey',
                           'pubkey',
                           'p2pkh_script',
                           'p2pkh_addr',
                           'p2wpkh_script',
                           'p2wpkh_addr',
                           'p2sh_p2wpkh_script',
                           'p2sh_p2wpkh_redeem_script',
                           'p2sh_p2wpkh_addr'])
  
  Multisig = namedtuple('Multisig', ['privkeys',
                                     'pubkeys',
                                     'p2sh_script',
                                     'p2sh_addr',
                                     'redeem_script',
                                     'p2wsh_script',
                                     'p2wsh_addr',
                                     'p2sh_p2wsh_script',
                                     'p2sh_p2wsh_addr'])
  
  def get_key(node):
      """Generate a fresh key on node
  
      Returns a named tuple of privkey, pubkey and all address and scripts."""
      addr = node.getnewaddress()
      pubkey = node.getaddressinfo(addr)['pubkey']
      pkh = hash160(hex_str_to_bytes(pubkey))
      return Key(privkey=node.dumpprivkey(addr),
                 pubkey=pubkey,
                 p2pkh_script=CScript([OP_DUP, OP_HASH160, pkh, OP_EQUALVERIFY, OP_CHECKSIG]).hex(),
                 p2pkh_addr=key_to_p2pkh(pubkey),
                 p2wpkh_script=CScript([OP_0, pkh]).hex(),
                 p2wpkh_addr=key_to_p2wpkh(pubkey),
                 p2sh_p2wpkh_script=CScript([OP_HASH160, hash160(CScript([OP_0, pkh])), OP_EQUAL]).hex(),
                 p2sh_p2wpkh_redeem_script=CScript([OP_0, pkh]).hex(),
                 p2sh_p2wpkh_addr=key_to_p2sh_p2wpkh(pubkey))
  
  def get_multisig(node):
      """Generate a fresh 2-of-3 multisig on node
  
      Returns a named tuple of privkeys, pubkeys and all address and scripts."""
      addrs = []
      pubkeys = []
      for _ in range(3):
          addr = node.getaddressinfo(node.getnewaddress())
          addrs.append(addr['address'])
          pubkeys.append(addr['pubkey'])
      script_code = CScript([OP_2] + [hex_str_to_bytes(pubkey) for pubkey in pubkeys] + [OP_3, OP_CHECKMULTISIG])
      witness_script = CScript([OP_0, sha256(script_code)])
      return Multisig(privkeys=[node.dumpprivkey(addr) for addr in addrs],
                      pubkeys=pubkeys,
                      p2sh_script=CScript([OP_HASH160, hash160(script_code), OP_EQUAL]).hex(),
                      p2sh_addr=script_to_p2sh(script_code),
                      redeem_script=script_code.hex(),
                      p2wsh_script=witness_script.hex(),
                      p2wsh_addr=script_to_p2wsh(script_code),
                      p2sh_p2wsh_script=CScript([OP_HASH160, witness_script, OP_EQUAL]).hex(),
                      p2sh_p2wsh_addr=script_to_p2sh_p2wsh(script_code))
  
  def test_address(node, address, **kwargs):
      """Get address info for `address` and test whether the returned values are as expected."""
      addr_info = node.getaddressinfo(address)
      for key, value in kwargs.items():
          if value is None:
              if key in addr_info.keys():
                  raise AssertionError("key {} unexpectedly returned in getaddressinfo.".format(key))
          elif addr_info[key] != value:
              raise AssertionError("key {} value {} did not match expected value {}".format(key, addr_info[key], value))
  