Davide Grilli
409669e000
- Add pytest test suite for all 5 address types (P2PK, P2PKH, P2SH, P2WPKH, P2TR) - Test output fields, address prefixes per network, key lengths and uniqueness - Each test file prints the SecretScan URL for manual address verification - Add pytest to requirements.txt
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import sys
|
|
import os
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
|
|
|
|
from src.p2tr import generate_p2tr_address
|
|
|
|
SECRETSCAN_URL = "https://secretscan.org/Bitcoin?address={}"
|
|
|
|
NETWORKS = ['mainnet', 'testnet', 'regtest']
|
|
|
|
|
|
def test_p2tr_fields():
|
|
result = generate_p2tr_address('mainnet')
|
|
assert set(result.keys()) == {'network', 'script_type', 'private_key_hex', 'private_key_wif', 'internal_pubkey_x_hex', 'address'}
|
|
|
|
|
|
def test_p2tr_script_type():
|
|
result = generate_p2tr_address('mainnet')
|
|
assert result['script_type'] == 'p2tr'
|
|
|
|
|
|
def test_p2tr_address_mainnet():
|
|
result = generate_p2tr_address('mainnet')
|
|
assert result['address'].startswith('bc1p')
|
|
|
|
|
|
def test_p2tr_address_testnet():
|
|
result = generate_p2tr_address('testnet')
|
|
assert result['address'].startswith('tb1p')
|
|
|
|
|
|
def test_p2tr_address_regtest():
|
|
result = generate_p2tr_address('regtest')
|
|
assert result['address'].startswith('bcrt1p')
|
|
|
|
|
|
def test_p2tr_private_key_length():
|
|
result = generate_p2tr_address('mainnet')
|
|
assert len(result['private_key_hex']) == 64
|
|
|
|
|
|
def test_p2tr_internal_pubkey_length():
|
|
result = generate_p2tr_address('mainnet')
|
|
assert len(result['internal_pubkey_x_hex']) == 64 # x-only pubkey, 32 bytes
|
|
|
|
|
|
def test_p2tr_unique_addresses():
|
|
r1 = generate_p2tr_address('mainnet')
|
|
r2 = generate_p2tr_address('mainnet')
|
|
assert r1['address'] != r2['address']
|
|
|
|
|
|
def test_p2tr_print_secretscan(capsys):
|
|
result = generate_p2tr_address('mainnet')
|
|
url = SECRETSCAN_URL.format(result['address'])
|
|
print(f"\n[P2TR] Address: {result['address']}")
|
|
print(f"[P2TR] Verify on SecretScan: {url}")
|
|
captured = capsys.readouterr()
|
|
assert 'secretscan.org' in captured.out
|