Merge bitcoin/bitcoin#32114: test: Add encodable PUSHDATA1 examples to feature_taproot

f974359e21 test: Add encodable PUSHDATA1 examples to feature_taproot (Greg Sanders)

Pull request description:

  Inspired by discussion in https://github.com/bitcoin/bitcoin/pull/31640#issuecomment-2743492906 I made an example adding coverage I think is missing, with some extra commentary that might help future contributors (including myself when I forget how it works again).

  Open for suggestions how we can make it more welcoming beyond this.

  cc darosior EthanHeilman sipa

ACKs for top commit:
  janb84:
    Re-ACK [f974359](f974359e21)
  rkrux:
    ACK f974359e21

Tree-SHA512: 7544d41c39c13d245a8a33522e53f22b4dd7593c069631978303e5a349cd12cf9d45bed648c391618c4732831232c4b82b8de2bf6cba5bf5e1232501db926122
This commit is contained in:
Ryan Ofsky 2025-04-08 13:56:58 -04:00
commit 24d5033a62
No known key found for this signature in database
GPG key ID: 46800E30FC748A66

View file

@ -1231,6 +1231,41 @@ def spenders_taproot_nonstandard():
return spenders return spenders
def sample_spenders():
# Create key(s) for output creation, as well as key and script-spends
secs = [generate_privkey() for _ in range(2)]
pubs = [compute_xonly_pubkey(sec)[0] for sec in secs]
# Create a list of scripts which will be built into a taptree
scripts = [
# leaf label, followed by CScript
("encodeable_pushdata1", CScript([OP_DROP, OP_PUSHDATA1, b'aa' * 75])),
("nonstd_encodeable_pushdata1", CScript([OP_PUSHDATA1, b'aa'])),
("dummyleaf", CScript([])),
]
# Build TaprootInfo using scripts and appropriate pubkey for output creation
tap = taproot_construct(pubs[0], scripts)
# Finally, add spender(s).
# Each spender embodies a test with an optional failure condition.
# These failure conditions allow for fine-grained success/failure
# conditions that are tested randomly.
spenders = []
# Named comment, using first leaf from scripts, with empty string as witness data, no optional fail condition
add_spender(spenders, comment="tutorial/pushdata1", tap=tap, leaf="encodeable_pushdata1", inputs=[b'\x00'], no_fail=True)
# Spender with alternative failure tapscript via over-riding "failure" dictionary, along with the failure's expected err_msg / ERR_*
add_spender(spenders, comment="tutorial/pushdata1redux", tap=tap, leaf="encodeable_pushdata1", inputs=[b'\x00'], failure={"leaf": "dummyleaf"}, **ERR_NO_SUCCESS)
# Spender that is non-standard but otherwise valid, with extraneous signature data from inner key for optional failure condition
add_spender(spenders, comment="tutorial/nonminpushdata1", tap=tap, leaf="nonstd_encodeable_pushdata1", key=secs[0], standard=False, failure={"inputs": [getter("sign")]}, **ERR_CLEANSTACK)
# New scripts=[] can be defined, and rinse-repeated as necessary until the spenders list is returned for execution
return spenders
# Consensus validation flags to use in dumps for tests with "legacy/" or "inactive/" prefix. # Consensus validation flags to use in dumps for tests with "legacy/" or "inactive/" prefix.
LEGACY_FLAGS = "P2SH,DERSIG,CHECKLOCKTIMEVERIFY,CHECKSEQUENCEVERIFY,WITNESS,NULLDUMMY" LEGACY_FLAGS = "P2SH,DERSIG,CHECKLOCKTIMEVERIFY,CHECKSEQUENCEVERIFY,WITNESS,NULLDUMMY"
# Consensus validation flags to use in dumps for all other tests. # Consensus validation flags to use in dumps for all other tests.
@ -1757,12 +1792,20 @@ class TaprootTest(BitcoinTestFramework):
self.gen_test_vectors() self.gen_test_vectors()
self.log.info("Post-activation tests...") self.log.info("Post-activation tests...")
self.test_spenders(self.nodes[0], spenders_taproot_active(), input_counts=[1, 2, 2, 2, 2, 3])
# New sub-tests not checking standardness can be added to consensus_spenders
# to allow for increased coverage across input types.
# See sample_spenders for a minimal example
consensus_spenders = sample_spenders()
consensus_spenders += spenders_taproot_active()
self.test_spenders(self.nodes[0], consensus_spenders, input_counts=[1, 2, 2, 2, 2, 3])
# Run each test twice; once in isolation, and once combined with others. Testing in isolation # Run each test twice; once in isolation, and once combined with others. Testing in isolation
# means that the standardness is verified in every test (as combined transactions are only standard # means that the standardness is verified in every test (as combined transactions are only standard
# when all their inputs are standard). # when all their inputs are standard).
self.test_spenders(self.nodes[0], spenders_taproot_nonstandard(), input_counts=[1]) nonstd_spenders = spenders_taproot_nonstandard()
self.test_spenders(self.nodes[0], spenders_taproot_nonstandard(), input_counts=[2, 3]) self.test_spenders(self.nodes[0], nonstd_spenders, input_counts=[1])
self.test_spenders(self.nodes[0], nonstd_spenders, input_counts=[2, 3])
if __name__ == '__main__': if __name__ == '__main__':