Add python bech32 impl round-trip test

pull/16864/head
Gregory Sanders 5 years ago
parent fb4f5beb6e
commit ae0add8dfe

@ -64,7 +64,10 @@ from test_framework.util import (
assert_raises_rpc_error, assert_raises_rpc_error,
connect_nodes_bi, connect_nodes_bi,
) )
from test_framework.segwit_addr import (
encode,
decode,
)
class AddressTypeTest(BitcoinTestFramework): class AddressTypeTest(BitcoinTestFramework):
def set_test_params(self): def set_test_params(self):
@ -97,6 +100,13 @@ class AddressTypeTest(BitcoinTestFramework):
else: else:
return [self.nodes[i].getunconfirmedbalance() for i in range(4)] return [self.nodes[i].getunconfirmedbalance() for i in range(4)]
# Quick test of python bech32 implementation
def test_python_bech32(self, addr):
hrp = addr[:4]
assert_equal(hrp, "bcrt")
(witver, witprog) = decode(hrp, addr)
assert_equal(encode(hrp, witver, witprog), addr)
def test_address(self, node, address, multisig, typ): def test_address(self, node, address, multisig, typ):
"""Run sanity checks on an address.""" """Run sanity checks on an address."""
info = self.nodes[node].getaddressinfo(address) info = self.nodes[node].getaddressinfo(address)
@ -121,6 +131,7 @@ class AddressTypeTest(BitcoinTestFramework):
assert_equal(info['witness_version'], 0) assert_equal(info['witness_version'], 0)
assert_equal(len(info['witness_program']), 40) assert_equal(len(info['witness_program']), 40)
assert 'pubkey' in info assert 'pubkey' in info
self.test_python_bech32(info["address"])
elif typ == 'legacy': elif typ == 'legacy':
# P2SH-multisig # P2SH-multisig
assert info['isscript'] assert info['isscript']
@ -146,6 +157,7 @@ class AddressTypeTest(BitcoinTestFramework):
assert_equal(info['witness_version'], 0) assert_equal(info['witness_version'], 0)
assert_equal(len(info['witness_program']), 64) assert_equal(len(info['witness_program']), 64)
assert 'pubkeys' in info assert 'pubkeys' in info
self.test_python_bech32(info["address"])
else: else:
# Unknown type # Unknown type
assert False assert False

Loading…
Cancel
Save