def test_exceptions() -> None:
# invalid size: 11 bytes instead of 20
err_msg = "invalid size: "
with pytest.raises(BTClibValueError, match=err_msg):
script_pubkey_from_payload("p2wpkh", "00" * 11)
# invalid size: 33 bytes instead of 32
with pytest.raises(BTClibValueError, match=err_msg):
script_pubkey_from_payload("p2wsh", "00" * 33)
err_msg = "unknown script_pubkey type: "
with pytest.raises(BTClibValueError, match=err_msg):
script_pubkey_from_payload("p2unkn", "00" * 32)
err_msg = "unknown script_pubkey: "
with pytest.raises(BTClibValueError, match=err_msg):
script_pubkey = [16, 20 * b"\x00"]
address_from_script_pubkey(script_pubkey)
# Unhandled witness version (16)
err_msg = "unmanaged witness version: "
address = b32address_from_witness(16, 20 * b"\x00")
with pytest.raises(BTClibValueError, match=err_msg):
script_pubkey_from_address(address)
def test_p2wsh() -> None:
# self-consistency
pubkey = "02 cc71eb30d653c0c3163990c47b976f3fb3f37cccdcbedb169a1dfef58bbfbfaf"
pubkey_hash = hash160(pubkey)
redeem_script = script_pubkey_from_payload("p2pkh", pubkey_hash)
payload = sha256(redeem_script)
script_pubkey = script.serialize([0, payload])
assert script_pubkey == p2wsh(script.deserialize(redeem_script))
# to the script_pubkey in two steps (through payload)
script_type = "p2wsh"
assert script_pubkey == script_pubkey_from_payload(script_type, payload)
# back from the script_pubkey to the payload
assert (script_type, payload,
0) == payload_from_script_pubkey(script_pubkey)
# bech32 address
network = "mainnet"
address = bech32address.p2wsh(redeem_script, network)
assert address == address_from_script_pubkey(script_pubkey, network)
wit_ver = 0
assert address == b32address_from_witness(wit_ver, payload, network)
# back from the address to the script_pubkey
assert (script_pubkey, network) == script_pubkey_from_address(address)
# p2sh-wrapped base58 address
address = base58address.p2wsh_p2sh(redeem_script, network)
assert address == b58address_from_witness(payload, network)
def test_p2pkh() -> None:
# self-consistency
pubkey = (
"04 "
"cc71eb30d653c0c3163990c47b976f3fb3f37cccdcbedb169a1dfef58bbfbfaf"
"f7d8a473e7e2e6d317b87bafe8bde97e3cf8f065dec022b51d11fcdd0d348ac4")
payload = hash160(pubkey)
script_pubkey = script.serialize(
["OP_DUP", "OP_HASH160", payload, "OP_EQUALVERIFY", "OP_CHECKSIG"])
assert script_pubkey == p2pkh(pubkey)
# to the script_pubkey in two steps (through payload)
script_type = "p2pkh"
assert script_pubkey == script_pubkey_from_payload(script_type, payload)
# back from the script_pubkey to the payload
assert (script_type, payload,
0) == payload_from_script_pubkey(script_pubkey)
# base58 address
network = "mainnet"
address = base58address.p2pkh(pubkey, network)
assert address == address_from_script_pubkey(script_pubkey, network)
prefix = NETWORKS[network].p2pkh
assert address == b58address_from_h160(prefix, payload, network)
# back from the address to the script_pubkey
assert (script_pubkey, network) == script_pubkey_from_address(address)
# documented test case: https://learnmeabitcoin.com/guide/p2pkh
payload = "12ab8dc588ca9d5787dde7eb29569da63c3a238c"
script_pubkey = "76a914" + payload + "88ac"
assert script_pubkey == script_pubkey_from_payload(script_type,
payload).hex()
address = b"12higDjoCCNXSA95xZMWUdPvXNmkAduhWv"
assert address == address_from_script_pubkey(script_pubkey, network)
assert (bytes.fromhex(script_pubkey),
network) == script_pubkey_from_address(address)
# invalid size: 11 bytes instead of 20
err_msg = "invalid size: "
with pytest.raises(BTClibValueError, match=err_msg):
script_pubkey_from_payload(script_type, "00" * 11)
def test_p2sh() -> None:
# self-consistency
pubkey = "02 cc71eb30d653c0c3163990c47b976f3fb3f37cccdcbedb169a1dfef58bbfbfaf"
pubkey_hash = hash160(pubkey)
redeem_script = script_pubkey_from_payload("p2pkh", pubkey_hash)
payload = hash160(redeem_script)
script_pubkey = script.serialize(["OP_HASH160", payload, "OP_EQUAL"])
assert script_pubkey == p2sh(redeem_script)
# to the script_pubkey in two steps (through payload)
script_type = "p2sh"
assert script_pubkey == script_pubkey_from_payload(script_type, payload)
# back from the script_pubkey to the payload
assert (script_type, payload,
0) == payload_from_script_pubkey(script_pubkey)
# base58 address
network = "mainnet"
address = base58address.p2sh(script.deserialize(redeem_script), network)
assert address == address_from_script_pubkey(script_pubkey, network)
prefix = NETWORKS[network].p2sh
assert address == b58address_from_h160(prefix, payload, network)
# back from the address to the script_pubkey
assert (script_pubkey, network) == script_pubkey_from_address(address)
# documented test case: https://learnmeabitcoin.com/guide/p2sh
payload = "748284390f9e263a4b766a75d0633c50426eb875"
script_pubkey = "a914" + payload + "87"
assert script_pubkey == script_pubkey_from_payload(script_type,
payload).hex()
address = b"3CK4fEwbMP7heJarmU4eqA3sMbVJyEnU3V"
assert address == address_from_script_pubkey(script_pubkey, network)
assert (bytes.fromhex(script_pubkey),
network) == script_pubkey_from_address(address)
# invalid size: 21 bytes instead of 20
err_msg = "invalid size: "
with pytest.raises(BTClibValueError, match=err_msg):
script_pubkey_from_payload(script_type, "00" * 21)