Skip to content

Commit

Permalink
Merge bitcoin#23120: test: Remove unused and confusing main parameter…
Browse files Browse the repository at this point in the history
… from script_util

fa54efd test: pep-8 touched test (MarcoFalke)
fa46768 test: Remove unused and confusing main parameter from script_util (MarcoFalke)

Pull request description:

ACKs for top commit:
  fanquake:
    ACK fa54efd

Tree-SHA512: 124e888e16c92bb24ab4d6f68a768d295500a48f8c6c0b4f069493effcd761f80be78dd93b31edbb529ebe4c8aaca764aaff48d1e3b23659dde722981dc787a5
  • Loading branch information
MarcoFalke committed Sep 29, 2021
2 parents 3c776fd + fa54efd commit 7d4ea7c
Showing 1 changed file with 39 additions and 18 deletions.
57 changes: 39 additions & 18 deletions test/functional/test_framework/script_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,17 @@
# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
"""Useful Script constants and utils."""
from test_framework.script import CScript, hash160, sha256, OP_0, OP_DUP, OP_HASH160, OP_CHECKSIG, OP_EQUAL, OP_EQUALVERIFY
from test_framework.script import (
CScript,
hash160,
sha256,
OP_0,
OP_DUP,
OP_HASH160,
OP_CHECKSIG,
OP_EQUAL,
OP_EQUALVERIFY,
)

# To prevent a "tx-size-small" policy rule error, a transaction has to have a
# non-witness size of at least 82 bytes (MIN_STANDARD_TX_NONWITNESS_SIZE in
Expand All @@ -25,58 +35,69 @@
DUMMY_P2WPKH_SCRIPT = CScript([b'a' * 21])
DUMMY_2_P2WPKH_SCRIPT = CScript([b'b' * 21])

def keyhash_to_p2pkh_script(hash, main = False):

def keyhash_to_p2pkh_script(hash):
assert len(hash) == 20
return CScript([OP_DUP, OP_HASH160, hash, OP_EQUALVERIFY, OP_CHECKSIG])

def scripthash_to_p2sh_script(hash, main = False):

def scripthash_to_p2sh_script(hash):
assert len(hash) == 20
return CScript([OP_HASH160, hash, OP_EQUAL])

def key_to_p2pkh_script(key, main = False):

def key_to_p2pkh_script(key):
key = check_key(key)
return keyhash_to_p2pkh_script(hash160(key), main)
return keyhash_to_p2pkh_script(hash160(key))

def script_to_p2sh_script(script, main = False):

def script_to_p2sh_script(script):
script = check_script(script)
return scripthash_to_p2sh_script(hash160(script), main)
return scripthash_to_p2sh_script(hash160(script))


def key_to_p2sh_p2wpkh_script(key, main = False):
def key_to_p2sh_p2wpkh_script(key):
key = check_key(key)
p2shscript = CScript([OP_0, hash160(key)])
return script_to_p2sh_script(p2shscript, main)
return script_to_p2sh_script(p2shscript)


def program_to_witness_script(version, program, main = False):
def program_to_witness_script(version, program):
if isinstance(program, str):
program = bytes.fromhex(program)
assert 0 <= version <= 16
assert 2 <= len(program) <= 40
assert version > 0 or len(program) in [20, 32]
return CScript([version, program])

def script_to_p2wsh_script(script, main = False):

def script_to_p2wsh_script(script):
script = check_script(script)
return program_to_witness_script(0, sha256(script), main)
return program_to_witness_script(0, sha256(script))


def key_to_p2wpkh_script(key, main = False):
def key_to_p2wpkh_script(key):
key = check_key(key)
return program_to_witness_script(0, hash160(key), main)
return program_to_witness_script(0, hash160(key))


def script_to_p2sh_p2wsh_script(script, main = False):
def script_to_p2sh_p2wsh_script(script):
script = check_script(script)
p2shscript = CScript([OP_0, sha256(script)])
return script_to_p2sh_script(p2shscript, main)
return script_to_p2sh_script(p2shscript)


def check_key(key):
if isinstance(key, str):
key = bytes.fromhex(key) # Assuming this is hex string
key = bytes.fromhex(key) # Assuming this is hex string
if isinstance(key, bytes) and (len(key) == 33 or len(key) == 65):
return key
assert False


def check_script(script):
if isinstance(script, str):
script = bytes.fromhex(script) # Assuming this is hex string
script = bytes.fromhex(script) # Assuming this is hex string
if isinstance(script, bytes) or isinstance(script, CScript):
return script
assert False

0 comments on commit 7d4ea7c

Please sign in to comment.